0

我正在尝试使用ssh2crate通过 SCP 下载一个大型二进制文件v0.3.3。

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()

4

1 回答 1

3

由于Streamimplements Read，您应该能够只使用std::io::copy(&mut stream, &mut File::create(...).unwrap())将所有数据复制到文件中。如果你想要一个进度条，我建议使用indicatifcrate，它有一个包装Read实例的方法，在读取数据时自动更新进度条。

于 2019-07-21T14:55:58.263 回答