Merge pull request #109 from FundingCircle/loga-sidekiq-integration

Loga Sidekiq integration

Author: joromir

.gitignore

@@ -25,3 +25,5 @@ spec/fixtures/**/*.log
 gemfiles/*.lock
 vendor/bundle
 .byebug_history
+.DS_Store
+

.rubocop.yml

@@ -4,6 +4,7 @@ AllCops:
     - 'gemfiles/*'
     - 'spec/fixtures/**/*'
     - 'vendor/bundle/**/*'
+    - 'gemfiles/vendor/bundle/**/*'
 
 require: rubocop-rspec
 

Appraisals

@@ -24,5 +24,9 @@ appraise 'rails52' do
   gem 'rails', '~> 5.2.0'
 end
 
+appraise 'sidekiq51' do
+  gem 'sidekiq', '~> 5.1.0'
+end
+
 appraise 'unit' do
 end

CHANGELOG.md

@@ -4,6 +4,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/)
 and this project adheres to [Semantic Versioning](http://semver.org/).
 
+## [2.3.0] - 2018-06-29
+### Added
+Support for Sidekiq `~> 5.0`.
+
 ## [2.2.0] - 2018-05-10
 ### Added
 Subscribe to `ActionMailer` events

Guardfile

@@ -39,6 +39,20 @@ group :rails do
   end
 end
 
+group :sidekiq do
+  cmd = 'bundle exec appraisal sidekiq51 rspec'
+
+  guard :rspec, all_on_start: true, cmd: cmd do
+    watch('lib/loga/sidekiq/job_logger.rb') do
+      [
+        'spec/integration/sidekiq_spec.rb',
+        'spec/loga/sidekiq/job_logger_spec.rb',
+        'spec/loga/sidekiq_spec.rb',
+      ]
+    end
+  end
+end
+
 group :unit do
   guard :rspec, cmd: 'bundle exec appraisal unit rspec' do
     watch(%r{^spec/unit/.+_spec\.rb$})

README.md

@@ -148,6 +148,10 @@ environment variable therefore it must be removed.
 LOGA_FORMAT=simple rackup
 ```
 
+### Sidekiq
+
+Loga `2.3` provides an out-of-the-box support for `Sidekiq ~> 5.0`.
+
 ## Output Example
 
 ### GELF Format

gemfiles/sidekiq51.gemfile

@@ -0,0 +1,11 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "sidekiq", "~> 5.1.0"
+
+group :test do
+  gem "simplecov"
+end
+
+gemspec path: "../"

lib/loga.rb

@@ -8,8 +8,8 @@
 require 'loga/rack/request'
 require 'loga/rack/request_id'
 require 'loga/railtie' if defined?(Rails)
+require 'loga/sidekiq'
 
-# rubocop:disable Naming/MemoizedInstanceVariableName
 module Loga
   ConfigurationError = Class.new(StandardError)
 
@@ -26,6 +26,8 @@ def self.configure(options, framework_options = {})
     raise ConfigurationError, 'Loga has already been configured' if @configuration
 
     @configuration ||= Configuration.new(options, framework_options)
+
+    Loga::Sidekiq.configure_logging
   end
 
   def self.logger
@@ -36,4 +38,3 @@ def self.reset
     @configuration = nil
   end
 end
-# rubocop:enable Naming/MemoizedInstanceVariableName

lib/loga/sidekiq.rb

@@ -0,0 +1,16 @@
+require 'loga/sidekiq/job_logger'
+
+module Loga
+  module Sidekiq
+    def self.configure_logging
+      return unless defined?(::Sidekiq)
+      return if Gem::Version.new(::Sidekiq::VERSION) < Gem::Version.new('5.0')
+
+      ::Sidekiq.configure_server do |config|
+        config.options[:job_logger] = Loga::Sidekiq::JobLogger
+      end
+
+      ::Sidekiq::Logging.logger = Loga.configuration.logger
+    end
+  end
+end

lib/loga/sidekiq/job_logger.rb

@@ -0,0 +1,62 @@
+module Loga
+  module Sidekiq
+    # The approach of using a custom job logger in sidekiq was introduced
+    # in v5.0: https://github.com/mperham/sidekiq/pull/3235
+    # This job logger does not support logging for Sidekiq versions
+    # that are before 5.0
+    class JobLogger
+      include Loga::Utilities
+
+      EVENT_TYPE = 'sidekiq'.freeze
+
+      attr_reader :started_at, :data
+
+      def initialize
+        @started_at = Time.now
+        @data = {}
+      end
+
+      def call(item, _queue)
+        yield
+      rescue Exception => ex # rubocop:disable Lint/RescueException
+        data['exception'] = ex
+
+        raise
+      ensure
+        assign_data(item)
+        send_message
+      end
+
+      private
+
+      def assign_data(item)
+        data['created_at']  = item['created_at']
+        data['enqueued_at'] = item['enqueued_at']
+        data['jid']         = item['jid']
+        data['queue']       = item['queue']
+        data['retry']       = item['retry']
+        data['params']      = item['args']
+        data['class']       = item['class']
+        data['duration']    = duration_in_ms(started_at)
+      end
+
+      def short_message
+        "#{data['class']} with jid: '#{data['jid']}' executed in #{data['duration']}ms"
+      end
+
+      def send_message
+        event = Event.new(data: data, message: short_message, type: EVENT_TYPE)
+
+        logger.public_send(compute_level, event)
+      end
+
+      def compute_level
+        data.key?('exception') ? :warn : :info
+      end
+
+      def logger
+        Loga.logger
+      end
+    end
+  end
+end