Skip to content

Commit

Permalink
Reorganized the methods in DrequeWorker
Browse files Browse the repository at this point in the history
  • Loading branch information
samuel committed Nov 27, 2009
1 parent d302ec1 commit da4db24
Showing 1 changed file with 16 additions and 14 deletions.
30 changes: 16 additions & 14 deletions dreque/worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,20 +54,6 @@ def unregister_worker(self):
self.stats.clear("processed:"+self.worker_id)
self.stats.clear("failed:"+self.worker_id)

def workers(self):
return self.redis.smembers(self._redis_key("workers"))

def working(self):
workers = self.list_workers()
if not workers:
return []

keys = [self._redis_key("worker:"+x) for x in workers]
return dict((x, y) for x, y in zip(self.redis.mget(workers, keys)))

def worker_exists(self, worker_id):
return self.redis.sismember(self._redis_key("workers"), worker_id)

def working_on(self, job):
self.redis.set(self._redis_key("worker:"+self.worker_id),
dict(
Expand Down Expand Up @@ -102,3 +88,19 @@ def lookup_function(self, name):
func = getattr(mod, func_name)
self.function_cache[name] = func
return func

#

def workers(self):
return self.redis.smembers(self._redis_key("workers"))

def working(self):
workers = self.list_workers()
if not workers:
return []

keys = [self._redis_key("worker:"+x) for x in workers]
return dict((x, y) for x, y in zip(self.redis.mget(workers, keys)))

def worker_exists(self, worker_id):
return self.redis.sismember(self._redis_key("workers"), worker_id)

0 comments on commit da4db24

Please sign in to comment.