版权声明:欢迎交流,转载请注明出处。 https://blog.csdn.net/u013034226/article/details/82290498
我在上一篇博客写到了创建tcp文件下载器,因为是单任务,所以在本篇博客加入学习的线程有关知识,实现多任务文件下载器的服务端和客户端,博客地址。
目录
一、发现原因
服务端如果有客户端连接时,新的客户端不能接入,原因是服务端套接字在处理客户端的文件下载请求时,会阻塞主线程,导致不能连接其它客户端。
二、解决思路
将服务端处理客户端文件下载请求的这段代码拿出来,放到一个子线程中,来一个客户端就创建一个子线程,这样就不会阻塞主线程中的与客户端建立连接的功能,实现多任务文件下载。
三、具体实现方法
服务端:
import socket
import os
import threading
# 处理客户端请求下载文件的操作(从主线程提出来的代码)
def deal_client_request(ip_port, service_client_socket):
# 连接成功后,输出“客户端连接成功”和客户端的ip和端口
print("客户端连接成功", ip_port)
# 我将查找位置进行了更改,改为到我的桌面去查找文件
os.chdir("C:/Users/用户名/Desktop")
# 接收客户端的请求信息
file_name = service_client_socket.recv(1024)
# 解码
file_name_data = file_name.decode("utf-8")
# 判断文件是否存在
if os.path.exists(file_name_data):
# 打开文件
with open(file_name_data, "rb") as f:
while True:
file_data = f.read(1024)
if file_data:
# 只要读取到数据,就向客户端进行发送
service_client_socket.send(file_data)
# 数据读完,退出循环
else:
print("请求的文件数据发送完成")
break
else:
print("下载的文件不存在")
# 关闭服务当前客户端的套接字
service_client_socket.close()
if __name__ == '__main__':
# 创建套接字
tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定端口号
tcp_server_socket.bind(("", 3356))
# 设置监听,将主动套接字变为被动套接字
tcp_server_socket.listen(128)
# 循环调用accept,可以支持多个客户端同时连接,和多个客户端同时下载文件
while True:
service_client_socket, ip_port = tcp_server_socket.accept()
# 连接成功后打印套接字号
print(id(service_client_socket))
# 创建子线程
sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket))
# 启动子线程
sub_thread.start()
客户端:
# 多任务文件下载器客户端
import socket
if __name__ == '__main__':
# 创建套接字
tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 和服务端连接
tcp_client_socket.connect(("服务端ip地址", 3356))
# 发送下载文件的请求
file_name = input("请输入要下载的文件名:")
# 编码
file_name_data = file_name.encode("utf-8")
# 发送文件下载请求数据
tcp_client_socket.send(file_name_data)
# 把数据写入到文件里
with open("我的桌面地址,在这里不要忘了最后的/" + file_name, "wb") as file:
while True:
# 循环接收文件数据
file_data = tcp_client_socket.recv(1024)
# 接收到数据
if file_data:
# 写入数据
file.write(file_data)
# 接收完成
else:
print("服务端把数据发送完成并关闭了连接")
break
# 在这里,客户端的套接字是要关闭的,服务端套接字因为要服务多个客户端所以不用关闭
tcp_client_socket.close()
四、传输文件、音频 、图片、视频
注意:
在测试时文件存储位置的格式不要写错,客户端的文件存储位置最后的“/”不能少,不然会变成一个新文件:Desktop+文件名,放在桌面的上一层目录中。