python asyncio uvloop socket server

来源:互联网 发布:哈登什么数据是第一 编辑:程序博客网 时间:2024/05/22 22:41
用asyncio搭建高并发的socket server,如果使用python3.5,则使用uvloop,性能将更好
import asyncio
import sysclients = []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()