91 lines
3.3 KiB
Ruby
91 lines
3.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Tracking
|
|
class << self
|
|
def enabled?
|
|
tracker.enabled?
|
|
end
|
|
|
|
def event(category, action, label: nil, property: nil, value: nil, context: [], project: nil, user: nil, namespace: nil, **extra) # rubocop:disable Metrics/ParameterLists
|
|
action = action.to_s
|
|
|
|
project_id = project.is_a?(Integer) ? project : project&.id
|
|
|
|
contexts = [
|
|
Tracking::StandardContext.new(
|
|
namespace_id: namespace&.id,
|
|
plan_name: namespace&.actual_plan_name,
|
|
project_id: project_id,
|
|
user_id: user&.id,
|
|
**extra).to_context, *context
|
|
]
|
|
|
|
track_struct_event(tracker, category, action, label: label, property: property, value: value, contexts: contexts)
|
|
end
|
|
|
|
def database_event(category, action, label: nil, property: nil, value: nil, context: [], project: nil, user: nil, namespace: nil, **extra) # rubocop:disable Metrics/ParameterLists
|
|
action = action.to_s
|
|
destination = Gitlab::Tracking::Destinations::DatabaseEventsSnowplow.new
|
|
contexts = [
|
|
Tracking::StandardContext.new(
|
|
namespace_id: namespace&.id,
|
|
plan_name: namespace&.actual_plan_name,
|
|
project_id: project&.id,
|
|
user_id: user&.id,
|
|
**extra).to_context, *context
|
|
]
|
|
|
|
track_struct_event(destination, category, action, label: label, property: property, value: value, contexts: contexts)
|
|
end
|
|
|
|
def definition(basename, category: nil, action: nil, label: nil, property: nil, value: nil, context: [], project: nil, user: nil, namespace: nil, **extra) # rubocop:disable Metrics/ParameterLists
|
|
definition = YAML.load_file(Rails.root.join("config/events/#{basename}.yml"))
|
|
|
|
dispatch_from_definition(definition, label: label, property: property, value: value, context: context, project: project, user: user, namespace: namespace, **extra)
|
|
end
|
|
|
|
def dispatch_from_definition(definition, **event_data)
|
|
definition = definition.with_indifferent_access
|
|
|
|
category ||= definition[:category]
|
|
action ||= definition[:action]
|
|
|
|
event(category, action, **event_data)
|
|
end
|
|
|
|
def options(group)
|
|
tracker.options(group)
|
|
end
|
|
|
|
def collector_hostname
|
|
tracker.hostname
|
|
end
|
|
|
|
def snowplow_micro_enabled?
|
|
Rails.env.development? && Gitlab.config.snowplow_micro.enabled
|
|
rescue GitlabSettings::MissingSetting
|
|
false
|
|
end
|
|
|
|
private
|
|
|
|
def track_struct_event(destination, category, action, label:, property:, value:, contexts:) # rubocop:disable Metrics/ParameterLists
|
|
destination
|
|
.event(category, action, label: label, property: property, value: value, context: contexts)
|
|
rescue StandardError => error
|
|
Gitlab::ErrorTracking.track_and_raise_for_dev_exception(error, snowplow_category: category, snowplow_action: action)
|
|
end
|
|
|
|
def tracker
|
|
@tracker ||= if snowplow_micro_enabled?
|
|
Gitlab::Tracking::Destinations::SnowplowMicro.new
|
|
else
|
|
Gitlab::Tracking::Destinations::Snowplow.new
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
Gitlab::Tracking.prepend_mod_with('Gitlab::Tracking')
|