Welcome to mirror list, hosted at ThFree Co, Russian Federation.

batch_count_service.rb « projects « services « app - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: b8a030e7bc5f5c4444626dbcb13b8c7fa28c0438 (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
37
38
39
module Projects
  class BatchCountService
    def initialize(projects)
      @projects = projects
    end

    def count
      @projects.map do |project|
        [project.id, current_count_service(project).count]
      end.to_h
    end

    def refresh_cache
      @projects.each do |project|
        unless current_count_service(project).count_stored?
          current_count_service(project).refresh_cache { global_count[project.id].to_i }
        end
      end
    end

    def current_count_service(project)
      if defined? @service
        @service.project = project
      else
        @service = count_service.new(project)
      end

      @service
    end

    def global_count(project)
      raise NotImplementedError, 'global_count must be implemented and return an hash indexed by the project id'
    end

    def count_service
      raise NotImplementedError, 'count_service must be implemented and return a Projects::CountService object'
    end
  end
end