64 lines
2.6 KiB
Ruby
64 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab::UsageDataCounters
|
|
class CiTemplateUniqueCounter
|
|
REDIS_SLOT = 'ci_templates'
|
|
KNOWN_EVENTS_FILE_PATH = File.expand_path('known_events/ci_templates.yml', __dir__)
|
|
|
|
class << self
|
|
def track_unique_project_event(project:, template:, config_source:, user:)
|
|
expanded_template_name = expand_template_name(template)
|
|
return unless expanded_template_name
|
|
|
|
event_name = ci_template_event_name(expanded_template_name, config_source)
|
|
Gitlab::UsageDataCounters::HLLRedisCounter.track_event(event_name, values: project.id)
|
|
|
|
namespace = project.namespace
|
|
if Feature.enabled?(:route_hll_to_snowplow, namespace)
|
|
context = Gitlab::Tracking::ServicePingContext.new(data_source: :redis_hll,
|
|
event: event_name).to_context
|
|
label = 'redis_hll_counters.ci_templates.ci_templates_total_unique_counts_monthly'
|
|
Gitlab::Tracking.event(name, 'ci_templates_unique', namespace: namespace,
|
|
project: project, context: [context], user: user,
|
|
label: label)
|
|
end
|
|
end
|
|
|
|
def ci_templates(relative_base = 'lib/gitlab/ci/templates')
|
|
Dir.glob('**/*.gitlab-ci.yml', base: Rails.root.join(relative_base))
|
|
end
|
|
|
|
def ci_template_event_name(template_name, config_source)
|
|
prefix = 'implicit_' if config_source.to_s == 'auto_devops_source'
|
|
|
|
"p_#{REDIS_SLOT}_#{prefix}#{template_to_event_name(template_name)}"
|
|
end
|
|
|
|
def expand_template_name(template_name)
|
|
Gitlab::Template::GitlabCiYmlTemplate.find(template_name.chomp('.gitlab-ci.yml'))&.full_name
|
|
end
|
|
|
|
def all_included_templates(template_name)
|
|
expanded_template_name = expand_template_name(template_name)
|
|
results = [expanded_template_name].tap do |result|
|
|
template = Gitlab::Template::GitlabCiYmlTemplate.find(template_name.chomp('.gitlab-ci.yml'))
|
|
data = YAML.safe_load(template.content, aliases: true)
|
|
[data['include']].compact.flatten.each do |ci_include|
|
|
if ci_include_template = ci_include['template']
|
|
result.concat(all_included_templates(ci_include_template))
|
|
end
|
|
end
|
|
end
|
|
|
|
results.uniq.sort_by { _1['name'] }
|
|
end
|
|
|
|
private
|
|
|
|
def template_to_event_name(template)
|
|
ActiveSupport::Inflector.parameterize(template.chomp('.gitlab-ci.yml'), separator: '_').underscore
|
|
end
|
|
end
|
|
end
|
|
end
|