python网络编程之多任务的TCP服务端开发

需求

同时多个客户端与服务器通信

思路

• 对原单任务的服务端改进的地方
  • 加入多线程
  • 从接收到数据开始建立连接循环建立多线程
  • 对于数据的接收和发送处理独立出来作为一个函数
  • 对于独立出的函数创建子线程并启动

原单任务代码示例

import socket

if __name__ == '__main__':
    # 1. 创建服务端套接字
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 解决关闭服务端后需要1到2分钟才能释放端口号，立即释放端口号
    server_socket.setsockopt(65535, 4, True)

    # 2. 绑定IP和端口
    # 绑定元祖参数为IP地址和端口号，IP地址一般不绑定，表示该服务器的所有IP
    server_socket.bind(('', 9090))

    # 3. 监听服务请求
    # 参数表示最大等待连接数量
    server_socket.listen(128)

    # 4. 接受客户端的请求,生成一个元祖包含：
    # 生成新的套接字专门和该请求连接服务
    # 客户端的IP 和 端口
    # 在没有接受到服务请求时会一直停留在此处
    new_server_socket, ip_port = server_socket.accept()
    print('新的服务端套接字为：', new_server_socket)
    print('客户端的IP和端口：', ip_port)

    # 5. 接受报文
    recv_data = new_server_socket.recv(1024)
    recv_content = recv_data.decode('utf-8')
    print('接受到消息为：', recv_content)

    # 6. 发送报文
    send_data = '哈哈'
    send_content = send_data.encode('utf-8')
    new_server_socket.send(send_content)

    # 7. 关闭专门服务该连接的新套接字
    new_server_socket.close()

    # 8. 关闭客户端的套接字
    server_socket.close()

多任务的代码示例

import socket
import threading

# 定义函数：对新的套接字接受和发送消息，并且循环直到客户端停止时关闭与该套接字连接
def new_server(new_socket, ip_port):
    while True:
        # 接受数据
        recv_data = new_socket.recv(1024)
        if recv_data:
            recv_content = recv_data.decode('utf-8')
            print(f'收到来自 {ip_port} 的消息：{recv_content}')

            # 发送数据
            send_content = '哈哈'
            send_data = send_content.encode('utf-8')
            new_socket.send(send_data)
        else:
            print(f'{ip_port}已关闭！！！')
            break
    new_socket.close()

if __name__ == '__main__':

    # 创建服务端套接字并端口复用
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)

    # 绑定IP和端口
    server_socket.bind(('',9999))

    # 监听消息
    server_socket.listen(128)

    # 设置多线程
    while True:

        # 生成新的套接字，一直阻塞直到接收发送的请求
        new_socket, ip_port = server_socket.accept()
        print(f'与客户端{ip_port}建立连接成功！！！')

        # 创建子线程对象并设置为守护主线程
        sub_thread = threading.Thread(target=new_server, args=(new_socket, ip_port), daemon=True)
        # 启动子线程
        sub_thread.start()

    # 关闭被动套接字
    server_socket.close()