手写微信群聊服务器

需求分析

1. 能够连接多个客户端；

2. 客户端长时间未响应能够断开节省资源；

3. 能够群发，一个用户发来一条消息，所有客户端都可以接收；

 

思路

直接使用socket.socket类就可以实现，但是这样的话我们需要自己手动写监听，写accept，把accept单独放入一个线程，每连入一个线程就新开辟一个线程。

 

这里我们选用socketserver这个库来实现，这样我们只需要把核心步骤写出来即可，其他的都能自动化帮我们完成了。

mport socketserver
import datetime
import threading


class MyHandle(socketserver.BaseRequestHandler):  # 每一个线程就一个实例
    def self_server_init(self):
        if not hasattr(self.server, "clients"):
            setattr(self.server, "clients", {})  # 如果没有这个属性就给他增加一个
        if not hasattr(self.server, "_lock_clients"):
            setattr(self.server, "lock_clients", threading.Lock())
        self.server.__dict__["hb_interval"] = 10
        self.server.clients[self.key] = datetime.datetime.now().timestamp()  # 记录时间

    def setup(self):
        # self init
        self.event = threading.Event()
        self.key = self.request, self.event
        # self.server init
        self.self_server_init()

    def handle(self):
        no_hb = set()
        while not self.event.is_set():
            data = self.request.recv(1024)
            print(data.decode())
            if data == b"^hb^":   # 收到了心跳包
                with self.server.lock_clients:
                    self.server.clients[self.key] = datetime.datetime.now().timestamp()
            if data == b"" or data.strip() == b"quit":
                with self.server.lock_clients:
                    self.server.clients.pop(self.key, None)  # 如果退出就弹出了
                break  # 因为一个线程是一个连接

            # 如果没有退出我们就刷新时间，也有可能是新的连接
            self.server.clients[self.key] = datetime.datetime.now().timestamp()

            # 发送消息的时候检查是否超时了
            with self.server.lock_clients:
                for key, t in self.server.clients.items():
                    if datetime.datetime.now().timestamp() - t > self.server.hb_interval:  # 如果超时了记录一下这个key
                        no_hb.add(key)  # s和e都可哈希
                        continue
                    key[0].send("from {}: {}".format(self.client_address, data.decode()).encode())
                for key in no_hb:
                    self.server.clients.pop(key)  # 移除没有心跳的客户端
                no_hb.clear()

    def finish(self):
        super().finish()
        self.event.set()


server = socketserver.ThreadingTCPServer(("127.0.0.1", 9999), MyHandle)
server.serve_forever()   # 启动