summaryrefslogtreecommitdiff
path: root/app/services/projects/housekeeping_service.rb
blob: 29b3981f49f76f1f3b824f56bd820869cec8c65b (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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# Projects::HousekeepingService class
#
# Used for git housekeeping
#
# Ex.
#   Projects::HousekeepingService.new(project).execute
#
module Projects
  class HousekeepingService < BaseService
    LEASE_TIMEOUT = 3600

    class LeaseTaken < StandardError
      def to_s
        "Somebody already triggered housekeeping for this project in the past #{LEASE_TIMEOUT / 60} minutes"
      end
    end

    def initialize(project)
      @project = project
    end

    def execute
      raise LeaseTaken unless try_obtain_lease

      execute_gitlab_shell_gc
    end

    def needed?
      @project.pushes_since_gc >= 10
    end

    def increment!
      if Gitlab::ExclusiveLease.new("project_housekeeping:increment!:#{@project.id}", timeout: 60).try_obtain
        Gitlab::Metrics.measure(:increment_pushes_since_gc) do
          update_pushes_since_gc(@project.pushes_since_gc + 1)
        end
      end
    end

    private

    def execute_gitlab_shell_gc
      GitGarbageCollectWorker.perform_async(@project.id)
    ensure
      Gitlab::Metrics.measure(:reset_pushes_since_gc) do
        update_pushes_since_gc(0)
      end
    end

    def update_pushes_since_gc(new_value)
      @project.update_column(:pushes_since_gc, new_value)
    end

    def try_obtain_lease
      Gitlab::Metrics.measure(:obtain_housekeeping_lease) do
        lease = ::Gitlab::ExclusiveLease.new("project_housekeeping:#{@project.id}", timeout: LEASE_TIMEOUT)
        lease.try_obtain
      end
    end
  end
end