Python-Socket练习（TCP回显服务器）

1. 练习Python套接字

创建TCP服务器流程

创建UDP服务器流程

跟上面的流程类似，这里直接练习Python套接字语法

TCP Sever

TCP服务器代码：

import socket
import threading
from concurrent.futures import ThreadPoolExecutor


# 处理客户端请求
def work(_sock, _port):
    print(f'分配给客户端sock为{_sock},端口号为{_port}')

    # 循环接受客户端数据
    while True:
        recvData = sock.recv(1024)
        if recvData:
            print('get client:', recvData.decode('gbk'), f'数据长度为{len(recvData)}')
            # 向客户端发送数据
            Str = "服务器已收到TCP客户端发送的消息"
            sock.send(Str.encode('gbk'))
        else:
            print(f"{_port}客户端下线")
            break
        # 关闭套接字，释放资源
    sock.close()


if __name__ == '__main__':

    # 创建套接字
    tcpSever = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 设置端口复用
    # socket.SOL_SOCKET 代表当前套接字
    # socket.SO_REUSEADDR 代表复用端口号选项
    # True 确定开启套接字复用
    tcpSever.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)

    # 绑定端口号
    tcpSever.bind(("", 8091))

    # 监听
    MAX_CONNECT = 128  # 最大监听数目
    tcpSever.listen(MAX_CONNECT)
    print("INFO:Sever Start!")

    # 接受客户端连接请求
    pool = ThreadPoolExecutor(max_workers=10)
    while True:
        # 接受客户端连接请求
        sock, port = tcpSever.accept()
        pthread = threading.Thread(target=work, args=(sock, port))
        # 设置守护线程，主线程退出之后子线程直接销毁
        pthread.setDaemon(True)
        pthread.start()

测试服务器的客户端代码

import socket

# 创建套接字
tcpClient = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # IPV4 TCP

# 与服务器建立链接
severIp = input('输入服务器ip地址')
severPort = int(input('输入服务器端口'))  # 8091
tcpClient.connect((severIp, severPort))

# 向服务器发送数据
Str = "你好TCP服务器,测试TCP客户端"
tcpClient.send(Str.encode('gbk'))

# 接受服务器响应
recvData = tcpClient.recv(1024)
# windows中文编码格式改成gbk，否则会出现乱码
print('get sever:', recvData.decode("gbk"))

# 关闭套接字
tcpClient.close()

练习结果

需要注意，如果没有设置端口复用，同一个端口不能再短时间内重启服务器。因为TCP断开连接时四次挥手，TIME_WAIT状态时重启服务器时会绑定失败，因为连接没有被完全释放。

相关资料