Skip to content

Commit 183f22f

Browse files
committed
rework setproctitle() usage to be more easily greppable
Makes it easier to use a monitoring tool to ensure that the worker is running and processing the right queues.
1 parent 429d78a commit 183f22f

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

pyres/worker.py

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,11 @@ def __str__(self):
111111
return self.id
112112
return '%s:%s:%s' % (self.hostname, self.pid, ','.join(self.queues))
113113

114+
def _setproctitle(self, msg):
115+
setproctitle("pyres_worker-%s [%s]: %s" % (__version__,
116+
','.join(self.queues),
117+
msg))
118+
114119
def work(self, interval=5):
115120
"""Invoked by ``run`` method. ``work`` listens on a list of queues and sleeps
116121
for ``interval`` time.
@@ -124,7 +129,7 @@ def work(self, interval=5):
124129
Finally, the ``process`` method actually processes the job by eventually calling the Job instance's ``perform`` method.
125130
126131
"""
127-
setproctitle('pyres_worker-%s: Starting' % __version__)
132+
self._setproctitle("Starting")
128133
self.startup()
129134

130135
while True:
@@ -140,10 +145,9 @@ def work(self, interval=5):
140145
self.before_fork(job)
141146
self.child = os.fork()
142147
if self.child:
143-
setproctitle("pyres_worker%s: Forked %s at %s" %
144-
(__version__,
145-
self.child,
146-
datetime.datetime.now()))
148+
self._setproctitle("Forked %s at %s" %
149+
(self.child,
150+
datetime.datetime.now()))
147151
logger.info('Forked %s at %s' % (self.child,
148152
datetime.datetime.now()))
149153

@@ -157,9 +161,9 @@ def work(self, interval=5):
157161
#os.wait()
158162
logger.debug('done waiting')
159163
else:
160-
setproctitle("pyres_worker-%s: Processing %s since %s" %
161-
(__version__, job._queue,
162-
datetime.datetime.now()))
164+
self._setproctitle("Processing %s since %s" %
165+
(job._queue,
166+
datetime.datetime.now()))
163167
logger.info('Processing %s since %s' %
164168
(job._queue, datetime.datetime.now()))
165169
self.after_fork(job)
@@ -170,8 +174,7 @@ def work(self, interval=5):
170174
if interval == 0:
171175
break
172176
#procline @paused ? "Paused" : "Waiting for #{@queues.join(',')}"
173-
setproctitle("pyres_worker-%s: Waiting for %s " %
174-
(__version__, ','.join(self.queues)))
177+
self._setproctitle("Waiting")
175178
#time.sleep(interval)
176179
self.unregister_worker()
177180

0 commit comments

Comments
 (0)