Python学习---基于多线程的TCP服务端聊天框架搭建案例

"""
# 导入模块
# 创建套接字
# 设置地址重用
# 绑定端口号
# 设置监听，套接字主动设置为被动
# 接收客户端连接
# 接收客户端发送的消息
# 解码数据并且传输
# 关闭和当前客户端的连接
"""
# 导入模块
import socket
import threading


def recv_msg(new_client_socket_avg, ip_port_avg):
    try:
        while True:
            # 接收客户端发送的消息
            recv_data = new_client_socket_avg.recv(1024)
            if recv_data:
                # 解码数据并且传输
                recv_text = recv_data.decode("gbk")
                print(f"收到来自ip地址{ip_port_avg[0]}，端口号为{ip_port_avg[1]}的信息:{recv_text}")
            else:
                # 如果recv返回空数据，表示客户端已经关闭连接
                print(f"客户端 {ip_port_avg[0]} 已断开连接")
                break
        # 关闭和当前客户端的连接
        new_client_socket_avg.close()
    except ConnectionAbortedError as e:
        print(f"连接被客户端中断，错误信息：{e}")
    finally:
        # 确保连接被关闭
        new_client_socket_avg.close()
        print(f"与{ip_port_avg}的连接已关闭。")


# 创建套接字
tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 设置地址重用
tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
# 绑定端口号
tcp_server_socket.bind(("", 10000))
# 设置监听，套接字主动设置为被动
tcp_server_socket.listen(128)
while True:
    # 接收客户端连接
    new_client_socket, ip_port = tcp_server_socket.accept()
    print(f"新用户: {ip_port[0]} 上线")
    # recv_msg(new_client_socket, ip_port)
    # 创建线程
    thread_recvmsg = threading.Thread(target=recv_msg, args=(new_client_socket, ip_port))
    # 设置守护线程
    thread_recvmsg.setDaemon(True)
    # 启动线程
    thread_recvmsg.start()