当rust中不存在文件或目录时，std：：FS：：File：：Create().unwire()会抛出错误

发布于02月09日

pub fn create_batched_writer(schema: &polars::prelude::Schema) -> BatchedWriter<File> {
    let mut file = std::fs::File::create("output/test.parquet").unwrap();
    let writer = ParquetWriter::new(file);
    writer.batched(schema).unwrap()
}

当文件或目录不存在时，它会死机，如下所示.

called `Result::unwrap()` on an `Err` value: Os { code: 2, kind: NotFound, message: "No such file or directory" }

我的问题是，当发现文件和目录不存在时，我应该如何创建

推荐答案

我假设发生这个错误是因为目录"output"不存在，在此基础上，您可以try 以下代码:

pub fn create_batched_writer(schema: &Schema) -> BatchedWriter<File> {
    let file_path = "output/test.parquet";
    
    // Checking if the directory exists and creating if not
    let parent_dir = Path::new(file_path).parent().unwrap();
    if !parent_dir.exists() {
        fs::create_dir_all(parent_dir).expect("Failed to create directory");
    }

    let file = File::create(file_path);
    let writer = ParquetWriter::new(file);
    writer.batched(schema).unwrap()
}