dead_bro 0.2.5 → 0.2.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 95efdf7e1aadb84f27739eb59612c8ad3bf0c6a5315d12497322140fdd0ca676
-  data.tar.gz: c9285dd8f024f10d974b04b76e98a79f365bb6e07ab1fa665a8340cc34bc69e7
+  metadata.gz: 7e977f41766dfbcf7ca52cf14aafb217595f35860746a553bd9386b1ddfe7797
+  data.tar.gz: 24858e179f45181d6d144352347ed3f760a19abe53cead83c80eb4644e281f39
 SHA512:
-  metadata.gz: 7b12f73db0672c45f624b0a3916b65c95c8d0929e491ef173832b5f960cfb073b2cf0b337a2071908f5a4bbe768551e851747c56e5a90a704c76719e13449773
-  data.tar.gz: b99e770dae2b68d773eee279b02c78821244bf9c0322e91776301611d0204a0ebdd3c3b5e0c4bf62aec9e5a403bdb4c66ee35f29907fb7d5e68ae978eca33c4d
+  metadata.gz: 7310571657927f6404088b2605d710cb9e107d5e63231cddcf785ffc9cc22a78ff634d5bc5969e34d08e634ed0cdc46f85f8f2d6937bc05ebe9af14b00c4968c
+  data.tar.gz: eaaec05db2cdf1d2da03cfa873c333286d891e33cb2e37d6b347b319a3ae8481f142182d87c3ce20bd2990829075aa236e52c9f52388d2799fc862b1511ceabd

data/lib/dead_bro/memory_tracking_subscriber.rb

@@ -6,6 +6,7 @@ module DeadBro
   class MemoryTrackingSubscriber
     # Object allocation events
     ALLOCATION_EVENT = "object_allocations.active_support"
+    PROCESS_ACTION_EVENT = "process_action.action_controller"
 
     THREAD_LOCAL_KEY = :dead_bro_memory_events
     # Consider objects larger than this many bytes as "large"
@@ -28,6 +29,23 @@ module DeadBro
             next unless rand < ALLOCATION_SAMPLING_RATE
             track_allocation(data, started, finished)
           end
+
+          # Subscribe to process_action to capture request-level allocation counters
+          ActiveSupport::Notifications.subscribe(PROCESS_ACTION_EVENT) do |*args|
+            event = if args.length == 1 && args.first.is_a?(ActiveSupport::Notifications::Event)
+              args.first
+            else
+              ActiveSupport::Notifications::Event.new(*args)
+            end
+            allocations = event.respond_to?(:allocations) ? event.allocations : event.payload[:allocations]
+            allocated_bytes = event.respond_to?(:allocated_bytes) ? event.allocated_bytes : event.payload[:allocated_bytes]
+            next unless allocations || allocated_bytes
+
+            record_request_allocations(
+              allocations: allocations,
+              allocated_bytes: allocated_bytes
+            )
+          end
         rescue
           # Allocation tracking might not be available in all Ruby versions
         end
@@ -44,9 +62,10 @@ module DeadBro
         allocations: [],
         memory_snapshots: [],
         large_objects: [],
+        request_allocations: nil,
         gc_before: gc_stats,
         memory_before: memory_usage_mb,
-        start_time: Time.now.utc.to_i,
+        start_time: Time.now.to_f,
         object_counts_before: count_objects_snapshot
       }
     end
@@ -58,7 +77,7 @@ module DeadBro
       if events
         events[:gc_after] = gc_stats
         events[:memory_after] = memory_usage_mb
-        events[:end_time] = Time.now.utc.to_i
+        events[:end_time] = Time.now.to_f
         events[:duration_seconds] = events[:end_time] - events[:start_time]
         events[:object_counts_after] = count_objects_snapshot
 
@@ -71,6 +90,16 @@ module DeadBro
       events || {}
     end
 
+    # Record request-level allocation counters from Rails instrumentation.
+    def self.record_request_allocations(allocations:, allocated_bytes:)
+      return unless Thread.current[THREAD_LOCAL_KEY]
+
+      Thread.current[THREAD_LOCAL_KEY][:request_allocations] = {
+        allocations: allocations,
+        allocated_bytes: allocated_bytes
+      }
+    end
+
     def self.track_allocation(data, started, finished)
       return unless Thread.current[THREAD_LOCAL_KEY]
 
@@ -122,6 +151,7 @@ module DeadBro
       allocations = memory_events[:allocations] || []
       large_objects = memory_events[:large_objects] || []
       snapshots = memory_events[:memory_snapshots] || []
+      request_allocations = memory_events[:request_allocations]
 
       # Calculate memory growth
       memory_growth = 0
@@ -132,6 +162,19 @@ module DeadBro
       # Calculate allocation totals
       total_allocations = allocations.sum { |a| a[:count] }
       total_allocated_size = allocations.sum { |a| a[:size] }
+      if request_allocations
+        total_allocated_size = request_allocations[:allocated_bytes].to_i if total_allocated_size.zero?
+      end
+      gc_allocations = nil
+      if memory_events[:gc_before] && memory_events[:gc_after]
+        gc_allocations = (memory_events[:gc_after][:total_allocated_objects] || 0) -
+          (memory_events[:gc_before][:total_allocated_objects] || 0)
+      end
+      if gc_allocations.to_i > 0
+        total_allocations = gc_allocations
+      elsif total_allocations.zero? && request_allocations
+        total_allocations = request_allocations[:allocations].to_i
+      end
 
       # Group allocations by class
       allocations_by_class = allocations.group_by { |a| a[:class_name] }
@@ -174,7 +217,8 @@ module DeadBro
           (total_allocations.to_f / memory_events[:duration_seconds]).round(2) : 0,
         top_allocating_classes: top_allocating_classes.map { |class_name, data|
           {
-            class_name: class_name,
+            class: class_name,
+            name: class_name,
             count: data[:count],
             size: data[:size],
             size_mb: (data[:size] / 1_000_000.0).round(2)

data/lib/dead_bro/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DeadBro
-  VERSION = "0.2.5"
+  VERSION = "0.2.6"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dead_bro
 version: !ruby/object:Gem::Version
-  version: 0.2.5
+  version: 0.2.6
 platform: ruby
 authors:
 - Emanuel Comsa