From 62df5b9e2ebf28a6cdc1c81ec810849dfe696a75 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ma=C5=82gorzata=20Ksionek?= Date: Fri, 31 May 2019 11:29:11 +0200 Subject: Add query for code stage --- lib/gitlab/cycle_analytics/code_base_query.rb | 11 +++++++++++ lib/gitlab/cycle_analytics/code_event_fetcher.rb | 2 ++ lib/gitlab/cycle_analytics/code_stage.rb | 2 ++ 3 files changed, 15 insertions(+) create mode 100644 lib/gitlab/cycle_analytics/code_base_query.rb 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 -- cgit v1.2.1