summaryrefslogtreecommitdiff
path: root/lib/gitlab/import_export/project/sample/date_calculator.rb
blob: 543fd25d883eaab69efe816f6e366db4c2fe259c (plain)
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
# frozen_string_literal: true

module Gitlab
  module ImportExport
    module Project
      module Sample
        class DateCalculator
          include Gitlab::Utils::StrongMemoize

          def initialize(dates)
            @dates = dates.dup
            @dates.compact!
            @dates.sort!
            @dates.map! { |date| date.to_time.to_f }
          end

          def closest_date_to_average
            strong_memoize(:closest_date_to_average) do
              next if @dates.empty?

              average_date = (@dates.first + @dates.last) / 2.0
              closest_date = @dates.min_by { |date| (date - average_date).abs }
              Time.zone.at(closest_date)
            end
          end

          def calculate_by_closest_date_to_average(date)
            return date unless closest_date_to_average && closest_date_to_average < Time.current

            date + (Time.current - closest_date_to_average).seconds
          end
        end
      end
    end
  end
end