feat(yabeda): Fire :closed callbacks on Sentry.close for final flush

Add a :closed callback event to Sentry::Configuration, fired by
Sentry.close before client.flush. sentry-yabeda registers an
after(:closed) hook that performs one final Yabeda.collect! and
then kills the collector, ensuring any gauge values set by
collection blocks are captured in the last envelope.

The existing after(:configured) guard (collector&.kill) is kept to
handle re-initialisation without an explicit close call.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: dingsdax

sentry-ruby/lib/sentry-ruby.rb

@@ -268,6 +268,7 @@ def close
       end
 
       if client = get_current_client
+        client.configuration.run_after_close_callbacks
         client.flush
 
         if client.configuration.include_local_variables

sentry-ruby/lib/sentry/configuration.rb

@@ -461,7 +461,8 @@ def after(event, &block)
       def callbacks
         @callbacks ||= {
           initialize: { before: [], after: [] },
-          configured: { before: [], after: [] }
+          configured: { before: [], after: [] },
+          closed: { before: [], after: [] }
         }
       end
 
@@ -798,6 +799,11 @@ def error_messages
       @errors.join(", ")
     end
 
+    # @api private
+    def run_after_close_callbacks
+      run_callbacks(:after, :closed)
+    end
+
     private
 
     def init_dsn(dsn_string)

sentry-ruby/spec/sentry/configuration_spec.rb

@@ -569,6 +569,20 @@ class SentryConfigurationSample < Sentry::Configuration
     end
   end
 
+  describe '#run_after_close_callbacks' do
+    it 'calls hooks registered with after(:closed)' do
+      called = false
+
+      config = Class.new(Sentry::Configuration) do
+        after(:closed) do
+          called = true
+        end
+      end.new
+
+      expect { config.run_after_close_callbacks }.to change { called }.from(false).to(true)
+    end
+  end
+
   describe "#skip_rake_integration" do
     it "returns false by default" do
       expect(subject.skip_rake_integration).to eq(false)

sentry-yabeda/lib/sentry/yabeda/configuration.rb

@@ -8,5 +8,13 @@ class Configuration
         Sentry::Yabeda.collector = Sentry::Yabeda::Collector.new(self)
       end
     end
+
+    after(:closed) do
+      if (collector = Sentry::Yabeda.collector)
+        collector.run
+        collector.kill
+        Sentry::Yabeda.collector = nil
+      end
+    end
   end
 end

sentry-yabeda/spec/sentry/yabeda/collector_spec.rb

@@ -21,14 +21,32 @@
     end
   end
 
+  describe "on close" do
+    it "performs a final collect before shutting down" do
+      expect(::Yabeda).to receive(:collect!)
+      Sentry.close
+    end
+
+    it "kills the collector and sets it to nil" do
+      collector = Sentry::Yabeda.collector
+      expect(collector).to receive(:kill).and_call_original
+      Sentry.close
+      expect(Sentry::Yabeda.collector).to be_nil
+    end
+
+    it "does nothing when no collector is running" do
+      Sentry::Yabeda.collector = nil
+      expect { Sentry.close }.not_to raise_error
+    end
+  end
+
   describe "auto-start" do
     it "starts automatically when Sentry is initialized with enable_metrics" do
       expect(Sentry::Yabeda.collector).to be_a(described_class)
     end
 
     it "does not start when enable_metrics is false" do
       Sentry.close
-      Sentry::Yabeda.collector = nil
 
       Sentry.init do |config|
         config.dsn = DUMMY_DSN
@@ -40,7 +58,7 @@
       expect(Sentry::Yabeda.collector).to be_nil
     end
 
-    it "replaces an existing collector on re-initialization" do
+    it "replaces an existing collector on re-initialization via close" do
       first = Sentry::Yabeda.collector
 
       Sentry.close
@@ -49,5 +67,15 @@
       expect(Sentry::Yabeda.collector).to be_a(described_class)
       expect(Sentry::Yabeda.collector).not_to equal(first)
     end
+
+    it "replaces an existing collector on re-initialization without close" do
+      first = Sentry::Yabeda.collector
+      expect(first).to receive(:kill).and_call_original
+
+      perform_basic_setup
+
+      expect(Sentry::Yabeda.collector).to be_a(described_class)
+      expect(Sentry::Yabeda.collector).not_to equal(first)
+    end
   end
 end