debian-mirror-gitlab/lib/sentry/client.rb

175 lines
4.3 KiB
Ruby
Raw Normal View History

2019-02-15 15:39:39 +05:30
# frozen_string_literal: true
module Sentry
class Client
Error = Class.new(StandardError)
2019-07-07 11:18:12 +05:30
MissingKeysError = Class.new(StandardError)
2019-02-15 15:39:39 +05:30
attr_accessor :url, :token
def initialize(api_url, token)
@url = api_url
@token = token
end
def list_issues(issue_status:, limit:)
issues = get_issues(issue_status: issue_status, limit: limit)
2019-07-07 11:18:12 +05:30
handle_mapping_exceptions do
map_to_errors(issues)
end
2019-02-15 15:39:39 +05:30
end
2019-03-02 22:35:43 +05:30
def list_projects
projects = get_projects
2019-07-07 11:18:12 +05:30
handle_mapping_exceptions do
map_to_projects(projects)
end
2019-03-02 22:35:43 +05:30
end
2019-02-15 15:39:39 +05:30
private
2019-07-07 11:18:12 +05:30
def handle_mapping_exceptions(&block)
yield
rescue KeyError => e
Gitlab::Sentry.track_acceptable_exception(e)
raise Client::MissingKeysError, "Sentry API response is missing keys. #{e.message}"
end
2019-02-15 15:39:39 +05:30
def request_params
{
headers: {
'Authorization' => "Bearer #{@token}"
},
follow_redirects: false
}
end
2019-03-02 22:35:43 +05:30
def http_get(url, params = {})
2019-07-07 11:18:12 +05:30
response = handle_request_exceptions do
Gitlab::HTTP.get(url, **request_params.merge(params))
end
2019-02-15 15:39:39 +05:30
2019-07-07 11:18:12 +05:30
handle_response(response)
2019-02-15 15:39:39 +05:30
end
2019-03-02 22:35:43 +05:30
def get_issues(issue_status:, limit:)
http_get(issues_api_url, query: {
query: "is:#{issue_status}",
limit: limit
})
end
def get_projects
http_get(projects_api_url)
end
2019-07-07 11:18:12 +05:30
def handle_request_exceptions
yield
2019-10-12 21:52:04 +05:30
rescue Gitlab::HTTP::Error => e
2019-07-07 11:18:12 +05:30
Gitlab::Sentry.track_acceptable_exception(e)
raise_error 'Error when connecting to Sentry'
rescue Net::OpenTimeout
raise_error 'Connection to Sentry timed out'
rescue SocketError
raise_error 'Received SocketError when trying to connect to Sentry'
rescue OpenSSL::SSL::SSLError
raise_error 'Sentry returned invalid SSL data'
rescue Errno::ECONNREFUSED
raise_error 'Connection refused'
rescue => e
Gitlab::Sentry.track_acceptable_exception(e)
raise_error "Sentry request failed due to #{e.class}"
end
2019-02-15 15:39:39 +05:30
def handle_response(response)
unless response.code == 200
2019-07-07 11:18:12 +05:30
raise_error "Sentry response status code: #{response.code}"
2019-02-15 15:39:39 +05:30
end
2019-07-07 11:18:12 +05:30
response
end
def raise_error(message)
raise Client::Error, message
2019-02-15 15:39:39 +05:30
end
2019-03-02 22:35:43 +05:30
def projects_api_url
projects_url = URI(@url)
projects_url.path = '/api/0/projects/'
projects_url
end
2019-02-15 15:39:39 +05:30
def issues_api_url
issues_url = URI(@url + '/issues/')
issues_url.path.squeeze!('/')
issues_url
end
def map_to_errors(issues)
2019-03-02 22:35:43 +05:30
issues.map(&method(:map_to_error))
end
def map_to_projects(projects)
projects.map(&method(:map_to_project))
2019-02-15 15:39:39 +05:30
end
def issue_url(id)
issues_url = @url + "/issues/#{id}"
issues_url = ErrorTracking::ProjectErrorTrackingSetting.extract_sentry_external_url(issues_url)
uri = URI(issues_url)
uri.path.squeeze!('/')
uri.to_s
end
def map_to_error(issue)
id = issue.fetch('id')
count = issue.fetch('count', nil)
frequency = issue.dig('stats', '24h')
message = issue.dig('metadata', 'value')
external_url = issue_url(id)
Gitlab::ErrorTracking::Error.new(
id: id,
first_seen: issue.fetch('firstSeen', nil),
last_seen: issue.fetch('lastSeen', nil),
title: issue.fetch('title', nil),
type: issue.fetch('type', nil),
user_count: issue.fetch('userCount', nil),
count: count,
message: message,
culprit: issue.fetch('culprit', nil),
external_url: external_url,
short_id: issue.fetch('shortId', nil),
status: issue.fetch('status', nil),
frequency: frequency,
2019-07-07 11:18:12 +05:30
project_id: issue.dig('project', 'id'),
project_name: issue.dig('project', 'name'),
project_slug: issue.dig('project', 'slug')
2019-02-15 15:39:39 +05:30
)
end
2019-03-02 22:35:43 +05:30
def map_to_project(project)
organization = project.fetch('organization')
Gitlab::ErrorTracking::Project.new(
2019-07-07 11:18:12 +05:30
id: project.fetch('id', nil),
2019-03-02 22:35:43 +05:30
name: project.fetch('name'),
slug: project.fetch('slug'),
status: project.dig('status'),
organization_name: organization.fetch('name'),
2019-07-07 11:18:12 +05:30
organization_id: organization.fetch('id', nil),
2019-03-02 22:35:43 +05:30
organization_slug: organization.fetch('slug')
)
end
2019-02-15 15:39:39 +05:30
end
end