refactored a bunch of stuff based on feedback

1 files changed, 6 insertions, 5 deletions

diff --git a/lib/gitlab/cycle_analytics/issue_stage.rb b/lib/gitlab/cycle_analytics/issue_stage.rb
index d320458d7fd..d2068fbc38f 100644
--- a/lib/gitlab/cycle_analytics/issue_stage.rb
+++ b/lib/gitlab/cycle_analytics/issue_stage.rb
@@ -1,12 +1,13 @@
 module Gitlab
   module CycleAnalytics
     class IssueStage < BaseStage
-      def initialize(*args)
-        @start_time_attrs = issue_table[:created_at]
-        @end_time_attrs = [issue_metrics_table[:first_associated_with_milestone_at],
-                           issue_metrics_table[:first_added_to_board_at]]
+      def start_time_attrs
+        @start_time_attrs ||= issue_table[:created_at]
+      end
 
-        super(*args)
+      def end_time_attrs
+        @end_time_attrs ||= [issue_metrics_table[:first_associated_with_milestone_at],
+                             issue_metrics_table[:first_added_to_board_at]]
       end
 
       def name