Use with statement.

This commit is contained in:
Yuxin Wang 2018-10-01 00:14:26 -04:00
parent 590e9cd93f
commit 8db86a7ab4

View file

@ -90,9 +90,8 @@ class MessageServer:
logger.debug('Message {} from {}'.format(self.__message_log(msg), client.getpeername()))
# process the packets in order
# TODO: remove this lock for better parallelism
self._process_lock.acquire()
self._process_message(client, msg)
self._process_lock.release()
with self._process_lock:
self._process_message(client, msg)
except (EOFError, OSError):
if self._is_running:
logger.warning('{} closed unexpectedly'.format(client.getpeername()))