有没有办法在工作过程中重新启动 Sidekiq 之前运行代码？

Question

我有一个 Sidekiq 作业，每 4 分钟运行一次。

此作业在再次执行代码之前检查当前代码块是否正在执行

process = ProcessTime.where("name = 'ad_queue_process'").first

# Return if job is running
return if process.is_running == true

如果 Sidekiq 在代码块中途重新启动，则更新作业状态的代码永远不会运行

# Done running, update the process times and allow it to be ran again
process.update_attributes(is_running: false, last_execution_time: Time.now)

除非我运行更新语句来设置is_running = false，否则会导致作业永远不会运行

有什么方法可以在 Sidekiq 重启之前执行代码？

Answer 1

更新：

• 感谢@Aaron，并根据我们的讨论（下面的cmets），ensure 块（由分叉的工作线程执行）只能运行几毫秒，然后主线程强制终止这些工作线程，以便主线程对异常堆栈进行一些“清理”，以避免被 Heroku SIGKILL。因此，请确保您的 ensure 代码应该非常快！

TL;DR：

def perform(*args)
  # your code here
ensure
  process.update_attributes(is_running: false, last_execution_time: Time.now)
end

• 上面的ensure 总是被调用，无论方法“成功”还是引发异常。我测试了这个：看到这个repl code，然后点击“运行”

• 换句话说，即使信号是SIGTERM（正常关闭信号），它也总是在SignalException 上调用，但ONLY EXCEPT 在SIGKILL 上（强制无法挽救的关闭） )。您可以通过检查我的repl code 来验证此行为，然后将Process.kill('TERM', Process.pid) 更改为Process.kill('KILL', Process.pid)，然后再次单击“运行”（您会注意到puts 不会被调用）

• 看着Heroku docs，我引用：

  当 Heroku 将关闭一个测功机（重启或新部署等）时，它首先向测功机中的进程发送一个 SIGTERM 信号。

  Heroku 向您的应用程序发送 SIGTERM 后，它将等待几秒钟然后发送 SIGKILL 以强制它关闭，即使它还没有完成清理。在这个例子中，根本没有调用 ensure 块，程序直接退出

  ... 这意味着将调用 ensure 块，因为它是 SIGTERM 而不是 SIGKILL，除非关闭需要很长时间，这可能是由于（某些原因我可以想想ATM）：

  • perform 代码（或堆栈中的任何 ruby​​ 代码；甚至是 gem）中的某些东西也拯救了 SignalException，甚至拯救了根 Exception 类，因为 SignalException 是 Exception 的子类) 但需要很长时间清理（即清理 connections 到 DB 或其他东西，或者挂起你的应用程序的 I/O 东西）

  • 或者，您自己的ensure 块需要很长时间。即在执行process.update_attributes(...) 时，由于某种原因数据库临时挂起/网络延迟或超时，那么update 可能根本不会成功！并且会用完时间，其中从我上面的引述来看，在SIGTERM 之后几秒钟后，应用程序将被 Heroku 发送SIGKILL 强制停止。

...这一切都意味着我的解决方案仍然不完全可靠，但在正常情况下应该可以工作

Answer 2

处理 sidekiq 关闭异常

class SomeWorker
  include Sidekiq::Worker

  sidekiq_options queue: :default

  def perform(params)
    ...

  rescue Sidekiq::Shutdown
    SomeWorker.perform_async(params)
  end
end