用asyncio搭建高并发的socket server,如果使用python3.5,则使用uvloop,性能将更好
import asyncio
import sys
clients = []
class SimpleChatClientProtocol(asyncio.Protocol):
def connection_made(self, transport):
self.transport = transport
self.peername = transport.get_extra_info("peername")
print("connection_made: {}".format(self.peername))
clients.append(self)
def data_received(self, data):
print("data_received: {}".format(data.decode()))
for client in clients:
if client is not self:
client.transport.write("{}: {}".format(self.peername, data.decode()).encode())
def connection_lost(self, ex):
print("connection_lost: {}".format(self.peername))
clients.remove(self)
if __name__ == '__main__':
print("starting up..")
PY35 = sys.version_info >= (3, 5)
if PY35:
try:
import uvloop
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
print('using uvloop')
except:
print('using asyncio default loop')
else:
print('using asyncio default loop')
loop = asyncio.get_event_loop()
coro = loop.create_server(SimpleChatClientProtocol, port=8000)
server = loop.run_until_complete(coro)
for socket in server.sockets:
print("serving on {}".format(socket.getsockname()))
loop.run_forever()