gitlab-org--gitlab-foss/app/services/ci/stop_environments_service.rb

# frozen_string_literal: true

module Ci
  class StopEnvironmentsService < BaseService
    attr_reader :ref

    def execute(branch_name)
      @ref = branch_name

      return unless @ref.present?

      environments.each { |environment| stop(environment) }
    end

    def execute_for_merge_request(merge_request)
      merge_request.environments.each { |environment| stop(environment) }
    end

    ##
    # This method is for stopping multiple environments in a batch style.
    # The maximum acceptable count of environments is roughly 5000. Please
    # apply acceptable `LIMIT` clause to the `environments` relation.
    def self.execute_in_batch(environments)
      stop_actions = environments.stop_actions.load

      environments.update_all(auto_stop_at: nil, state: 'stopped')

      stop_actions.each do |stop_action|
        stop_action.play(stop_action.user)
      rescue => e
        Gitlab::ErrorTracking.track_exception(e, deployable_id: stop_action.id)
      end
    end

    private

    def environments
      @environments ||= EnvironmentsFinder
        .new(project, current_user, ref: @ref, recently_updated: true)
        .execute
    end

    def stop(environment)
      return unless environment.stop_action_available?
      return unless can?(current_user, :stop_environment, environment)

      environment.stop_with_action!(current_user)
    end
  end
end