聊天室系统:多任务版TCP服务端程序开发详细代码解释
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))# 设置监听队列长度为128server_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()