Sidekiq jobs support (silva96#50)

* - Add logger for better development debug
- Take care of orphan logs (before their parent request exist, or after)
- Add sidekiq jobs support

* Fix standard

Author: silva96

lib/log_bench.rb

@@ -20,6 +20,22 @@ def setup
       yield(configuration) if block_given?
       configuration
     end
+
+    def logger
+      @logger ||= create_debug_logger
+    end
+
+    private
+
+    def create_debug_logger
+      require "logger"
+      logger = Logger.new("logbench_log.log")
+      logger.level = Logger::DEBUG
+      logger.formatter = proc do |severity, datetime, _progname, msg|
+        "[#{datetime.strftime("%Y-%m-%d %H:%M:%S")}] #{severity}: #{msg}\n"
+      end
+      logger
+    end
   end
 end
 

lib/log_bench/app/monitor.rb

@@ -32,6 +32,7 @@ def monitor_loop
           begin
             log_file.watch do |new_collection|
               add_new_requests(new_collection.requests)
+              add_orphan_requests(new_collection.orphan_requests)
             end
           rescue
             sleep 1
@@ -46,6 +47,23 @@ def add_new_requests(new_requests)
         keep_recent_requests
       end
 
+      def add_orphan_requests(orphan_requests)
+        state.orphan_requests.concat(orphan_requests)
+        return if state.orphan_requests.empty?
+
+        # Try to attach orphaned logs to existing requests and remove them if successful
+        state.orphan_requests.reject! do |orphan_request|
+          request = state.requests.find { |req| req.request_id == orphan_request.request_id }
+
+          if request
+            orphan_request.related_logs.each { |log| request.add_related_log(log) }
+            true # Remove this orphan request from the list
+          else
+            false # Keep this orphan request for later
+          end
+        end
+      end
+
       def keep_recent_requests
         # Keep only the last 1000 requests to prevent memory issues
         state.requests = state.requests.last(1000) if state.requests.size > 1000

lib/log_bench/app/state.rb

@@ -4,10 +4,11 @@ module LogBench
   module App
     class State
       attr_reader :main_filter, :sort, :detail_filter, :cleared_requests
-      attr_accessor :requests, :auto_scroll, :scroll_offset, :selected, :detail_scroll_offset, :detail_selected_entry, :text_selection_mode, :update_available, :update_version
+      attr_accessor :requests, :orphan_requests, :auto_scroll, :scroll_offset, :selected, :detail_scroll_offset, :detail_selected_entry, :text_selection_mode, :update_available, :update_version
 
       def initialize
         self.requests = []
+        self.orphan_requests = []
         self.selected = 0
         self.scroll_offset = 0
         self.auto_scroll = true

lib/log_bench/log/collection.rb

@@ -11,10 +11,6 @@ def initialize(input)
         self.entries = parse_input(input)
       end
 
-      def each(&block)
-        entries.each(&block)
-      end
-
       def size
         entries.size
       end
@@ -24,7 +20,11 @@ def empty?
       end
 
       def requests
-        entries.select { |entry| entry.is_a?(Request) }
+        entries.select { |entry| entry.is_a?(Request) && !entry.orphan }
+      end
+
+      def orphan_requests
+        entries.select { |entry| entry.is_a?(Request) && entry.orphan }
       end
 
       def filter_by_method(method)
@@ -57,10 +57,6 @@ def sort_by_timestamp
         create_collection_from_requests(sorted_requests)
       end
 
-      def to_a
-        entries
-      end
-
       private
 
       def create_collection_from_requests(requests)

lib/log_bench/log/parser.rb

@@ -43,8 +43,7 @@ def self.build_requests_from_groups(grouped)
         grouped.each do |request_id, entries|
           next unless request_id
 
-          request = find_request_entry(entries)
-          next unless request
+          request = find_request_entry(entries) || Request.new_orphan(request_id)
 
           related_logs = find_related_logs(entries)
           related_logs.each { |log| request.add_related_log(log) }

lib/log_bench/log/request.rb

@@ -3,10 +3,10 @@
 module LogBench
   module Log
     class Request < Entry
-      attr_reader :method, :path, :status, :duration, :controller, :action, :params, :related_logs
+      attr_reader :method, :path, :status, :duration, :controller, :action, :params, :related_logs, :orphan
 
-      def initialize(json_data)
-        super
+      def initialize(json_data, orphan: false)
+        super(json_data)
         self.type = :http_request
         self.related_logs = []
         self.method = json_data["method"]
@@ -16,6 +16,11 @@ def initialize(json_data)
         self.controller = json_data["controller"]
         self.action = json_data["action"]
         self.params = parse_params(json_data["params"])
+        self.orphan = orphan
+      end
+
+      def self.new_orphan(request_id)
+        new({"request_id" => request_id}, orphan: true)
       end
 
       def add_related_log(log_entry)
@@ -63,7 +68,7 @@ def server_error?
 
       private
 
-      attr_writer :method, :path, :status, :duration, :controller, :action, :params
+      attr_writer :method, :path, :status, :duration, :controller, :action, :params, :orphan
 
       def related_logs=(value)
         @related_logs = value

lib/log_bench/railtie.rb

@@ -48,6 +48,7 @@ class Railtie < Rails::Railtie
       if LogBench.configuration.enabled
         LogBench::Railtie.setup_rails_logger_final
         LogBench::Railtie.setup_current_attributes
+        LogBench::Railtie.setup_sidekiq_middleware
         LogBench::Railtie.validate_configuration!
       end
     end
@@ -96,6 +97,12 @@ def set_current_request_id
         end
       end
 
+      def setup_sidekiq_middleware
+        # Use Sidekiq's built-in CurrentAttributes middleware
+        require "sidekiq/middleware/current_attributes"
+        Sidekiq::CurrentAttributes.persist("LogBench::Current")
+      end
+
       # Validate that LogBench setup worked correctly
       def validate_configuration!
         ConfigurationValidator.validate_rails_config!