blob: 54b3bbb3ce6a3df63317fe3fb87cddf6e67dd713 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
# frozen_string_literal: true
module Gitlab
module Analytics
module InstanceStatistics
class WorkersArgumentBuilder
def initialize(measurement_identifiers: [], recorded_at: Time.zone.now)
@measurement_identifiers = measurement_identifiers
@recorded_at = recorded_at
end
def execute
measurement_identifiers.map do |measurement_identifier|
query_scope = query_mappings[measurement_identifier]&.call
next if query_scope.nil?
[measurement_identifier, *determine_start_and_finish(measurement_identifier, query_scope), recorded_at]
end.compact
end
private
attr_reader :measurement_identifiers, :recorded_at
# Determining the query range (id range) as early as possible in order to get more accurate counts.
def determine_start_and_finish(measurement_identifier, query_scope)
queries = custom_min_max_queries[measurement_identifier]
if queries
[queries[:minimum_query].call, queries[:maximum_query].call]
else
[query_scope.minimum(:id), query_scope.maximum(:id)]
end
end
def custom_min_max_queries
::Analytics::InstanceStatistics::Measurement.identifier_min_max_queries
end
def query_mappings
::Analytics::InstanceStatistics::Measurement.identifier_query_mapping
end
end
end
end
end
|
