我正在尝试使用ssh2
crate v0.3.3
通过SCP下载一个大的二进制文件。
extern crate ssh2;
use std::net::TcpStream;
use ssh2::Session;
use std::path::Path;
use std::fs::File;
use indicatif::ProgressBar;
fn main() -> Result<(), Box<std::error::Error>> {
let ssh_host_port = "prokerala.com:22";
let ssh_user = "prokeral";
let remote_temp_file = "/tmp/dbimport_Rk6Iwwm5.sql.bz2";
let tcp = TcpStream::connect(&ssh_host_port).unwrap();
let mut sess = Session::new().unwrap();
sess.handshake(&tcp).expect("SSH handshake failed");
let _ = sess.userauth_agent(ssh_user);
let path = Path::new(&remote_temp_file);
let (mut remote_file, stat) = sess.scp_recv(path).unwrap();
let stream:ssh2::Stream = remote_file.stream(1);
// Update: solved by using io::copy as suggested by @apemanzilla below
// let mut target = File::create("/tmp/done.txt").unwrap();
// let pb = ProgressBar::new(stat.size());
// std::io::copy(&mut pb.wrap_read(remote_file), &mut target)?;
Ok(())
}
我想将stream
写到一个文件中,并显示一个进度条。是否可以直接将stream
写入文件,而无需在循环中使用stream.read()
将数据读取到内存中?
发布于 2019-07-21 22:55:58
由于Stream
实现了Read
,因此您应该能够只使用std::io::copy(&mut stream, &mut File::create(...).unwrap())
将所有数据复制到一个文件中。如果你想要一个进度条,我推荐使用indicatif
机箱,它有一个method to wrap a Read
instance,可以在读取数据时自动更新进度条。
https://stackoverflow.com/questions/57134077
复制相似问题