Rustにてフォルダ内のファイル数を事前に把握してからfor inを回したい

Question

実現したいこと

ここに実現したいことを箇条書きで書いてください。

• フォルダ内の特定の拡張子ファイルに対して処理を行う
• どのくらいのファイル数があり、どのくらいの数を処理したのか表示をしたい

前提

勉強を兼ねてRustでCLIツールを作っているのでRustで作りたいです。
表示に関してindicatifクレートを使う予定です
フォルダ内のファイル捜査に関してはread_dirを使っています。

発生している問題・エラーメッセージ

error[E0382]: use of moved value: `files`
   --> src\main.rs:35:17

ファイル数を把握する時点で所有権が移ってしまっています。
所有権を移さずに個数を把握することはできないのでしょうか？

該当のソースコード

fn main() -> Result<(), Box<dyn Error>> {
    println!("ファイルをまとめます");
    // エクセルファイルが格納されているフォルダのパス
    let folder_path = r"\example";
    let files = read_dir(folder_path)?;

    // 出力先のパス
    let output_path = r"\example.csv";
    // csv::WriterBuilderに渡す前にBOMを書き込んでおく。

    println!("保存フォルダ：{}",folder_path);
    println!("ファイル作成中...：{}",output_path);

    let output = File::create(output_path)?;
    let mut wtr = BufWriter::new(output);
    wtr.write_all(BOM)?;

    let mut wtr = WriterBuilder::new()
        .terminator(Terminator::CRLF)
        .from_writer(wtr);

    let max_count = files.cloned();

    let mut is_head_writed = false;
    // フォルダ内の各Excelファイルに対して処理を実行
    for file in files {
        let file_path = file?.path();
        if let Some(extension) = file_path.extension() {
            // 拡張子がxlsxのファイルのみ処理する
            if extension == "xlsx" {
                // エクセルファイルを開く
                let mut workbook: Xlsx<_> = open_workbook(&file_path)?;
                workbook
                    .load_tables()
                    .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                
                if let Some(Ok(table)) = workbook.table_by_name("exampleTable"){
                    if  false == is_head_writed {
                        wtr
                            .write_record(table.columns())
                            .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                        is_head_writed = true;
                    }
                    for row in table.data().rows(){
                        if "" != row[1].to_string() {
                            wtr
                                .write_record(row.iter().map(|f|f.to_string()).collect::<Vec<_>>())
                                .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                        }else {
                            // データが空なのでskip
                        }
                    }
                }
            }
        }
    }
    wtr.flush()?;
    println!("作成完了！");
    Ok(())
}

試したこと

以下のようにしたら実現するのですが、
カウント時とファイル処理時の2回、フォルダ捜査と絞り込みを行っているのが気に食わないです。

fn main() -> Result<(), Box<dyn Error>> {
     println!("ファイルをまとめます");
    // エクセルファイルが格納されているフォルダのパス
    let folder_path = r"\example";

    // 出力先のパス
    let output_path = r"\example.csv";
    // csv::WriterBuilderに渡す前にBOMを書き込んでおく。

    println!("保存フォルダ：{}",folder_path);
    println!("ファイル作成中...：{}",output_path);

    let output = File::create(output_path)?;
    let mut wtr = BufWriter::new(output);
    wtr.write_all(BOM)?;

    let mut wtr = WriterBuilder::new()
        .terminator(Terminator::CRLF)
        .from_writer(wtr);
    
    let file_count: u64 = read_dir(folder_path)?
        .filter_map(|entry| {
            let path = entry.ok()?.path();
            if path.extension().map_or(false, |ext| ext == "xlsx"){
                Some(path)
            }else {
                None
            }
        })
        .count() as u64;
    let pb = ProgressBar::new(file_count);

    let mut is_head_writed = false;

    // フォルダ内の各Excelファイルに対して処理を実行
    for file in read_dir(folder_path)? {
        let file_path = file?.path();
        if let Some(extension) = file_path.extension() {
            // 拡張子がxlsxのファイルのみ処理する
            if extension == "xlsx" {
                // エクセルファイルを開く
                let mut workbook: Xlsx<_> = open_workbook(&file_path)?;
                workbook
                    .load_tables()
                    .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                
                if let Some(Ok(table)) = workbook.table_by_name("exampleTable"){
                    if  false == is_head_writed {
                        wtr
                            .write_record(table.columns())
                            .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                        is_head_writed = true;
                    }
                    for row in table.data().rows(){
                        if "" != row[1].to_string() {
                            wtr
                                .write_record(row.iter().map(|f|f.to_string()).collect::<Vec<_>>())
                                .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                        }else {
                            // データが空なのでskip
                        }
                    }
                }
            }
        }
        thread::sleep(Duration::from_millis(5));
        pb.inc(1);
    }
    wtr.flush()?;
    pb.finish_with_message("done");
    println!("作成完了！");
    Ok(())
}

