module ResqueAdmin

Constants

DEFAULT_HEARTBEAT_INTERVAL
DEFAULT_PRUNE_INTERVAL
Version

Attributes

enqueue_front[W]
heartbeat_interval[W]
inline[RW]
inline?[RW]
logger[RW]

Set or retrieve the current logger object

prune_interval[W]

Public Instance Methods

after_fork(&block) click to toggle source

The `after_fork` hook will be run in the child process and is passed the current job. Any changes you make, therefore, will only live as long as the job currently being processed.

Call with a block to register a hook. Call with no arguments to return all registered hooks.

# File lib/resque_admin.rb, line 209
def after_fork(&block)
  block ? register_hook(:after_fork, block) : hooks(:after_fork)
end
after_fork=(block) click to toggle source

Register an after_fork proc.

# File lib/resque_admin.rb, line 214
def after_fork=(block)
  register_hook(:after_fork, block)
end
after_pause(&block) click to toggle source

The `after_pause` hook will be run in the parent process after the worker has paused (via SIGCONT).

# File lib/resque_admin.rb, line 231
def after_pause(&block)
  block ? register_hook(:after_pause, block) : hooks(:after_pause)
end
after_pause=(block) click to toggle source

Register an after_pause proc.

# File lib/resque_admin.rb, line 236
def after_pause=(block)
  register_hook(:after_pause, block)
end
before_first_fork(&block) click to toggle source

The `before_first_fork` hook will be run in the parent process only once, before forking to run the first job. Be careful- any changes you make will be permanent for the lifespan of the worker.

Call with a block to register a hook. Call with no arguments to return all registered hooks.

# File lib/resque_admin.rb, line 179
def before_first_fork(&block)
  block ? register_hook(:before_first_fork, block) : hooks(:before_first_fork)
end
before_first_fork=(block) click to toggle source

Register a before_first_fork proc.

# File lib/resque_admin.rb, line 184
def before_first_fork=(block)
  register_hook(:before_first_fork, block)
end
before_fork(&block) click to toggle source

The `before_fork` hook will be run in the parent process before every job, so be careful- any changes you make will be permanent for the lifespan of the worker.

Call with a block to register a hook. Call with no arguments to return all registered hooks.

# File lib/resque_admin.rb, line 194
def before_fork(&block)
  block ? register_hook(:before_fork, block) : hooks(:before_fork)
end
before_fork=(block) click to toggle source

Register a before_fork proc.

# File lib/resque_admin.rb, line 199
def before_fork=(block)
  register_hook(:before_fork, block)
end
before_pause(&block) click to toggle source

The `before_pause` hook will be run in the parent process before the worker has paused processing (via pause_processing or SIGUSR2).

# File lib/resque_admin.rb, line 220
def before_pause(&block)
  block ? register_hook(:before_pause, block) : hooks(:before_pause)
end
before_pause=(block) click to toggle source

Register a before_pause proc.

# File lib/resque_admin.rb, line 225
def before_pause=(block)
  register_hook(:before_pause, block)
end
classify(dashed_word) click to toggle source

Given a word with dashes, returns a camel cased version of it.

classify('job-name') # => 'JobName'

# File lib/resque_admin.rb, line 58
def classify(dashed_word)
  dashed_word.split('-').map(&:capitalize).join
end
constantize(camel_cased_word) click to toggle source

Tries to find a constant with the name specified in the argument string:

constantize(“Module”) # => Module constantize(“Test::Unit”) # => Test::Unit

The name is assumed to be the one of a top-level constant, no matter whether it starts with “::” or not. No lexical context is taken into account:

C = 'outside' module M

C = 'inside'
C # => 'inside'
constantize("C") # => 'outside', same as ::C

end

NameError is raised when the constant is unknown.

# File lib/resque_admin.rb, line 79
def constantize(camel_cased_word)
  camel_cased_word = camel_cased_word.to_s

  if camel_cased_word.include?('-')
    camel_cased_word = classify(camel_cased_word)
  end

  names = camel_cased_word.split('::')
  names.shift if names.empty? || names.first.empty?

  constant = Object
  names.each do |name|
    args = Module.method(:const_get).arity != 1 ? [false] : []

    if constant.const_defined?(name, *args)
      constant = constant.const_get(name)
    else
      constant = constant.const_missing(name)
    end
  end
  constant
end
data_store()
Alias for: redis
decode(object) click to toggle source

Given a string, returns a Ruby object.

