1. 需求
目前我们开发的TCP服务端程序只能服务于一个客户端,如何开发一个多任务版的TCP服务端程序能够服务于多个客户端呢?
完成多任务,可以使用线程,比进程更加节省内存资源。
2. 具体实现步骤
- 编写一个TCP服务端程序,循环等待接受客户端的连接请求
- 当客户端和服务端建立连接成功,创建子线程,使用子线程专门处理客户端的请求,防止主线程阻塞
- 把创建的子线程设置成为守护主线程,防止主线程无法退出。
3. 实现代码:
# 导入socket模块,用于实现网络通信功能
import socket
# 导入threading模块,用于创建和管理线程
import threading
# 定义处理客户端请求的函数
def handle_client_request(service_client_socket, ip_port):
# 无限循环,持续接收客户端消息
while True:
# 接收来自客户端的数据,最大接收1024字节
recv_data = service_client_socket.recv(1024)
# 如果接收到的数据不为空
if recv_data:
# 打印客户端发送的消息,格式为“客户端IP:端口说:消息内容”
print("客户端%s:%s说:%s" % (ip_port[0], ip_port[1], recv_data.decode()))
# 向客户端发送响应消息“ok,问题正在处理中...”,并将其编码为utf-8格式
service_client_socket.send("ok,问题正在处理中...".encode('utf-8'))
else:
# 如果没有接收到数据,说明客户端已断开连接
print("客户端%s:%s已经退出聊天室" % (ip_port[0], ip_port[1]))
# 跳出循环
break
# 关闭与客户端的连接
service_client_socket.close()
# 如果此脚本作为主程序运行
if __name__ == '__main__':
# 创建一个TCP/IP套接字对象
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定服务器地址和端口,此处使用的是本地地址和8080端口
server_socket.bind(("", 8080))
# 设置监听队列长度为128
server_socket.listen(128)
# 无限循环,等待客户端连接
while True:
# 接受客户端连接,返回一个新的套接字对象和客户端地址信息
service_client_socket, ip_port = server_socket.accept()
# 打印客户端连接信息,格式为“客户端IP:端口已经进入聊天室”
print("客户端%s:%s已经进入聊天室" % (ip_port[0], ip_port[1]))
# 创建一个新线程来处理客户端请求
sub_thread = threading.Thread(target=handle_client_request, args=(service_client_socket, ip_port))
# 设置线程为守护线程,主线程结束时子线程也会被终止
sub_thread.daemon = True
# 启动线程
sub_thread.start()