Rust Pattern - 写文件

最新推荐文章于 2025-02-03 07:00:00 发布

转载最新推荐文章于 2025-02-03 07:00:00 发布 · 359 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://my.oschina.net/kuerant/blog/1501381

文章标签：

#rust #python

本文介绍了使用Rust语言进行文件输出的多种方法，包括简单直接的写法、利用缓冲提高效率的方法，以及如何将输出定向至标准输出。此外，还探讨了在不同情况下选择合适方法的建议。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2019独角兽企业重金招聘Python工程师标准>>>

简单的写法

#[allow(dead_code)]
fn output(filename: &str, bytes: &[u8]) -> Result<()> {
    let mut fp = OpenOptions::new().truncate(true).create(true).write(true).open(Path::new(filename)).chain_err(|| format!("fail to open {}", filename))?;
    fp.write_all( bytes )?;
    fp.write_all( &['\n' as u8] )?;
    fp.flush()?;

    Ok(())
}

或者，如下写法更有 Rust 的感觉

#[allow(dead_code)]
fn output(filename: &str, bytes: &[u8]) -> Result<()> {
    OpenOptions::new().truncate(true).create(true).write(true).open(Path::new(filename)).and_then(|mut fp| {
        fp.write_all( bytes )?;
        fp.write_all( &['\n' as u8] )?;
        fp.flush()?;
        
        Ok(())
    })
}

引入 BufWriter，增加写缓冲

#[allow(dead_code)]
fn output(filename: &str, bytes: &[u8]) -> Result<()> {
    let fp = OpenOptions::new().truncate(true).create(true).write(true).open(Path::new(filename)).chain_err(|| format!("fail to open {}", filename))?;
    let mut writer = BufWriter::with_capacity( 1024*1024*4, fp );
    writer.write_all( bytes )?;
    writer.write_all( &['\n' as u8] )?;
    writer.flush()?;

    Ok(())
}

当 filename == '-' 时，写到标准输出（STDOUT）

#[allow(dead_code)]
fn output(filename: &str, bytes: &[u8]) -> Result<()> {
    let fp = match filename {
        "-" => Box::new(stdout()) as Box<Write>,
        filename => {
            let path = Path::new(filename);
            //let fp = OpenOptions::new().append(true).create(true).write(true).open(path).unwrap();
            let fp = OpenOptions::new().truncate(true).create(true).write(true).open(path).unwrap();
            Box::new(fp) as Box<Write>
        },
    };

    let mut writer = BufWriter::with_capacity( 1024*1024*4, fp );
    writer.write_all( bytes )?;
    writer.write_all( &['\n' as u8] )?;
    writer.flush()?;

    Ok(())
}

转载于:https://my.oschina.net/kuerant/blog/1501381