Add listener_address as Instance attributes of Promenade::Raindrops::…

…Stats
errm · Aug 15, 2024 · 1581a4b · 1581a4b
1 parent 59195bf
commit 1581a4b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/lib/promenade/raindrops/stats.rb b/lib/promenade/raindrops/stats.rb
@@ -16,14 +16,15 @@ module Raindrops
 
     class Stats
 
-      attr_reader :active_workers, :queued_requests
+      attr_reader :active_workers, :queued_requests, :listener_address
 
-      def initialize
+      def initialize(listener_address: nil)
         return unless defined?(::Raindrops)
         return unless defined?(::Raindrops::Linux.tcp_listener_stats)
 
-        listener_address = "0.0.0.0:#{ENV.fetch('PORT', 3000)}"
-        stats = ::Raindrops::Linux.tcp_listener_stats([listener_address])[listener_address]
+        @listener_address = listener_address || "127.0.0.1:#{ENV.fetch('PORT', 3000)}"
+
+        stats = ::Raindrops::Linux.tcp_listener_stats([@listener_address])[@listener_address]
 
         @active_workers = stats.active
         @queued_requests = stats.queued
@@ -34,8 +35,8 @@ def instrument
         Promenade.metric(:rack_queued_requests).set({}, queued_requests)
       end
 
-      def self.instrument
-        new.instrument
+      def self.instrument(listener_address: nil)
+        new(listener_address: listener_address).instrument
       end
     end
   end

diff --git a/spec/promenade/raindrops/stats_spec.rb b/spec/promenade/raindrops/stats_spec.rb
@@ -3,7 +3,7 @@
 
 RSpec.describe Promenade::Raindrops::Stats do
   let(:listen_stats) { instance_double("Raindrops::Linux::ListenStats", active: 1, queued: 1) }
-  let(:listener_address) { "0.0.0.0:#{ENV.fetch('PORT', 3000)}" }
+  let(:listener_address) { "127.0.0.1:#{ENV.fetch('PORT', 3000)}" }
 
   before do
     allow(Raindrops::Linux).to receive(:tcp_listener_stats).and_return({ listener_address => listen_stats })