class Resque::DataStore::Workers

Public Instance Methods

all_heartbeats() click to toggle source
# File lib/resque/data_store.rb, line 290
def all_heartbeats
  @redis.hgetall(HEARTBEAT_KEY)
end
get_worker_payload(worker_id) click to toggle source

return the worker’s payload i.e. job

# File lib/resque/data_store.rb, line 247
def get_worker_payload(worker_id)
  @redis.get("worker:#{worker_id}")
end
heartbeat(worker) click to toggle source
# File lib/resque/data_store.rb, line 281
def heartbeat(worker)
  heartbeat = @redis.hget(HEARTBEAT_KEY, worker.to_s)
  heartbeat && Time.parse(heartbeat)
end
heartbeat!(worker, time) click to toggle source
# File lib/resque/data_store.rb, line 286
def heartbeat!(worker, time)
  @redis.hset(HEARTBEAT_KEY, worker.to_s, time.iso8601)
end
register_worker(worker) click to toggle source
# File lib/resque/data_store.rb, line 255
def register_worker(worker)
  @redis.pipelined do
    @redis.sadd(:workers, worker)
    worker_started(worker)
  end
end
remove_heartbeat(worker) click to toggle source
# File lib/resque/data_store.rb, line 277
def remove_heartbeat(worker)
  @redis.hdel(HEARTBEAT_KEY, worker.to_s)
end
set_worker_payload(worker, data) click to toggle source
# File lib/resque/data_store.rb, line 294
def set_worker_payload(worker, data)
  @redis.set(redis_key_for_worker(worker), data)
end
unregister_worker(worker, &block) click to toggle source
# File lib/resque/data_store.rb, line 266
def unregister_worker(worker, &block)
  @redis.pipelined do
    @redis.srem(:workers, worker)
    @redis.del(redis_key_for_worker(worker))
    @redis.del(redis_key_for_worker_start_time(worker))
    @redis.hdel(HEARTBEAT_KEY, worker.to_s)

    block.call
  end
end
worker_done_working(worker, &block) click to toggle source
# File lib/resque/data_store.rb, line 302
def worker_done_working(worker, &block)
  @redis.pipelined do
    @redis.del(redis_key_for_worker(worker))
    block.call
  end
end
worker_exists?(worker_id) click to toggle source
# File lib/resque/data_store.rb, line 251
def worker_exists?(worker_id)
  @redis.sismember(:workers, worker_id)
end
worker_ids() click to toggle source
# File lib/resque/data_store.rb, line 235
def worker_ids
  Array(@redis.smembers(:workers))
end
worker_start_time(worker) click to toggle source
# File lib/resque/data_store.rb, line 298
def worker_start_time(worker)
  @redis.get(redis_key_for_worker_start_time(worker))
end
worker_started(worker) click to toggle source
# File lib/resque/data_store.rb, line 262
def worker_started(worker)
  @redis.set(redis_key_for_worker_start_time(worker), Time.now.to_s)
end
workers_map(worker_ids) click to toggle source

Given a list of worker ids, returns a map of those ids to the worker’s value in redis, even if that value maps to nil

# File lib/resque/data_store.rb, line 241
def workers_map(worker_ids)
  redis_keys = worker_ids.map { |id| "worker:#{id}" }
  @redis.mapped_mget(*redis_keys)
end

Public Class Methods

new(redis) click to toggle source
# File lib/resque/data_store.rb, line 231
def initialize(redis)
  @redis = redis
end