app/graphql/resolvers/analytics/cycle_analytics/base_count_resolver.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

# frozen_string_literal: true

module Resolvers
  module Analytics
    module CycleAnalytics
      class BaseCountResolver < BaseResolver
        type Types::Analytics::CycleAnalytics::MetricType, null: true

        argument :from, Types::TimeType,
          required: true,
          description: 'Timestamp marking the start date and time.'

        argument :to, Types::TimeType,
          required: true,
          description: 'Timestamp marking the end date and time.'

        def ready?(**args)
          start_date = args[:from]
          end_date = args[:to]

          if start_date >= end_date
            raise Gitlab::Graphql::Errors::ArgumentError,
              '`from` argument must be before `to` argument'
          end

          max_days = Gitlab::Analytics::CycleAnalytics::RequestParams::MAX_RANGE_DAYS

          if (end_date.beginning_of_day - start_date.beginning_of_day) > max_days
            raise Gitlab::Graphql::Errors::ArgumentError,
              "Max of #{max_days.inspect} timespan is allowed"
          end

          super
        end
      end
    end
  end
end