summaryrefslogtreecommitdiff
path: root/lib/gitlab/metrics/elasticsearch_rack_middleware.rb
blob: 870ab1480046e87451f30575c82f138f093914d5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# frozen_string_literal: true

module Gitlab
  module Metrics
    # Rack middleware for tracking Elasticsearch metrics from Grape and Web requests.
    class ElasticsearchRackMiddleware
      HISTOGRAM_BUCKETS = [0.1, 0.5, 1, 10, 50].freeze

      def initialize(app)
        @app = app
      end

      def call(env)
        transaction = Gitlab::Metrics.current_transaction

        @app.call(env)
      ensure
        record_metrics(transaction)
      end

      private

      def record_metrics(transaction)
        query_time = ::Gitlab::Instrumentation::ElasticsearchTransport.query_time
        request_count = ::Gitlab::Instrumentation::ElasticsearchTransport.get_request_count

        return unless request_count > 0

        transaction.increment(:http_elasticsearch_requests_total, request_count) do
          docstring 'Amount of calls to Elasticsearch servers during web requests'
        end

        transaction.observe(:http_elasticsearch_requests_duration_seconds, query_time) do
          docstring 'Query time for Elasticsearch servers during web requests'
          buckets HISTOGRAM_BUCKETS
        end
      end
    end
  end
end