129 lines
3.5 KiB
Python
Executable File
129 lines
3.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""TCP echo server example."""
|
|
import argparse
|
|
import trollius as asyncio
|
|
import sys
|
|
try:
|
|
import signal
|
|
except ImportError:
|
|
signal = None
|
|
|
|
|
|
class EchoServer(asyncio.Protocol):
|
|
|
|
TIMEOUT = 5.0
|
|
|
|
def timeout(self):
|
|
print('connection timeout, closing.')
|
|
self.transport.close()
|
|
|
|
def connection_made(self, transport):
|
|
print('connection made')
|
|
self.transport = transport
|
|
|
|
# start 5 seconds timeout timer
|
|
self.h_timeout = asyncio.get_event_loop().call_later(
|
|
self.TIMEOUT, self.timeout)
|
|
|
|
def data_received(self, data):
|
|
print('data received: ', data.decode())
|
|
self.transport.write(b'Re: ' + data)
|
|
|
|
# restart timeout timer
|
|
self.h_timeout.cancel()
|
|
self.h_timeout = asyncio.get_event_loop().call_later(
|
|
self.TIMEOUT, self.timeout)
|
|
|
|
def eof_received(self):
|
|
pass
|
|
|
|
def connection_lost(self, exc):
|
|
print('connection lost:', exc)
|
|
self.h_timeout.cancel()
|
|
|
|
|
|
class EchoClient(asyncio.Protocol):
|
|
|
|
message = 'This is the message. It will be echoed.'
|
|
|
|
def connection_made(self, transport):
|
|
self.transport = transport
|
|
self.transport.write(self.message.encode())
|
|
print('data sent:', self.message)
|
|
|
|
def data_received(self, data):
|
|
print('data received:', data)
|
|
|
|
# disconnect after 10 seconds
|
|
asyncio.get_event_loop().call_later(10.0, self.transport.close)
|
|
|
|
def eof_received(self):
|
|
pass
|
|
|
|
def connection_lost(self, exc):
|
|
print('connection lost:', exc)
|
|
asyncio.get_event_loop().stop()
|
|
|
|
|
|
def start_client(loop, host, port):
|
|
t = asyncio.Task(loop.create_connection(EchoClient, host, port))
|
|
loop.run_until_complete(t)
|
|
|
|
|
|
def start_server(loop, host, port):
|
|
f = loop.create_server(EchoServer, host, port)
|
|
return loop.run_until_complete(f)
|
|
|
|
|
|
ARGS = argparse.ArgumentParser(description="TCP Echo example.")
|
|
ARGS.add_argument(
|
|
'--server', action="store_true", dest='server',
|
|
default=False, help='Run tcp server')
|
|
ARGS.add_argument(
|
|
'--client', action="store_true", dest='client',
|
|
default=False, help='Run tcp client')
|
|
ARGS.add_argument(
|
|
'--host', action="store", dest='host',
|
|
default='127.0.0.1', help='Host name')
|
|
ARGS.add_argument(
|
|
'--port', action="store", dest='port',
|
|
default=9999, type=int, help='Port number')
|
|
ARGS.add_argument(
|
|
'--iocp', action="store_true", dest='iocp',
|
|
default=False, help='Use IOCP event loop')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
args = ARGS.parse_args()
|
|
|
|
if ':' in args.host:
|
|
args.host, port = args.host.split(':', 1)
|
|
args.port = int(port)
|
|
|
|
if (not (args.server or args.client)) or (args.server and args.client):
|
|
print('Please specify --server or --client\n')
|
|
ARGS.print_help()
|
|
else:
|
|
if args.iocp:
|
|
from trollius import windows_events
|
|
loop = windows_events.ProactorEventLoop()
|
|
asyncio.set_event_loop(loop)
|
|
else:
|
|
loop = asyncio.get_event_loop()
|
|
print ('Using backend: {0}'.format(loop.__class__.__name__))
|
|
|
|
if signal is not None and sys.platform != 'win32':
|
|
loop.add_signal_handler(signal.SIGINT, loop.stop)
|
|
|
|
if args.server:
|
|
server = start_server(loop, args.host, args.port)
|
|
else:
|
|
start_client(loop, args.host, args.port)
|
|
|
|
try:
|
|
loop.run_forever()
|
|
finally:
|
|
if args.server:
|
|
server.close()
|
|
loop.close()
|