summaryrefslogtreecommitdiff
path: root/lib/gitlab/storage_check/response.rb
blob: 326ab236e3eaf76630c2e3450e0b63dd0eda5fa6 (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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
require 'json'

module Gitlab
  module StorageCheck
    class Response
      attr_reader :http_response

      def initialize(http_response)
        @http_response = http_response
      end

      def valid?
        @http_response && (200...299).cover?(@http_response.status) &&
          @http_response.headers['Content-Type'].include?('application/json') &&
          parsed_response
      end

      def check_interval
        return nil unless parsed_response

        parsed_response['check_interval']
      end

      def responsive_shards
        divided_results[:responsive_shards]
      end

      def skipped_shards
        divided_results[:skipped_shards]
      end

      def failing_shards
        divided_results[:failing_shards]
      end

      private

      def results
        return [] unless parsed_response

        parsed_response['results']
      end

      def divided_results
        return @divided_results if @divided_results

        @divided_results = {}
        @divided_results[:responsive_shards] = []
        @divided_results[:skipped_shards] = []
        @divided_results[:failing_shards] = []

        results.each do |info|
          name = info['storage']

          case info['success']
          when true
            @divided_results[:responsive_shards] << name
          when false
            @divided_results[:failing_shards] << name
          else
            @divided_results[:skipped_shards] << name
          end
        end

        @divided_results
      end

      def parsed_response
        return @parsed_response if defined?(@parsed_response)

        @parsed_response = JSON.parse(@http_response.body)
      rescue JSON::JSONError
        @parsed_response = nil
      end
    end
  end
end