selector群聊最终版

selector群聊最终版

class A:
    def __init__(self, ip='127.0.0.1', port=9999):
        self.addr = ip, port
        self.sock = socket.socket()
        self.s = selectors.DefaultSelector()

    def start(self):
        self.sock.bind(self.addr)
        self.sock.listen()
        self.sock.setblocking(False)
        self.s.register(self.sock, selectors.EVENT_READ, self.accept)
        threading.Thread(target=self.select, name='select', daemon=True).start()

    def accept(self, sock):
        nwsock, self.raddr = self.sock.accept()
        logger.info('{} comming !!!'.format(self.raddr))
        nwsock.setblocking(False)
        self.s.register(nwsock, selectors.EVENT_READ, self.recv)

    def recv(self, nwsock):
        data = nwsock.recv(1024)

        # logger.info(data)
        msg = '{} send: {}'.format(self.raddr, data).encode()
        for key in self.s.get_map().values():
            # logger.info(key)
            if key.data == self.recv:
                key.fileobj.send(msg)
        if data == b'' or data == b'quit':
            self.s.unregister(nwsock)
            nwsock.close()
            logger.info('{} quit !!!'.format(self.raddr))
            return

    def select(self):
        while True:
            events = self.s.select()
            for k, v in events:
                k.data(k.fileobj)

    def stop(self):
        fobj = []
        for fb, key in self.s.get_map().items():
            fobj.append(key.fileobj)
        for i in fobj:
            self.s.unregister(i)
            i.close()
        self.s.close()


a = A()
a.start()

while True:
    cmd = input('>>>')
    if cmd == 'quit':
        a.stop()
        break
    logger.info(threading.enumerate())