# File lib/resque_admin.rb, line 41
def decode(object)
  return unless object

  begin
    if MultiJson.respond_to?(:dump) && MultiJson.respond_to?(:load)
      MultiJson.load object
    else
      MultiJson.decode object
    end
  rescue ::MultiJson::DecodeError => e
    raise Helpers::DecodeException, e.message, e.backtrace
  end
end
dequeue(klass, *args) click to toggle source

This method can be used to conveniently remove a job from a queue. It assumes the class you're passing it is a real Ruby class (not a string or reference) which either:

a) has a @queue ivar set
b) responds to `queue`

If either of those conditions are met, it will use the value obtained from performing one of the above operations to determine the queue.

If no queue can be inferred this method will raise a `ResqueAdmin::NoQueueError`

If no args are given, this method will dequeue all jobs matching the provided class. See `ResqueAdmin::Job.destroy` for more information.

Returns the number of jobs destroyed.

Example:

# Removes all jobs of class `UpdateNetworkGraph`
ResqueAdmin.dequeue(GitHub::Jobs::UpdateNetworkGraph)

# Removes all jobs of class `UpdateNetworkGraph` with matching args.
ResqueAdmin.dequeue(GitHub::Jobs::UpdateNetworkGraph, 'repo:135325')

This method is considered part of the `stable` API.

# File lib/resque_admin.rb, line 408
def dequeue(klass, *args)
  # Perform before_dequeue hooks. Don't perform dequeue if any hook returns false
  before_hooks = Plugin.before_dequeue_hooks(klass).collect do |hook|
    klass.send(hook, *args)
  end
  return if before_hooks.any? { |result| result == false }

  destroyed = Job.destroy(queue_from_class(klass), klass, *args)

  Plugin.after_dequeue_hooks(klass).each do |hook|
    klass.send(hook, *args)
  end

  destroyed
end
encode(object) click to toggle source

Given a Ruby object, returns a string suitable for storage in a queue.

# File lib/resque_admin.rb, line 32
def encode(object)
  if MultiJson.respond_to?(:dump) && MultiJson.respond_to?(:load)
    MultiJson.dump object
  else
    MultiJson.encode object
  end
end
enqueue(klass, *args) click to toggle source

This method can be used to conveniently add a job to a queue. It assumes the class you're passing it is a real Ruby class (not a string or reference) which either:

a) has a @queue ivar set
b) responds to `queue`

If either of those conditions are met, it will use the value obtained from performing one of the above operations to determine the queue.

If no queue can be inferred this method will raise a `ResqueAdmin::NoQueueError`

Returns true if the job was queued, nil if the job was rejected by a before_enqueue hook.

This method is considered part of the `stable` API.

# File lib/resque_admin.rb, line 352
def enqueue(klass, *args)
  enqueue_to(queue_from_class(klass), klass, *args)
end
enqueue_front() click to toggle source
# File lib/resque_admin.rb, line 167
def enqueue_front
  return @enqueue_front unless @enqueue_front.nil?
  @enqueue_front = false
end
enqueue_to(queue, klass, *args) click to toggle source

Just like `enqueue` but allows you to specify the queue you want to use. Runs hooks.

`queue` should be the String name of the queue you're targeting.

Returns true if the job was queued, nil if the job was rejected by a before_enqueue hook.

This method is considered part of the `stable` API.

# File lib/resque_admin.rb, line 365
def enqueue_to(queue, klass, *args)
  # Perform before_enqueue hooks. Don't perform enqueue if any hook returns false
  before_hooks = Plugin.before_enqueue_hooks(klass).collect do |hook|
    klass.send(hook, *args)
  end
  return nil if before_hooks.any? { |result| result == false }

  Job.create(queue, klass, *args)

  Plugin.after_enqueue_hooks(klass).each do |hook|
    klass.send(hook, *args)
  end

  return true
end
heartbeat_interval() click to toggle source
# File lib/resque_admin.rb, line 157
def heartbeat_interval
  @heartbeat_interval || DEFAULT_HEARTBEAT_INTERVAL
end
info() click to toggle source

Returns a hash, similar to redis-rb's info, of interesting stats.

# File lib/resque_admin.rb, line 484
def info
  return {
    :pending   => queue_sizes.inject(0) { |sum, (queue_name, queue_size)| sum + queue_size },
    :processed => Stat[:processed],
    :queues    => queues.size,
    :workers   => workers.size.to_i,
    :working   => working.size,
    :failed    => data_store.num_failed,
    :servers   => [redis_id],
    :environment  => ENV['RAILS_ENV'] || ENV['RACK_ENV'] || 'development'
  }
