Richard Huang's Website

为resque写扩展

resque是基于redis的ruby类库，用于创建后台任务，把这些后台任务放在多个队列中去，之后在处理它们。github就是使用resque来处理它们的后台任务的。

对于需要长时间处理的任务，比如发送email，发tweet，图片resize等等，都是resque的用武之地。默认resque就是将任务加到redis的队列中去，然后定时取出来去处理，实际项目中我们往往需要对其增加额外的扩展，比如你需要增加日志功能，增加处理次数的限制，这个时候就可以给resque写一个plugin，就像rails的plugin一样。

resque定义了非常良好的HOOK，使得为其写扩展变得更加容易。

resque采取的是每隔n秒从队列中获取一个任务，然后fork一个子进程来执行这个任务。resque定义了before_fork, after_fork, before_perform, after_perform, around_perform, on_failure几个hook，执行顺序如下

before_fork
fork
after_fork
before_perform
around_perform
perform
around_perfomr
after_perform

还有就是发生错误的时候，on_failure会被执行。

再给一个我写的resque-restriction插件的实例

def before_perform_restriction(*args)
  settings.each do |period, number|
    key = redis_key(period)
    value = get_restrict(key)

    if value.nil? or value == ""
      set_restrict(key, seconds(period), number)
    elsif value.to_i <= 0
      Resque.push "restriction", :class => to_s, :args => args
      raise Resque::Job::DontPerform
    end
  end
end

def after_perform_restriction(*args)
  settings.each do |period, number|
    key = redis_key(period)
    Resque.redis.decrby(key, 1)
  end
end