73 lines
2.2 KiB
Ruby
73 lines
2.2 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module Usage
|
||
|
|
module Metrics
|
||
|
|
class Query
|
||
|
|
class << self
|
||
|
|
def for(operation, relation, column = nil, **extra)
|
||
|
|
case operation
|
||
|
|
when :count
|
||
|
|
count(relation, column)
|
||
|
|
when :distinct_count
|
||
|
|
distinct_count(relation, column)
|
||
|
|
when :sum
|
||
|
|
sum(relation, column)
|
||
|
|
when :estimate_batch_distinct_count
|
||
|
|
estimate_batch_distinct_count(relation, column)
|
||
|
|
when :histogram
|
||
|
|
histogram(relation, column, **extra)
|
||
|
|
else
|
||
|
|
raise ArgumentError, "#{operation} operation not supported"
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def count(relation, column = nil)
|
||
|
|
raw_sql(relation, column)
|
||
|
|
end
|
||
|
|
|
||
|
|
def distinct_count(relation, column = nil)
|
||
|
|
raw_sql(relation, column, true)
|
||
|
|
end
|
||
|
|
|
||
|
|
def sum(relation, column)
|
||
|
|
relation.select(relation.all.table[column].sum).to_sql
|
||
|
|
end
|
||
|
|
|
||
|
|
def estimate_batch_distinct_count(relation, column = nil)
|
||
|
|
raw_sql(relation, column, true)
|
||
|
|
end
|
||
|
|
|
||
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
||
|
|
def histogram(relation, column, buckets:, bucket_size: buckets.size)
|
||
|
|
count_grouped = relation.group(column).select(Arel.star.count.as('count_grouped'))
|
||
|
|
cte = Gitlab::SQL::CTE.new(:count_cte, count_grouped)
|
||
|
|
|
||
|
|
bucket_segments = bucket_size - 1
|
||
|
|
width_bucket = Arel::Nodes::NamedFunction
|
||
|
|
.new('WIDTH_BUCKET', [cte.table[:count_grouped], buckets.first, buckets.last, bucket_segments])
|
||
|
|
.as('buckets')
|
||
|
|
|
||
|
|
query = cte
|
||
|
|
.table
|
||
|
|
.project(width_bucket, cte.table[:count])
|
||
|
|
.group('buckets')
|
||
|
|
.order('buckets')
|
||
|
|
.with(cte.to_arel)
|
||
|
|
|
||
|
|
query.to_sql
|
||
|
|
end
|
||
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
||
|
|
|
||
|
|
def raw_sql(relation, column, distinct = false)
|
||
|
|
column ||= relation.primary_key
|
||
|
|
relation.select(relation.all.table[column].count(distinct)).to_sql
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|