補足情報（FW/ツールのバージョンなど）

[dependencies]
calamine = "0.20.0"
csv = "1.2"
indicatif = "0.15.0"

Comments

[hoshi-takanori]

Rust よく分かりませんが、read_dir の結果は Iterator っぽいので、個数カウントを含め、一度読み込んだら消費済みになるのかも。配列か何かに読み込んで処理すれば良いのでは？

[watya1]

コメントありがとうございます。
ひとまず試してみましたがいけそうです。
基本的にイテレータは1回こっきりで消費されるみたいですね。
覚えておきます。

Answer 1

以下のコメントをいただき、試したところうまくいきました

read_dir の結果は Iterator っぽいので、
個数カウントを含め一度読み込んだら消費済みになるのかも。
配列か何かに読み込んで処理すれば良いのでは？

fn main() -> Result<(), Box<dyn Error>> {
    println!("棚卸し原表をまとめます");
    // エクセルファイルが格納されているフォルダのパス
    let folder_path = r"\example"

    // 出力先のパス
    let output_path = r"\example.csv"
    // csv::WriterBuilderに渡す前にBOMを書き込んでおく。

    println!("保存フォルダ：{}",folder_path);
    println!("ファイル作成中...：{}",output_path);

    let output = File::create(output_path)?;
    let mut wtr = BufWriter::new(output);
    wtr.write_all(BOM)?;

    let mut wtr = WriterBuilder::new()
        .terminator(Terminator::CRLF)
        .from_writer(wtr);
    
// -----------------------↓↓ここでイテレータ取得/xlsxが含まれるファイル走査/配列にmove/を行う
    let files_path = read_dir(folder_path)?
        .filter_map(|entry| {
            let path = entry.ok()?.path();
            // 拡張子がxlsxのファイルのみを対象とする。
            if path.extension().map_or(false, |ext| ext == "xlsx"){
                Some(path)
            }else {
                None
            }
        })
        .collect::<Vec<_>>();
// ----------------------------------------------------------------------------------------------------------
    
    let pb = ProgressBar::new(files_path.len() as u64); // ← moveされない！
    let mut is_first_file = true;

    // フォルダ内の各Excelファイルに対して処理を実行
    for file_path in files_path {
// -----------------------↓↓files_pathはxlsxのpathのみになっているので条件分岐も必要なくなった。moveもされていない。
        // ファイル処理
        let mut workbook: Xlsx<_> = open_workbook(&file_path)?;
        workbook
            .load_tables()
            .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
        
        if let Some(Ok(table)) = workbook.table_by_name("T_Data"){
            // 最初のファイルだけヘッダを取り込む
            if  is_first_file {
                wtr
                    .write_record(table.columns())
                    .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                is_first_file = false;
            }
            for row in table.data().rows(){
                if "" != row[1].to_string() {
                    wtr
                        .write_record(row.iter().map(|f|f.to_string()).collect::<Vec<_>>())
                        .unwrap_or_else(|err| eprintln!("IO Error -> {}",err));
                }else {
                    // データが空なのでskip
                }
            }
        }
        thread::sleep(Duration::from_millis(5));
        pb.inc(1);
// ----------------------------------------------------------------------------------------------------------
    }
    wtr.flush()?;
    pb.finish_with_message("done");
    println!("作成完了！");
    Ok(())
}