add open/read timeout kwargs

prometheus · Mar 23, 2021 · c053118 · c053118
1 parent 76e3cb8
commit c053118
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/prometheus/client/push.rb b/lib/prometheus/client/push.rb
@@ -20,7 +20,7 @@ class Push
 
       attr_reader :job, :instance, :gateway, :path
 
-      def initialize(job, instance = nil, gateway = nil)
+      def initialize(job, instance = nil, gateway = nil, **kwargs)
         @mutex = Mutex.new
         @job = job
         @instance = instance
@@ -30,6 +30,8 @@ def initialize(job, instance = nil, gateway = nil)
 
         @http = Net::HTTP.new(@uri.host, @uri.port)
         @http.use_ssl = (@uri.scheme == 'https')
+        @http.open_timeout = kwargs[:open_timeout] if kwargs[:open_timeout]
+        @http.read_timeout = kwargs[:read_timeout] if kwargs[:read_timeout]
       end
 
       def add(registry)