Add query for code stage

3 files changed, 15 insertions, 0 deletions

diff --git a/lib/gitlab/cycle_analytics/code_base_query.rb b/lib/gitlab/cycle_analytics/code_base_query.rb
new file mode 100644
index 00000000000..ef96419c021
--- /dev/null
+++ b/lib/gitlab/cycle_analytics/code_base_query.rb
@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module CycleAnalytics
+    module CodeBaseQuery
+      def stage_query(project_ids)
+        super(project_ids).where(issue_metrics_table[:first_mentioned_in_commit_at].not_eq(nil))
+      end
+    end
+  end
+end
diff --git a/lib/gitlab/cycle_analytics/code_event_fetcher.rb b/lib/gitlab/cycle_analytics/code_event_fetcher.rb
index 591db3c35e6..3292614e4a4 100644
--- a/lib/gitlab/cycle_analytics/code_event_fetcher.rb
+++ b/lib/gitlab/cycle_analytics/code_event_fetcher.rb
@@ -3,6 +3,8 @@
 module Gitlab
   module CycleAnalytics
     class CodeEventFetcher < BaseEventFetcher
+      include CodeBaseQuery
+
       def initialize(*args)
         @projections = [mr_table[:title],
                         mr_table[:iid],
diff --git a/lib/gitlab/cycle_analytics/code_stage.rb b/lib/gitlab/cycle_analytics/code_stage.rb
index 2e5f9ef5a40..b6bd08255a9 100644
--- a/lib/gitlab/cycle_analytics/code_stage.rb
+++ b/lib/gitlab/cycle_analytics/code_stage.rb
@@ -3,6 +3,8 @@
 module Gitlab
   module CycleAnalytics
     class CodeStage < BaseStage
+      include CodeBaseQuery
+
       def start_time_attrs
         @start_time_attrs ||= issue_metrics_table[:first_mentioned_in_commit_at]
       end