resque-restriction

Resque Restriction is a plugin for the Resque queueing system (http://github.com/defunkt/resque). It adds two functions:

it will limit the execution number of certain jobs in a period time. For example, it can limit a certain job can be executed 1000 times per day, 100 time per hour and 30 times per 300 seconds.
it will execute the exceeded jobs at the next period. For example, you restrict the email sending jobs to run 1000 times per day. If your system generates 1010 email sending jobs, only 1000 email sending jobs can be executed today, and the other 10 email sending jobs will be executed tomorrow.

Resque Restriction requires Resque 1.7.0.

Attention

The identifier method is renamed to restriction_identifier to solve the confliction with resque-retry from version 0.3.0.

Install

sudo gem install resque-restriction

To use

It is especially useful when a system has an email invitation resque job, because sending emails too frequentyly will be treated as a spam. What you should do for the InvitationJob is to inherit it from Resque::Plugins::RestrictionJob class and add restrict definition. Example:

class InvitationJob < Resque::Plugins::RestrictionJob
  restrict :per_day => 1000, :per_hour => 100, :per_300 => 30

  #rest of your class here
end

That means the InvitationJob can not be executed more than 1000 times per day, 100 times per hour and 30 times per 300 seconds. All restrictions have to be met for the job to execute.

The argument of restrict method is a hash, the key of the hash is a period time, including :concurrent, :per_minute, :per_hour, :per_day, :per_week, :per_month, :per_year, and you can also define any period like :per_300 means per 300 seconds. The value of the hash is the job execution limit number in a period. The :concurrent option restricts the number of jobs that run simultaneously.

Advance

You can also add customized restriction as you like. For example, we have a job to restrict the facebook post numbers 40 times per user per day, we can define as:

class GenerateFacebookShares < Resque::Plugins::RestrictionJob
  restrict :per_day => 40

  def self.restriction_identifier(options)
    [self.to_s, options["user_id"]].join(":")
  end

  #rest of your class here
end

options["user_id"] returns the user's facebook uid, the key point is that the different restriction_identifiers can restrict different job execution numbers.

Contributing

Once you've made your commits:

Fork Resque Restriction
Create a topic branch - git checkout -b my_branch
Push to your branch - git push origin my_branch
Create an Issue with a link to your branch
That's it!

Author

Richard Huang :: flyerhzm@gmail.com :: @flyerhzm

Contributors

Matt Conway :: matt@conwaysplace.com :: @mattconway

Martin Fourcade :: fourcade.m@gmail.com :: @mfourcade

bioritmo / resque-restriction