我注意到使用 paramiko 的 sftp 执行 get 或 put 时无法获得相同的传输速度。
在我们的 mac mini 服务器(运行 Mac os 10.12.6)的千兆网络中,通过 rsync/sftp/scp/finder 的文件传输速度维持在 95-100MB/秒左右。如果我使用 paramiko 的 sftp.get,我发现它的最大速度达到 25MB/秒。
我使用的是 paramiko 1.17 并更新到 2.3.1,但速度几乎相同。
有什么想法可能导致限制吗?
谢谢!
亚当
我遇到了同样的问题并实施了其他人提出的一些建议。可以做三件事:
增加传输中的缓冲区大小。
transport = paramiko.Transport(ftp_host, ftp_port)
transport.default_window_size = 4294967294 # 2147483647
transport.packetizer.REKEY_BYTES = pow(2, 40)
transport.packetizer.REKEY_PACKETS = pow(2, 40)
Run Code Online (Sandbox Code Playgroud)
在获取文件之前执行预读。
ftp_file = ftp_conn.file(file_name, "r")
ftp_file_size = ftp_file.stat().st_size
ftp_file.prefetch(ftp_file_size)
ftp_file.set_pipelined()
ftp_file_data = ftp_file.read(ftp_file_size)
Run Code Online (Sandbox Code Playgroud)
传输较大文件时可以做的另一件事是实施“块”。这会将文件分割成单独传输的较小部分。我仅通过传输到 s3 对此进行了测试。
chunk_size = 6000000 #6 MB
chunk_count = int(math.ceil(ftp_file_size / float(chunk_size)))
multipart_upload = s3_conn.create_multipart_upload(Bucket=bucket_name, Key=s3_key_val)
parts = []
for i in range(chunk_count):
print("Transferring chunk {}...".format(i + 1), "of ", chunk_count)
start_time = time.time()
ftp_file.prefetch(chunk_size * (i+1) # This statement is where the magic was to keep speeds high.
chunk = ftp_file.read(int(chunk_size))
part = s3_conn.upload_part(
Bucket=bucket_name,
Key=s3_file_path,
PartNumber=part_number,
UploadId=multipart_upload["UploadId"],
Body=chunk
)
end_time = time.time()
total_seconds = end_time - start_time
print("speed is {} kb/s total seconds taken {}".format(math.ceil((int(chunk_size) / 1024) / total_seconds), total_seconds))
part_output = {"PartNumber": i, "ETag": part["ETag"]}
parts.append(part)
print("Chunk {} Transferred Successfully!".format(i + 1))
part_info = {"Parts": parts}
s3_conn.complete_multipart_upload(
Bucket=bucket_name,
Key=s3_key_val,
UploadId=multipart_upload["UploadId"],
MultipartUpload=part_info
)
Run Code Online (Sandbox Code Playgroud)
处理 chunk 时的重要部分是 ftp_file.prefetch(chunk_size * (i+1)),它在每个循环中进一步增量读取。
实施所有这些后,下载速度从 200 kBps 提高到 5 MBps(最大隧道速度)。
在这段代码的后续迭代中,我遇到了来自 paramiko 的垃圾收集问题。我通过删除该行解决了它们:
ftp_file.set_pipelined()
Run Code Online (Sandbox Code Playgroud)