end
keys() click to toggle source

Returns an array of all known ResqueAdmin keys in Redis. Redis' KEYS operation is O(N) for the keyspace, so be careful - this can be slow for big databases.

# File lib/resque_admin.rb, line 499
def keys
  data_store.all_resque_keys
end
list_range(key, start = 0, count = 1) click to toggle source

Does the dirty work of fetching a range of items from a Redis list and converting them into Ruby objects.

# File lib/resque_admin.rb, line 306
def list_range(key, start = 0, count = 1)
  results = data_store.list_range(key, start, count)
  if count == 1
    decode(results)
  else
    results.map { |result| decode(result) }
  end
end
peek(queue, start = 0, count = 1) click to toggle source

Returns an array of items currently queued. Queue name should be a string.

start and count should be integer and can be used for pagination. start is the item to begin, count is how many items to return.

To get the 3rd page of a 30 item, paginatied list one would use:

ResqueAdmin.peek('my_list', 59, 30)
# File lib/resque_admin.rb, line 295
def peek(queue, start = 0, count = 1)
  results = data_store.peek_in_queue(queue,start,count)
  if count == 1
    decode(results)
  else
    results.map { |result| decode(result) }
  end
end
pop(queue) click to toggle source

Pops a job off a queue. Queue name should be a string.

Returns a Ruby object.

# File lib/resque_admin.rb, line 277
def pop(queue)
  decode(data_store.pop_from_queue(queue))
end
prune_interval() click to toggle source
# File lib/resque_admin.rb, line 162
def prune_interval
  @prune_interval || DEFAULT_PRUNE_INTERVAL
end
push(queue, item) click to toggle source

Pushes a job onto a queue. Queue name should be a string and the item should be any JSON-able Ruby object.

ResqueAdmin works generally expect the `item` to be a hash with the following keys:

class - The String name of the job to run.
 args - An Array of arguments to pass the job. Usually passed
        via `class.to_class.perform(*args)`.

Example

ResqueAdmin.push('archive', :class => 'Archive', :args => [ 35, 'tar' ])

Returns nothing

# File lib/resque_admin.rb, line 270
def push(queue, item)
  data_store.push_to_queue(queue,encode(item))
end
queue_from_class(klass) click to toggle source

Given a class, try to extrapolate an appropriate queue based on a class instance variable or `queue` method.

# File lib/resque_admin.rb, line 426
def queue_from_class(klass)
  klass.instance_variable_get(:@queue) ||
    (klass.respond_to?(:queue) and klass.queue)
end
queue_sizes() click to toggle source

Returns a hash, mapping queue names to queue sizes

# File lib/resque_admin.rb, line 504
def queue_sizes
  queue_names = queues

  sizes = redis.pipelined do
    queue_names.each do |name|
      redis.llen("queue:#{name}")
    end
  end

  Hash[queue_names.zip(sizes)]
end
queues() click to toggle source

Returns an array of all known ResqueAdmin queues as strings.

# File lib/resque_admin.rb, line 316
def queues
  data_store.queue_names
end
redis() click to toggle source

Returns the current Redis connection. If none has been created, will create a new one.

# File lib/resque_admin.rb, line 139
def redis
  return @data_store if @data_store
  self.redis = Redis.respond_to?(:connect) ? Redis.connect : "localhost:6379"
  self.redis
end
Also aliased as: data_store
redis=(server) click to toggle source

Accepts:

1. A 'hostname:port' String
2. A 'hostname:port:db' String (to select the Redis db)
3. A 'hostname:port/namespace' String (to set the Redis namespace)
4. A Redis URL String 'redis://host:port'
5. An instance of `Redis`, `Redis::Client`, `Redis::DistRedis`,
   or `Redis::Namespace`.
6. An Hash of a redis connection {:host => 'localhost', :port => 6379, :db => 0}
# File lib/resque_admin.rb, line 112
def redis=(server)
  case server
  when String
    if server =~ /redis\:\/\//
      redis = Redis.connect(:url => server, :thread_safe => true)
    else
      server, namespace = server.split('/', 2)
      host, port, db = server.split(':')
      redis = Redis.new(:host => host, :port => port,
        :thread_safe => true, :db => db)
    end
    namespace ||= :resque_admin

    @data_store = ResqueAdmin::DataStore.new(Redis::Namespace.new(namespace, :redis => redis))
  when Redis::Namespace
    @data_store = ResqueAdmin::DataStore.new(server)
  when ResqueAdmin::DataStore
    @data_store = server
  when Hash
    @data_store = ResqueAdmin::DataStore.new(Redis::Namespace.new(:resque_admin, :redis => Redis.new(server)))
  else
    @data_store = ResqueAdmin::DataStore.new(Redis::Namespace.new(:resque_admin, :redis => server))
  end
