EXPERIMENTAL

Add metrics for api_response_duration_seconds histograms.

config.ru

@@ -13,6 +13,8 @@ end
 Bundler.require
 Spectrum::Json.configure(__dir__, ENV["RAILS_RELATIVE_URL_ROOT"])
 
+require 'alma_rest_client/monkey_patch'
+
 use Rack::ReverseProxy do
   reverse_proxy %r{^/browse.css}, "https://#{ENV["BROWSE_HOST"]}/$1"
   reverse_proxy %r{^/catalog/browse/(.*)$}, "https://#{ENV["BROWSE_HOST"]}/$1"

config/puma.rb

@@ -65,6 +65,12 @@
       labels: %i[method path],
       buckets: Prometheus::Client::Histogram::DEFAULT_BUCKETS + [15, 20, 30]
     )
+    Prometheus::Client.registry.histogram(
+      :api_response_duration_seconds,
+      docstring: "The API response duration for requests made by Spectrum.",
+      labels: %i[source],
+      buckets: Prometheus::Client::Histogram::DEFAULT_BUCKETS + [15, 20, 30]
+    )
     Yabeda.configure!
 
     plugin :yabeda

lib/alma_rest_client/monkey_patch.rb

@@ -0,0 +1,16 @@
+module AlmaRestClient
+  class Client
+    [:get, :post, :delete, :put].each do |name|
+      old_method = instance_method(name)
+      define_method(name) do |url, options = {}|
+        @metrics ||= Prometheus::Client.registry.get(:api_response_duration_seconds)
+        response = nil
+        duration = Benchmark.realtime do
+          response = old_method.bind(self).(url, options)
+        end
+        @metrics.observe(duration, labels: {source: "alma"})
+        response
+      end
+    end
+  end
+end

local-gems/spectrum-config/lib/spectrum/search_engines/primo/engine.rb

@@ -24,6 +24,7 @@ def initialize(
           if defined?(Rails) && Rails.respond_to?(:logger)
             @logger = Rails.logger
           end
+          @metrics = Prometheus::Client.registry.get(:api_response_duration_seconds)
         end
 
         def results
@@ -36,8 +37,21 @@ def total_items
 
 
         def search
+          return @results if @results
           @logger&.info { url }
-          @results ||= Response.for_json(HTTParty.get(url)).with_libkey(libkey)
+
+          response = nil
+          duration = Benchmark.realtime do
+            response = Response.for_json(HTTParty.get(url))
+          end
+          @metrics.observe(duration, labels: {source: 'primo'})
+
+          duration = Benchmark.realtime do
+            @results = response.with_libkey(libkey)
+          end
+          @metrics.observe(duration, labels: {source: 'libkey'})
+
+          @results
         end
 
         def url

local-gems/spectrum-config/lib/spectrum/search_engines/solr.rb

@@ -32,7 +32,11 @@ def initialize(original_options)
 
         @params[:qq] ||= '"' + RSolr.solr_escape(@params[:q]) + '"'
 
-        @response = Response.for(@solr.post("select", params: @params))
+        @metrics = Prometheus::Client.registry.get(:api_response_duration_seconds)
+        duration = Benchmark.realtime do
+          @response = Response.for(@solr.post("select", params: @params))
+        end
+        @metrics.observe(duration, labels: {source: @source.id})
       end
 
       def total_items