debian-mirror-gitlab/app/models/project_services/prometheus_service.rb

144 lines
3.4 KiB
Ruby
Raw Normal View History

2018-11-20 20:47:30 +05:30
# frozen_string_literal: true
2017-08-17 22:00:37 +05:30
class PrometheusService < MonitoringService
2018-03-27 19:54:05 +05:30
include PrometheusAdapter
2017-08-17 22:00:37 +05:30
# Access to prometheus is directly through the API
prop_accessor :api_url
2018-03-17 18:26:18 +05:30
boolean_accessor :manual_configuration
2017-08-17 22:00:37 +05:30
2019-12-26 22:10:19 +05:30
# We need to allow the self-monitoring project to connect to the internal
# Prometheus instance.
# Since the internal Prometheus instance is usually a localhost URL, we need
# to allow localhost URLs when the following conditions are true:
# 1. project is the self-monitoring project.
# 2. api_url is the internal Prometheus URL.
2018-03-17 18:26:18 +05:30
with_options presence: true, if: :manual_configuration? do
2019-12-26 22:10:19 +05:30
validates :api_url, public_url: true, unless: proc { |object| object.allow_local_api_url? }
validates :api_url, url: true, if: proc { |object| object.allow_local_api_url? }
2017-08-17 22:00:37 +05:30
end
2018-03-27 19:54:05 +05:30
before_save :synchronize_service_state
2018-03-17 18:26:18 +05:30
2017-08-17 22:00:37 +05:30
after_save :clear_reactive_cache!
2020-01-01 13:55:28 +05:30
after_commit :track_events
2017-08-17 22:00:37 +05:30
def initialize_properties
if properties.nil?
self.properties = {}
end
end
2018-03-17 18:26:18 +05:30
def show_active_box?
false
end
def editable?
2018-12-13 13:39:08 +05:30
manual_configuration? || !prometheus_available?
2018-03-17 18:26:18 +05:30
end
2017-08-17 22:00:37 +05:30
def title
'Prometheus'
end
def description
2018-03-17 18:26:18 +05:30
s_('PrometheusService|Time-series monitoring service')
2017-08-17 22:00:37 +05:30
end
def self.to_param
'prometheus'
end
def fields
2018-03-17 18:26:18 +05:30
return [] unless editable?
2017-08-17 22:00:37 +05:30
[
2018-03-17 18:26:18 +05:30
{
type: 'checkbox',
name: 'manual_configuration',
title: s_('PrometheusService|Active'),
required: true
},
2017-08-17 22:00:37 +05:30
{
type: 'text',
name: 'api_url',
title: 'API URL',
2018-03-17 18:26:18 +05:30
placeholder: s_('PrometheusService|Prometheus API Base URL, like http://prometheus.example.com/'),
2017-09-10 17:25:29 +05:30
required: true
2017-08-17 22:00:37 +05:30
}
]
end
# Check we can connect to the Prometheus API
def test(*args)
2019-10-12 21:52:04 +05:30
prometheus_client.ping
2017-08-17 22:00:37 +05:30
{ success: true, result: 'Checked API endpoint' }
2018-03-27 19:54:05 +05:30
rescue Gitlab::PrometheusClient::Error => err
2017-08-17 22:00:37 +05:30
{ success: false, result: err }
end
2018-03-27 19:54:05 +05:30
def prometheus_client
2019-10-12 21:52:04 +05:30
return unless should_return_client?
Gitlab::PrometheusClient.new(api_url)
2018-03-17 18:26:18 +05:30
end
2018-12-13 13:39:08 +05:30
def prometheus_available?
2018-03-17 18:26:18 +05:30
return false if template?
return false unless project
2020-01-01 13:55:28 +05:30
project.all_clusters.enabled.any? { |cluster| cluster.application_prometheus_available? }
2017-09-10 17:25:29 +05:30
end
2019-12-26 22:10:19 +05:30
def allow_local_api_url?
self_monitoring_project? && internal_prometheus_url?
end
2020-01-01 13:55:28 +05:30
def configured?
should_return_client?
end
2017-09-10 17:25:29 +05:30
private
2019-12-26 22:10:19 +05:30
def self_monitoring_project?
2020-03-09 13:42:32 +05:30
project && project.id == current_settings.self_monitoring_project_id
2019-12-26 22:10:19 +05:30
end
def internal_prometheus_url?
api_url.present? && api_url == ::Gitlab::Prometheus::Internal.uri
end
2019-03-13 22:55:13 +05:30
def should_return_client?
2019-10-12 21:52:04 +05:30
api_url.present? && manual_configuration? && active? && valid?
2019-03-13 22:55:13 +05:30
end
2019-12-26 22:10:19 +05:30
def current_settings
Gitlab::CurrentSettings.current_application_settings
end
2018-03-27 19:54:05 +05:30
def synchronize_service_state
2018-12-13 13:39:08 +05:30
self.active = prometheus_available? || manual_configuration?
2018-03-17 18:26:18 +05:30
true
end
2020-01-01 13:55:28 +05:30
def track_events
if enabled_manual_prometheus?
Gitlab::Tracking.event('cluster:services:prometheus', 'enabled_manual_prometheus')
elsif disabled_manual_prometheus?
Gitlab::Tracking.event('cluster:services:prometheus', 'disabled_manual_prometheus')
end
true
end
def enabled_manual_prometheus?
manual_configuration_changed? && manual_configuration?
end
def disabled_manual_prometheus?
manual_configuration_changed? && !manual_configuration?
end
2017-08-17 22:00:37 +05:30
end