【Python高级353】python实现多线程版本的TCP服务器

我叫czc2024-12-26 13:13

前面学了了套接字编程、tcp服务端客户端开发、面向对象版的服务端客户端、带有端口复用的服务端。

这里使用多线程开发多任务版的服务端

多任务版本的TCP服务器

来一个客户，就为其创建一个线程

python 复制代码

import socket
import threading

class WebServer:
    # 3、定义一个__init__方法，初始化套接字对象
    def __init__(self):
        self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 设置端口复用（在上一次关闭服务器后，端口不会立即释放，需要设置端口复用）
        self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)  # 参数2：SOL_SOCKET表示当前套接字，参数3：SO_REUSEADDR表示端口复用，参数4：True表示开启端口复用（默认是false，要等待很长时间端口才会自动释放）
    
        self.tcp_server_socket.bind(("127.0.0.1", 8000))  # 如果是本机，可以不写ip地址
        self.tcp_server_socket.listen(128)  # 128表示最大连接数

    # 4、定义一个start方法，启动服务器，接收客户端连接
    def start(self):
        while True:
            # 等待接受客户端的连接请求
            new_socket, ip_port = self.tcp_server_socket.accept()

            # 当接收到客户端连接请求后创建一个线程（线程内调用自身的handle_request()方法，用于接收和发送消息（封装性））
            sub_thread = threading.Thread(target=self.handle_request, args=(new_socket, ip_port))
            sub_thread.start()

    # 5、定义一个handle_request方法，用于接收和发送消息
    def handle_request(self, new_socket, ip_port):
        # 接收某个客户端发送过来的消息
        recv_data = new_socket.recv(1024).decode("utf-8")  # 实际工作中一条数据大小在1~1.5k之间
        print(f"接收到的数据为：{recv_data}")

        # 发送消息给客户端
        new_socket.send("信息已收到".encode("utf-8"))

        # 关闭套接字
        new_socket.close()

# 定义一个程序的执行入口
if __name__ == "__main__":
    # 1、实例化服务器对象
    server = WebServer()

    # 2、启动服务器
    server.start()