debian-mirror-gitlab/app/services/auto_merge_service.rb

78 lines
2.3 KiB
Ruby
Raw Normal View History

2019-09-04 21:01:54 +05:30
# frozen_string_literal: true
class AutoMergeService < BaseService
2020-04-22 19:07:51 +05:30
include Gitlab::Utils::StrongMemoize
2019-12-04 20:38:33 +05:30
STRATEGY_MERGE_WHEN_PIPELINE_SUCCEEDS = 'merge_when_pipeline_succeeds'
2019-09-04 21:01:54 +05:30
STRATEGIES = [STRATEGY_MERGE_WHEN_PIPELINE_SUCCEEDS].freeze
class << self
2020-04-22 19:07:51 +05:30
def all_strategies_ordered_by_preference
2019-09-04 21:01:54 +05:30
STRATEGIES
end
def get_service_class(strategy)
2020-04-22 19:07:51 +05:30
return unless all_strategies_ordered_by_preference.include?(strategy)
2019-09-04 21:01:54 +05:30
"::AutoMerge::#{strategy.camelize}Service".constantize
end
end
2020-04-22 19:07:51 +05:30
def execute(merge_request, strategy = nil)
strategy ||= preferred_strategy(merge_request)
service = get_service_instance(merge_request, strategy)
2019-09-04 21:01:54 +05:30
return :failed unless service&.available_for?(merge_request)
service.execute(merge_request)
end
def update(merge_request)
return :failed unless merge_request.auto_merge_enabled?
2020-04-22 19:07:51 +05:30
strategy = merge_request.auto_merge_strategy
get_service_instance(merge_request, strategy).update(merge_request)
2019-09-04 21:01:54 +05:30
end
def process(merge_request)
return unless merge_request.auto_merge_enabled?
2020-04-22 19:07:51 +05:30
strategy = merge_request.auto_merge_strategy
get_service_instance(merge_request, strategy).process(merge_request)
2019-09-04 21:01:54 +05:30
end
def cancel(merge_request)
return error("Can't cancel the automatic merge", 406) unless merge_request.auto_merge_enabled?
2020-04-22 19:07:51 +05:30
strategy = merge_request.auto_merge_strategy
get_service_instance(merge_request, strategy).cancel(merge_request)
2019-09-04 21:01:54 +05:30
end
2019-09-30 21:07:59 +05:30
def abort(merge_request, reason)
return error("Can't abort the automatic merge", 406) unless merge_request.auto_merge_enabled?
2020-04-22 19:07:51 +05:30
strategy = merge_request.auto_merge_strategy
get_service_instance(merge_request, strategy).abort(merge_request, reason)
2019-09-30 21:07:59 +05:30
end
2019-09-04 21:01:54 +05:30
def available_strategies(merge_request)
2020-04-22 19:07:51 +05:30
self.class.all_strategies_ordered_by_preference.select do |strategy|
get_service_instance(merge_request, strategy).available_for?(merge_request)
2019-09-04 21:01:54 +05:30
end
end
2020-04-22 19:07:51 +05:30
def preferred_strategy(merge_request)
available_strategies(merge_request).first
end
2019-09-04 21:01:54 +05:30
private
2020-04-22 19:07:51 +05:30
def get_service_instance(merge_request, strategy)
strong_memoize("service_instance_#{merge_request.id}_#{strategy}") do
self.class.get_service_class(strategy)&.new(project, current_user, params)
end
2019-09-04 21:01:54 +05:30
end
end
2019-12-04 20:38:33 +05:30
2021-06-08 01:23:25 +05:30
AutoMergeService.prepend_mod_with('AutoMergeService')