memory-profiler 1.1.6 → 1.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ea6e765ad50aed3c3cf538d5c1a44f102f1bbe5251d5fd9c0995dd9d6890282a
-  data.tar.gz: dac87b8b0bd69f57ce3a290ef84f6eae56192ab422e554f861a2cb89b3cfa290
+  metadata.gz: 8bb750e881686089136fbcd20b073e857dc3a7dc7f4784f0a75cf8fbb17444ec
+  data.tar.gz: 75d398699455359a70dafbbd1c71ce8dd529faf492f8fd7f595cc9714101061f
 SHA512:
-  metadata.gz: cc91cd05c5cf0a2680fefe8745d8be48f7d85c0e7f358326da2ee0b58a07b01363fd9238b3b00bc9f17d5df8f216da726ebbf6dfdd77242685753ebf285183e9
-  data.tar.gz: df0498c707e90556d7df5319aa22775c1912f5961c5ea4f8d60f5314a9bdc8920ac60f367ef1d29e912789974857a3275af3e680a2ad2447651366e53d5685df
+  metadata.gz: 83261c47a3365c107239cd626b61afe496e51d17e04ee36ac81235d7493997089fd97648965af42747ff1b3f8074f1fe72df31a1ecf6ce774fe9ac1ae2cf0565
+  data.tar.gz: 5bf3be79ccabbc16f1d753c94a3af7d89affabfc34d4feb9fd001f0573cf516614363f7430dfbe3553166076c349ffe38cee7e39309f9f6eecef867ba4978d78

data/ext/memory/profiler/capture.c

@@ -694,6 +694,57 @@ static VALUE Memory_Profiler_Capture_aref(VALUE self, VALUE klass) {
 	return Qnil;
 }
 
