lib/peek/views/elasticsearch.rb


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
41
42
43
44
45
46
47

# frozen_string_literal: true

module Peek
  module Views
    class Elasticsearch < DetailedView
      DEFAULT_THRESHOLDS = {
        calls: 5,
        duration: 1000,
        individual_call: 1000
      }.freeze

      THRESHOLDS = {
        production: {
          calls: 5,
          duration: 1000,
          individual_call: 1000
        }
      }.freeze

      def key
        'es'
      end

      def self.thresholds
        @thresholds ||= THRESHOLDS.fetch(Rails.env.to_sym, DEFAULT_THRESHOLDS)
      end

      private

      def duration
        ::Gitlab::Instrumentation::ElasticsearchTransport.query_time * 1000
      end

      def calls
        ::Gitlab::Instrumentation::ElasticsearchTransport.get_request_count
      end

      def call_details
        ::Gitlab::Instrumentation::ElasticsearchTransport.detail_store
      end

      def format_call_details(call)
        super.merge(request: "#{call[:method]} #{call[:path]}")
      end
    end
  end
end