python asyncio uvloop socket server

用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()

你可能感兴趣的:(python)