end
redis_id() click to toggle source
# File lib/resque_admin.rb, line 146
def redis_id
  data_store.identifier
end
remove_queue(queue) click to toggle source

Given a queue name, completely deletes the queue.

# File lib/resque_admin.rb, line 321
def remove_queue(queue)
  data_store.remove_queue(queue)
end
remove_worker(worker_id) click to toggle source

A shortcut to unregister_worker useful for command line tool

# File lib/resque_admin.rb, line 474
def remove_worker(worker_id)
  worker = ResqueAdmin::Worker.find(worker_id)
  worker.unregister_worker
end
reserve(queue) click to toggle source

This method will return a `ResqueAdmin::Job` object or a non-true value depending on whether a job can be obtained. You should pass it the precise name of a queue: case matters.

This method is considered part of the `stable` API.

# File lib/resque_admin.rb, line 436
def reserve(queue)
  Job.reserve(queue)
end
sample_queues(sample_size = 1000) click to toggle source

Returns a hash, mapping queue names to (up to `sample_size`) samples of jobs in that queue

# File lib/resque_admin.rb, line 517
def sample_queues(sample_size = 1000)
  queue_names = queues

  samples = redis.pipelined do
    queue_names.each do |name|
      key = "queue:#{name}"
      redis.llen(key)
      redis.lrange(key, 0, sample_size - 1)
    end
  end

  hash = {}

  queue_names.zip(samples.each_slice(2).to_a) do |queue_name, (queue_size, serialized_samples)|
    samples = serialized_samples.map do |serialized_sample|
      Job.decode(serialized_sample)
    end

    hash[queue_name] = {
      :size => queue_size,
      :samples => samples
    }
  end

  hash
end
size(queue) click to toggle source

Returns an integer representing the size of a queue. Queue name should be a string.

# File lib/resque_admin.rb, line 283
def size(queue)
  data_store.queue_size(queue)
end
to_s() click to toggle source
# File lib/resque_admin.rb, line 240
def to_s
  "ResqueAdmin Client connected to #{redis_id}"
end
validate(klass, queue = nil) click to toggle source

Validates if the given klass could be a valid ResqueAdmin job

If no queue can be inferred this method will raise a `ResqueAdmin::NoQueueError`

If given klass is nil this method will raise a `ResqueAdmin::NoClassError`

# File lib/resque_admin.rb, line 445
def validate(klass, queue = nil)
  queue ||= queue_from_class(klass)

  if !queue
    raise NoQueueError.new("Jobs must be placed onto a queue. No queue could be inferred for class #{klass}")
  end

  if klass.to_s.empty?
    raise NoClassError.new("Jobs must be given a class.")
  end
end
watch_queue(queue) click to toggle source

Used internally to keep track of which queues we've created. Don't call this directly.

# File lib/resque_admin.rb, line 327
def watch_queue(queue)
  data_store.watch_queue(queue)
end
workers() click to toggle source

A shortcut to Worker.all

# File lib/resque_admin.rb, line 463
def workers
  Worker.all
end
working() click to toggle source

A shortcut to Worker.working

# File lib/resque_admin.rb, line 468
def working
  Worker.working
end

Private Instance Methods

clear_hooks(name) click to toggle source

Clear all hooks given a hook name.

# File lib/resque_admin.rb, line 561
def clear_hooks(name)
  @hooks && @hooks[name] = []
end
hooks() click to toggle source

Retrieve all hooks

# File lib/resque_admin.rb, line 566
def hooks
  @hooks || {}
end
register_hook(name, block) click to toggle source

Register a new proc as a hook. If the block is nil this is the equivalent of removing all hooks of the given name.

`name` is the hook that the block should be registered with.

# File lib/resque_admin.rb, line 550
def register_hook(name, block)
  return clear_hooks(name) if block.nil?

  @hooks ||= {}
  @hooks[name] ||= []

  block = Array(block)
  @hooks[name].concat(block)
end