测试 Sidekiq 的重试队列

Testing Sidekiq's retry queue

我们有一个相当关键的组件,如果满足特定条件,它会清除特定项目的 Sidekiq 重试队列。 Sidekiq 的测试 API 没有提供添加到重试队列的方法,所以我们不得不想出一些自定义的东西。

添加到重试队列:

  def add_retry(object: @object, klass: 'HardWorker', at: Time.now.to_f)
    payload = Sidekiq.dump_json(
      class: klass,
      args: [object.id, object.class, 'sidekiq_retry_test'],
      queue: 'user_integration',
      jid: rand(1..10000),
      retry_count: 20,
      failed_at: Time.now.to_f
    )

    Sidekiq.redis do |conn|
      conn.zadd('retry', at.to_s, payload)
    end
  end

您可以更改有效负载哈希中的任何内容。您可能想要指定队列,或添加更强大的解决方案来选择作业 ID。这正是对我们有用的。

事后您需要清理重试队列。您应该在 teardownafter(:all) 块中执行此操作。

  def cleanup_retry_queue
    Sidekiq::RetrySet.new.each do |job|
      job.delete if job.args[2] == 'sidekiq_retry_test'
    end
  end

请记住,这实际上会添加到队列中,所以要小心。仅当您有一个依赖重试队列的可测试组件时才使用它。不要用它来测试 Sidekiq 的排队功能本身。