gitlab-org--gitlab-foss/lib/gitlab/cycle_analytics/base_stage.rb

module Gitlab
  module CycleAnalytics
    class BaseStage
      attr_reader :stage, :description

      def initialize(project:, options:, stage:)
        @project = project
        @options = options
        @fetcher = Gitlab::CycleAnalytics::MetricsFetcher.new(project: project,
                                                              from: options[:from],
                                                              branch: options[:branch])
        @stage = stage
      end

      def events
        event_class.new(fetcher: @fetcher, stage: @stage, options: @options).fetch
      end

      def median_data
        AnalyticsStageSerializer.new.represent(self).as_json
      end

      private

      def event_class
        "Gitlab::CycleAnalytics::#{@stage.to_s.capitalize}Event".constantize
      end
    end
  end
end