ActiveJob extension for preserving logs

lib/good_job.rb

@@ -8,6 +8,7 @@
 require "good_job/adapter"
 require "active_job/queue_adapters/good_job_adapter"
 require "good_job/active_job_extensions/concurrency"
+require "good_job/active_job_extensions/logging"
 
 require "good_job/assignable_connection"
 require "good_job/cleanup_tracker"

lib/good_job/active_job_extensions/logging.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+module GoodJob
+  module ActiveJobExtensions
+    module Logging
+      extend ActiveSupport::Concern
+
+      included do
+        around_perform do |job, block|
+          original_logger = job.logger
+          job.logger = ActiveSupport::TaggedLogging.new(ActiveSupport::Logger.new(LogDevice.new(job)).extend(ActiveSupport::Logger.broadcast(original_logger)))
+          block.call
+        ensure
+          job.logger = original_logger
+        end
+      end
+
+      class LogDevice
+        cattr_accessor :logs, default: Concurrent::Array.new
+
+        def initialize(job)
+          @job = job
+        end
+
+        def write(message)
+          self.class.logs << [@job.provider_job_id, message.strip]
+        end
+
+        def close
+          nil
+        end
+
+        def reopen(_log = nil)
+          nil
+        end
+      end
+    end
+  end
+end

spec/lib/good_job/active_job_extensions/logging_spec.rb

@@ -0,0 +1,37 @@
+# spec/lib/good_job/active_job_extensions/logging_spec.rb
+
+# frozen_string_literal: true
+require 'rails_helper'
+
+RSpec.describe GoodJob::ActiveJobExtensions::Logging do
+  before do
+    # freeze_time
+    ActiveJob::Base.queue_adapter = GoodJob::Adapter.new(execution_mode: :external)
+
+    stub_const 'TestJob', (Class.new(ActiveJob::Base) do
+      include GoodJob::ActiveJobExtensions::Logging
+
+      def perform
+        logger.info "Hello, world!"
+        logger.debug "Debug level!"
+        logger.tagged("TAG") { logger.info "Tagged!" }
+      end
+    end)
+  end
+
+  # very naive test case, please modify based on implementation
+  describe '.logs' do
+    it 'stores the logs from the job in a tuple of execution ID and log' do
+      active_job = TestJob.perform_later
+      GoodJob.perform_inline
+
+      job_log = described_class::LogDevice.logs
+      # I expect this tuple would be replaced with a better object eventually, but that can be deferred to later checkpoints.
+      expect(job_log).to eq([
+                              [active_job.provider_job_id, 'Hello, world!'],
+                              [active_job.provider_job_id, 'Debug level!'],
+                              [active_job.provider_job_id, '[TAG] Tagged!'],
+                            ])
+    end
+  end
+end