add push metrics to push gateway

Author: niquerio

lib/process_ldap.rb

@@ -100,6 +100,7 @@ def script_type
   end
 
   def process
+    start = Time.now
     S.logger.info("start")
     S.logger.info("open files", file_base: @file_base) if @file_base
     Report.open(file_base: @file_base, script_type: script_type) do |report|
@@ -127,8 +128,9 @@ def process
       S.logger.error "ldap_error", code: ldap_result_code, message: ldap.get_operation_result.message
       Report.metrics.error.increment
     end
-
+    Report.metrics.job_duration_seconds.set({script_type: script_type}, Time.now - start)
     puts Report.print_metrics
+    Report.push_metrics
     S.logger.info("end")
   end
 

lib/report.rb

@@ -1,4 +1,5 @@
 require "yabeda/prometheus"
+require "uri"
 class Report
   def self.configure_yabeda!
     Yabeda.configure do
@@ -8,6 +9,7 @@ def self.configure_yabeda!
         gauge :found, comment: "Number of found patrons", tags: [:script_type]
         gauge :statistic_category, comment: "Number of loaded patrons in a statistic category", tags: [:script_type, :name]
         gauge :error, comment: "Number of errors encountered while running the patron load", tags: [:script_type]
+        gauge :job_duration_seconds, comment: "Number of seconds it took to run the patron load job", tags: [:script_type]
       end
     end
     Yabeda.configure!
@@ -21,6 +23,16 @@ def self.print_metrics
     Prometheus::Client::Formats::Text.marshal(Yabeda::Prometheus.registry)
   end
 
+  def self.push_metrics
+    # The env var needs to be set to the push gateway url
+    if ENV["PROMETHEUS_PUSH_GATEWAY"]&.match?(URI::DEFAULT_PARSER.make_regexp)
+      Yabeda::Prometheus.push_gateway.add(Yabeda::Prometheus.registry)
+      S.logger.info("Metrics sent to the push gateway")
+    else
+      S.logger.warn("PROMETHEUS_PUSH_GATEWAY env var not set. Metrics not sent to the push gateway")
+    end
+  end
+
   def self.open(file_base:, script_type:, &block)
     if file_base
       File.open("#{file_base}.tsv", "w") do |fh|