gitlab-org--gitlab-foss/lib/gitlab/prometheus_client.rb

88 lines
2.4 KiB
Ruby
Raw Normal View History

module Gitlab
PrometheusError = Class.new(StandardError)
# Helper methods to interact with Prometheus network services & resources
class PrometheusClient
attr_reader :rest_client, :headers
def initialize(rest_client)
@rest_client = rest_client
end
def ping
json_api_get('query', query: '1')
end
2017-05-09 00:15:34 -04:00
def query(query, time: Time.now)
get_result('vector') do
json_api_get('query', query: query, time: time.to_f)
end
end
2017-05-09 00:15:34 -04:00
def query_range(query, start: 8.hours.ago, stop: Time.now)
get_result('matrix') do
json_api_get('query_range',
query: query,
start: start.to_f,
2017-05-09 00:15:34 -04:00
end: stop.to_f,
step: 1.minute.to_i)
end
end
2017-06-05 15:00:57 -04:00
def label_values(name = '__name__')
json_api_get("label/#{name}/values")
end
def series(*matches, start: 8.hours.ago, stop: Time.now)
json_api_get('series', 'match': matches, start: start.to_f, end: stop.to_f)
end
private
def json_api_get(type, args = {})
2018-01-02 14:24:12 -05:00
path = ['api', 'v1', type].join('/')
get(path, args)
rescue JSON::ParserError
raise PrometheusError, 'Parsing response failed'
rescue Errno::ECONNREFUSED
raise PrometheusError, 'Connection refused'
end
2018-01-02 14:24:12 -05:00
def get(path, args)
response = rest_client[path].get(params: args)
2018-01-02 14:24:12 -05:00
handle_response(response)
2017-05-02 12:16:59 -04:00
rescue SocketError
2018-01-03 21:13:54 -05:00
raise PrometheusError, "Can't connect to #{rest_client.url}"
2017-05-02 12:16:59 -04:00
rescue OpenSSL::SSL::SSLError
2018-01-03 21:13:54 -05:00
raise PrometheusError, "#{rest_client.url} contains invalid SSL data"
rescue RestClient::ExceptionWithResponse => ex
handle_exception_response(ex.response)
rescue RestClient::Exception
raise PrometheusError, "Network connection error"
end
def handle_response(response)
json_data = JSON.parse(response.body)
2018-01-02 14:24:12 -05:00
if response.code == 200 && json_data['status'] == 'success'
json_data['data'] || {}
2018-01-03 21:13:54 -05:00
else
raise PrometheusError, "#{response.code} - #{response.body}"
end
end
def handle_exception_response(response)
if response.code == 400
json_data = JSON.parse(response.body)
2018-01-02 14:24:12 -05:00
raise PrometheusError, json_data['error'] || 'Bad data received'
else
raise PrometheusError, "#{response.code} - #{response.body}"
end
end
def get_result(expected_type)
data = yield
data['result'] if data['resultType'] == expected_type
end
end
end