service: centralize fetching thread name, and tidy up logs

pull/612/head
David Wilson 5 years ago
parent f0782ccd42
commit 11c7e3f561

@ -99,6 +99,10 @@ def get_or_create_pool(size=None, router=None):
return _pool return _pool
def get_thread_name():
return threading.currentThread().getName()
def call(service_name, method_name, call_context=None, **kwargs): def call(service_name, method_name, call_context=None, **kwargs):
""" """
Call a service registered with this pool, using the calling thread as a Call a service registered with this pool, using the calling thread as a
@ -611,10 +615,11 @@ class Pool(object):
try: try:
event = self._select.get_event() event = self._select.get_event()
except mitogen.core.LatchError: except mitogen.core.LatchError:
LOG.debug('%r: graceful exit', self) LOG.debug('thread %s exiting gracefully', get_thread_name())
return return
except mitogen.core.ChannelError: except mitogen.core.ChannelError:
LOG.debug('%r: exitting: %s', self, sys.exc_info()[1]) LOG.debug('thread %s exiting with error: %s',
get_thread_name(), sys.exc_info()[1])
return return
func = self._func_by_source[event.source] func = self._func_by_source[event.source]
@ -627,16 +632,14 @@ class Pool(object):
try: try:
self._worker_run() self._worker_run()
except Exception: except Exception:
th = threading.currentThread() LOG.exception('%r: worker %r crashed', self, get_thread_name())
LOG.exception('%r: worker %r crashed', self, th.getName())
raise raise
def __repr__(self): def __repr__(self):
th = threading.currentThread()
return 'Pool(%04x, size=%d, th=%r)' % ( return 'Pool(%04x, size=%d, th=%r)' % (
id(self) & 0xffff, id(self) & 0xffff,
len(self._threads), len(self._threads),
th.getName(), get_thread_name(),
) )
@ -752,7 +755,7 @@ class PushFileService(Service):
def store_and_forward(self, path, data, context): def store_and_forward(self, path, data, context):
LOG.debug('%r.store_and_forward(%r, %r, %r) %r', LOG.debug('%r.store_and_forward(%r, %r, %r) %r',
self, path, data, context, self, path, data, context,
threading.currentThread().getName()) get_thread_name())
self._lock.acquire() self._lock.acquire()
try: try:
self._cache[path] = data self._cache[path] = data

Loading…
Cancel
Save