+// Struct to accumulate statistics during iteration
+struct Memory_Profiler_Allocations_Statistics {
+	size_t total_tracked_objects;
+	VALUE per_class_counts;
+};
+
+// Iterator callback to count object_states per class
+static int Memory_Profiler_Capture_count_object_states(st_data_t key, st_data_t value, st_data_t argument) {
+	struct Memory_Profiler_Allocations_Statistics *statistics = (struct Memory_Profiler_Allocations_Statistics *)argument;
+	VALUE klass = (VALUE)key;
+	VALUE allocations = (VALUE)value;
+	struct Memory_Profiler_Capture_Allocations *record = Memory_Profiler_Allocations_get(allocations);
+	
+	size_t object_states_count = record->object_states ? record->object_states->num_entries : 0;
+	statistics->total_tracked_objects += object_states_count;
+	
+	rb_hash_aset(statistics->per_class_counts, klass, SIZET2NUM(object_states_count));
+	return ST_CONTINUE;
+}
+
+// Get internal statistics for debugging
+// Returns hash with internal state sizes
+static VALUE Memory_Profiler_Capture_statistics(VALUE self) {
+	struct Memory_Profiler_Capture *capture;
+	TypedData_Get_Struct(self, struct Memory_Profiler_Capture, &Memory_Profiler_Capture_type, capture);
+	
+	VALUE statistics = rb_hash_new();
+	
+	// Tracked classes count
+	rb_hash_aset(statistics, ID2SYM(rb_intern("tracked_classes_count")), SIZET2NUM(capture->tracked_classes->num_entries));
+	
+	// Queue sizes
+	rb_hash_aset(statistics, ID2SYM(rb_intern("newobj_queue_size")), SIZET2NUM(capture->newobj_queue.count));
+	rb_hash_aset(statistics, ID2SYM(rb_intern("newobj_queue_capacity")), SIZET2NUM(capture->newobj_queue.capacity));
+	rb_hash_aset(statistics, ID2SYM(rb_intern("freeobj_queue_size")), SIZET2NUM(capture->freeobj_queue.count));
+	rb_hash_aset(statistics, ID2SYM(rb_intern("freeobj_queue_capacity")), SIZET2NUM(capture->freeobj_queue.capacity));
+	
+	// Count object_states entries for each tracked class
+	struct Memory_Profiler_Allocations_Statistics allocations_statistics = {
+		.total_tracked_objects = 0,
+		.per_class_counts = rb_hash_new()
+	};
+	
+	st_foreach(capture->tracked_classes, Memory_Profiler_Capture_count_object_states, (st_data_t)&allocations_statistics);
+	
+	rb_hash_aset(statistics, ID2SYM(rb_intern("total_tracked_objects")), SIZET2NUM(allocations_statistics.total_tracked_objects));
+	rb_hash_aset(statistics, ID2SYM(rb_intern("tracked_objects_per_class")), allocations_statistics.per_class_counts);
+	
+	return statistics;
+}
+
 void Init_Memory_Profiler_Capture(VALUE Memory_Profiler)
 {
 	// Initialize event symbols
@@ -715,6 +766,7 @@ void Init_Memory_Profiler_Capture(VALUE Memory_Profiler)
 	rb_define_method(Memory_Profiler_Capture, "each", Memory_Profiler_Capture_each, 0);
 	rb_define_method(Memory_Profiler_Capture, "[]", Memory_Profiler_Capture_aref, 1);
 	rb_define_method(Memory_Profiler_Capture, "clear", Memory_Profiler_Capture_clear, 0);
+	rb_define_method(Memory_Profiler_Capture, "statistics", Memory_Profiler_Capture_statistics, 0);
 	
 	// Initialize Allocations class
 	Init_Memory_Profiler_Allocations(Memory_Profiler);

data/lib/memory/profiler/sampler.rb

@@ -85,8 +85,6 @@ module Memory
 				end
 			end
 			
-			attr_reader :depth
-			
 			# Create a new memory sampler.
 			#
 			# @parameter depth [Integer] Number of stack frames to capture for call path analysis.
@@ -104,6 +102,30 @@ module Memory
 				@call_trees = {}
 				@samples = {}
 			end
+
+			# @attribute [Integer] The depth of the call tree.
+			attr :depth
+			
+			# @attribute [Proc] The filter to exclude frames from call paths.
+			attr :filter
+
+			# @attribute [Integer] The number of increases before enabling detailed tracking.
+			attr :increases_threshold
+
+			# @attribute [Integer] The number of insertions before auto-pruning (nil = no auto-pruning).
+			attr :prune_limit
+
+			# @attribute [Integer | Nil] The number of insertions before auto-pruning (nil = no auto-pruning).
+			attr :prune_threshold
+
+			# @attribute [Capture] The capture object.
+			attr :capture
+
+			# @attribute [Hash] The call trees.
+			attr :call_trees
+
+			# @attribute [Hash] The samples for each class being tracked.
+			attr :samples
 			
 			# Start capturing allocations.
 			def start

data/lib/memory/profiler/version.rb

@@ -7,7 +7,7 @@
 module Memory
 	# @namespace
 	module Profiler
-		VERSION = "1.1.6"
+		VERSION = "1.1.7"
 	end
 end
 

data/readme.md

@@ -22,6 +22,10 @@ Please see the [project documentation](https://socketry.github.io/memory-profile
 
 Please see the [project releases](https://socketry.github.io/memory-profiler/releases/index) for all releases.
 
+### v1.1.7
+
+  - Expose `Capture#statistics` for debugging internal memory tracking state.
+
 ### v1.1.6
 
   - Write barriers all the things.

data/releases.md

@@ -1,5 +1,9 @@
 # Releases
 
+## v1.1.7
+
+  - Expose `Capture#statistics` for debugging internal memory tracking state.
+
 ## v1.1.6
 
   - Write barriers all the things.

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: memory-profiler
 version: !ruby/object:Gem::Version
-  version: 1.1.6
+  version: 1.1.7
 platform: ruby
 authors:
 - Samuel Williams