Add latest changes from gitlab-org/gitlab@13-4-stable-ee

1 files changed, 32 insertions, 0 deletions

diff --git a/lib/gitlab/analytics/instance_statistics/workers_argument_builder.rb b/lib/gitlab/analytics/instance_statistics/workers_argument_builder.rb
new file mode 100644
index 00000000000..636bba22c23
--- /dev/null
+++ b/lib/gitlab/analytics/instance_statistics/workers_argument_builder.rb
@@ -0,0 +1,32 @@
+# 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 = ::Analytics::InstanceStatistics::Measurement::IDENTIFIER_QUERY_MAPPING[measurement_identifier]&.call
+
+            next if query_scope.nil?
+
+            # Determining the query range (id range) as early as possible in order to get more accurate counts.
+            start = query_scope.minimum(:id)
+            finish = query_scope.maximum(:id)
+
+            [measurement_identifier, start, finish, recorded_at]
+          end.compact
+        end
+
+        private
+
+        attr_reader :measurement_identifiers, :recorded_at
+      end
+    end
+  end
+end