small fixes in example

This commit is contained in:
izderadicka 2016-05-26 18:06:04 +02:00
parent 7cccac7454
commit f66411e62b
2 changed files with 14 additions and 8 deletions

View File

@ -35,7 +35,7 @@ class MyComponent(ApplicationSession):
if __name__ == '__main__':
level = 'info'
if len(sys.argv) > 1 and sys.argv[1] == ' debug':
if len(sys.argv) > 1 and sys.argv[1] == 'debug':
level = 'debug'
runner = ApplicationRunnerRawSocket(
"tcp://localhost:9090",

View File

@ -78,6 +78,18 @@ class ApplicationRunnerRawSocket(object):
txaio.use_asyncio()
txaio.config.loop = loop
try:
loop.add_signal_handler(signal.SIGTERM, loop.stop)
except NotImplementedError:
# signals are not available on Windows
pass
def handle_error(loop, context):
self.log.error('Application Error: {err}', err=context)
loop.stop()
loop.set_exception_handler(handle_error)
if is_unix:
coro = loop.create_unix_connection(transport_factory, parsed_url.path)
else:
@ -86,17 +98,11 @@ class ApplicationRunnerRawSocket(object):
txaio.start_logging(level=logging_level) # @UndefinedVariable
try:
loop.add_signal_handler(signal.SIGTERM, loop.stop)
except NotImplementedError:
# signals are not available on Windows
pass
try:
loop.run_forever()
except KeyboardInterrupt:
pass
self.log('Left main loop waiting for completiotion')
self.log.debug('Left main loop waiting for completion')
# give Goodbye message a chance to go through, if we still
# have an active session
# it's not working now - because protocol is_closed must return Future