Class: Gitlab::UsageDataQueries

Inherits:
UsageData show all
Defined in:
lib/gitlab/usage_data_queries.rb

Overview

This class is used by the ‘gitlab:usage_data:dump_sql` rake tasks to output SQL instead of running it. See gitlab.com/gitlab-org/gitlab/-/merge_requests/41091

Constant Summary

Constants inherited from UsageData

Gitlab::UsageData::CE_MEMOIZED_VALUES, Gitlab::UsageData::MAX_GENERATION_TIME_FOR_SAAS

Constants included from Gitlab::Usage::TimeFrame

Gitlab::Usage::TimeFrame::ALL_TIME_TIME_FRAME_NAME, Gitlab::Usage::TimeFrame::DEFAULT_TIMESTAMP_COLUMN, Gitlab::Usage::TimeFrame::SEVEN_DAYS_TIME_FRAME_NAME, Gitlab::Usage::TimeFrame::TWENTY_EIGHT_DAYS_TIME_FRAME_NAME

Constants included from Gitlab::Utils::UsageData

Gitlab::Utils::UsageData::DISTRIBUTED_HLL_FALLBACK, Gitlab::Utils::UsageData::FALLBACK, Gitlab::Utils::UsageData::HISTOGRAM_FALLBACK, Gitlab::Utils::UsageData::MAX_BUCKET_SIZE

Class Method Summary collapse

Methods inherited from UsageData

components_usage_data, data, features_usage_data, object_store_config, object_store_usage_data, recorded_at, system_usage_data, system_usage_data_license, system_usage_data_weekly, usage_activity_by_stage, usage_activity_by_stage_configure, usage_activity_by_stage_create, usage_activity_by_stage_enablement, usage_activity_by_stage_manage, usage_activity_by_stage_monitor, usage_activity_by_stage_package, usage_activity_by_stage_plan, usage_activity_by_stage_release, usage_activity_by_stage_verify, user_preferences_usage

Methods included from Gitlab::Usage::TimeFrame

#monthly_time_range, #monthly_time_range_db_params, #weekly_time_range, #weekly_time_range_db_params

Methods included from Gitlab::Utils::UsageData

#add, #add_metric, #alt_usage_data, #average, #count, #distinct_count, #estimate_batch_distinct_count, #histogram, #maximum_id, #measure_duration, #metrics_collection_metadata, #minimum_id, #redis_usage_data, #sum, #track_usage_event, #with_finished_at, #with_metadata, #with_prometheus_client

Class Method Details

.add(*args) ⇒ Object 



41
42
43
 
# File 'lib/gitlab/usage_data_queries.rb', line 41

def add(*args)
  'SELECT ' + args.map { |arg| "(#{arg})" }.join(' + ')
end

.add_metric(metric, time_frame: 'none', options: {}) ⇒ Object 



12
13
14
15
16
 
# File 'lib/gitlab/usage_data_queries.rb', line 12

def add_metric(metric, time_frame: 'none', options: {})
  metric_class = "Gitlab::Usage::Metrics::Instrumentations::#{metric}".constantize

  metric_class.new(time_frame: time_frame, options: options).instrumentation
end

.alt_usage_data(value = nil, fallback: FALLBACK, &block) ⇒ Object 



53
54
55
56
57
58
59
 
# File 'lib/gitlab/usage_data_queries.rb', line 53

def alt_usage_data(value = nil, fallback: FALLBACK, &block)
  if block
    { alt_usage_data_block: "non-SQL usage data block" }
  else
    { alt_usage_data_value: value }
  end
end

.count(relation, column = nil, *args, **kwargs) ⇒ Object 



18
19
20
 
# File 'lib/gitlab/usage_data_queries.rb', line 18

def count(relation, column = nil, *args, **kwargs)
  Gitlab::Usage::Metrics::Query.for(:count, relation, column)
end

.distinct_count(relation, column = nil, *args, **kwargs) ⇒ Object 



22
23
24
 
# File 'lib/gitlab/usage_data_queries.rb', line 22

def distinct_count(relation, column = nil, *args, **kwargs)
  Gitlab::Usage::Metrics::Query.for(:distinct_count, relation, column)
end

.estimate_batch_distinct_count(relation, column = nil, *args, **kwargs) ⇒ Object

For estimated distinct count use exact query instead of hll buckets query, because it can’t be used to obtain estimations without supplementary ruby code present in Gitlab::Database::PostgresHll::BatchDistinctCounter



37
38
39
 
# File 'lib/gitlab/usage_data_queries.rb', line 37

def estimate_batch_distinct_count(relation, column = nil, *args, **kwargs)
  Gitlab::Usage::Metrics::Query.for(:estimate_batch_distinct_count, relation, column)
end

.histogram(relation, column, buckets:, bucket_size: buckets.size) ⇒ Object 



30
31
32
 
# File 'lib/gitlab/usage_data_queries.rb', line 30

def histogram(relation, column, buckets:, bucket_size: buckets.size)
  Gitlab::Usage::Metrics::Query.for(:histogram, relation, column, buckets: buckets, bucket_size: bucket_size)
end

.maximum_id(model, column = nil) ⇒ Object 



45
46
47
 
# File 'lib/gitlab/usage_data_queries.rb', line 45

def maximum_id(model, column = nil)
  # no-op: shadowing super for performance reasons
end

.minimum_id(model, column = nil) ⇒ Object 



49
50
51
 
# File 'lib/gitlab/usage_data_queries.rb', line 49

def minimum_id(model, column = nil)
  # no-op: shadowing super for performance reasons
end

.redis_usage_data(counter = nil, &block) ⇒ Object 



61
62
63
64
65
66
67
 
# File 'lib/gitlab/usage_data_queries.rb', line 61

def redis_usage_data(counter = nil, &block)
  if block
    { redis_usage_data_block: "non-SQL usage data block" }
  elsif counter.present?
    { redis_usage_data_counter: counter.to_s }
  end
end

.stage_manage_events(time_period) ⇒ Object 



76
77
78
79
80
 
# File 'lib/gitlab/usage_data_queries.rb', line 76

def stage_manage_events(time_period)
  # rubocop: disable CodeReuse/ActiveRecord
  estimate_batch_distinct_count(::Event.where(time_period), :author_id)
  # rubocop: enable CodeReuse/ActiveRecord
end

.sum(relation, column, *args, **kwargs) ⇒ Object 



26
27
28
 
# File 'lib/gitlab/usage_data_queries.rb', line 26

def sum(relation, column, *args, **kwargs)
  Gitlab::Usage::Metrics::Query.for(:sum, relation, column)
end

.topology_usage_data ⇒ Object 



69
70
71
72
73
74
 
# File 'lib/gitlab/usage_data_queries.rb', line 69

def topology_usage_data
  {
    duration_s: 0,
    failures: []
  }
end

.with_metadata ⇒ Object 



8
9
10
 
# File 'lib/gitlab/usage_data_queries.rb', line 8

def with_metadata
  yield
end