app/services/delete_merged_branches_service.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

class DeleteMergedBranchesService < BaseService
  def async_execute
    DeleteMergedBranchesWorker.perform_async(project.id, current_user.id)
  end

  def execute
    raise Gitlab::Access::AccessDeniedError unless can?(current_user, :push_code, project)

    # n+1: https://gitlab.com/gitlab-org/gitlab-ce/issues/37438
    Gitlab::GitalyClient.allow_n_plus_1_calls do
      branches = project.repository.branch_names
      branches = branches.select { |branch| project.repository.merged_to_root_ref?(branch) }
      # Prevent deletion of branches relevant to open merge requests
      branches -= merge_request_branch_names
      # Prevent deletion of protected branches
      branches = branches.reject { |branch| ProtectedBranch.protected?(project, branch) }

      branches.each do |branch|
        DeleteBranchService.new(project, current_user).execute(branch)
      end
    end
  end

  private

  def merge_request_branch_names
    # reorder(nil) is necessary for SELECT DISTINCT because default scope adds an ORDER BY
    source_names = project.origin_merge_requests.opened.reorder(nil).uniq.pluck(:source_branch)
    target_names = project.merge_requests.opened.reorder(nil).uniq.pluck(:target_branch)
    (source_names + target_names).uniq
  end
end