# This Resque extension changes the resque default logger to monologger
# and formats the log in json format.
# 
# Monologger supports printing logs in trap block.
# 
module Resque
  module Plugins
    module CustomLogger
      def before_perform(*args)
        case args.class.to_s
        when "Array"
          if args.first.class == Hash
            data = args.first.merge({:worker_class => self.to_s})
          else
            data = {:worker_class => self.to_s, :args => args}
          end
        when "Hash"
          data = args.merge({:worker_class => self.to_s})
        end
        if Rails.logger.is_a?(Ougai::Logger) 
          Rails.logger.with_fields = {job: data, trace_id: SecureRandom.uuid, name: "RailsWorker"}
        end
        data = {:msg => 'Starting job', :job => data}
        if data.dig(:job, 'app_instance_id').present?
          data[:app_instance_id] = data.dig(:job, 'app_instance_id')
        end

        Rails.logger.info(data) if data.present?
      end
    end
  end
end