debian-mirror-gitlab/app/workers/chat_notification_worker.rb

# frozen_string_literal: true

class ChatNotificationWorker
  include ApplicationWorker

  TimeoutExceeded = Class.new(StandardError)

  sidekiq_options retry: false
  feature_category :chatops
  latency_sensitive_worker!
  weight 2

  # TODO: break this into multiple jobs
  # as the `responder` uses external dependencies
  # See https://gitlab.com/gitlab-com/gl-infra/scalability/issues/34
  # worker_has_external_dependencies!

  RESCHEDULE_INTERVAL = 2.seconds
  RESCHEDULE_TIMEOUT = 5.minutes

  # rubocop: disable CodeReuse/ActiveRecord
  def perform(build_id, reschedule_count = 0)
    Ci::Build.find_by(id: build_id).try do |build|
      send_response(build)
    end
  rescue Gitlab::Chat::Output::MissingBuildSectionError
    raise TimeoutExceeded if timeout_exceeded?(reschedule_count)

    # The creation of traces and sections appears to be eventually consistent.
    # As a result it's possible for us to run the above code before the trace
    # sections are present. To better handle such cases we'll just reschedule
    # the job instead of producing an error.
    self.class.perform_in(RESCHEDULE_INTERVAL, build_id, reschedule_count + 1)
  end
  # rubocop: enable CodeReuse/ActiveRecord

  def send_response(build)
    Gitlab::Chat::Responder.responder_for(build).try do |responder|
      if build.success?
        output = Gitlab::Chat::Output.new(build)

        responder.success(output.to_s)
      else
        responder.failure
      end
    end
  end

  private

  def timeout_exceeded?(reschedule_count)
    (reschedule_count * RESCHEDULE_INTERVAL) >= RESCHEDULE_TIMEOUT
  end
end