Rust中如何将split()结果传递至main()？CSV读取编译报错解决

问题场景 #

用静态数组模拟CSV内容时编译正常，但替换为真实文件读取逻辑后，触发E0597生命周期错误。

原代码 #

use std::collections::BTreeMap;
use std::error::Error;
use std::fs::File;
use std::io::{BufRead, BufReader};

fn read_csv(guess_number: &mut BTreeMap<&str, &str>) -> Result<(), Box<dyn Error>> {
    // let lines = ["Tom,4", "John,6"];
    // for line in lines.iter() {
    let file = File::open("sample.csv")?;
    let reader = BufReader::new(file);
    for (_idx, line) in reader.lines().enumerate() {
        if let Ok(line) = line {
            let split = line.split(",");
            let vec: Vec<_> = split.collect();
            println!("{} {:?}", line, vec);
            //let number: i32 = vec[1].trim().parse().unwrap();
            // guess_number.insert(vec[0], number);
            guess_number.insert(vec[0], vec[1]);
        }
    }
    Ok(())
}
fn main() {
    let mut guess_number: BTreeMap<&str, &str> = BTreeMap::new();
    read_csv(&mut guess_number);
    for (k, v) in guess_number {
        println!("{} {:?}", k, v);
    }
}

编译错误信息 #

Compiling sample v0.1.0 (/home/hosthome/prj/rs/sample)
error[E0597]: `line` 生命周期不足
  --> src/main.rs:13:25
   |
6  | fn read_csv(guess_number: &mut BTreeMap<&str, &str>) -> Result<(), Box<dyn Error>> {
   |                                               - 我们称这个引用的生命周期为 `'1`
...
13 |             let split = line.split(",");
   |                         ^^^^^^^^^^^^^^^ 被借用的值生命周期不足
...
18 |             guess_number.insert(vec[0], vec[1]);
   |             ----------------------------------- 参数要求 `line` 必须被借用至生命周期 `'1`
19 |         }
   |         - `line` 在此处被销毁，但仍处于被借用状态

For more information about this error, try `rustc --explain E0597`.
error: could not compile `sample` due to previous error

错误原因 #

静态数组中的字符串是静态生命周期（'static），可以被BTreeMap长期持有引用；但文件读取的line是循环内的局部变量，生命周期仅存在于当前迭代，当循环结束时line会被销毁，此时BTreeMap中存储的引用会变成悬垂引用，Rust的所有权机制禁止这种不安全行为。

解决方法 #

将BTreeMap的类型从BTreeMap<&str, &str>改为BTreeMap<String, String>，通过拥有所有权的方式存储字符串，避免引用生命周期问题：

修改后的代码 #

use std::collections::BTreeMap;
use std::error::Error;
use std::fs::File;
use std::io::{BufRead, BufReader};

fn read_csv(guess_number: &mut BTreeMap<String, String>) -> Result<(), Box<dyn Error>> {
    let file = File::open("sample.csv")?;
    let reader = BufReader::new(file);
    for (_idx, line) in reader.lines().enumerate() {
        if let Ok(line) = line {
            // 用split_once更安全，避免索引越界
            if let Some((name, num_str)) = line.split_once(",") {
                guess_number.insert(name.trim().to_string(), num_str.trim().to_string());
            }
        }
    }
    Ok(())
}
fn main() {
    let mut guess_number: BTreeMap<String, String> = BTreeMap::new();
    if let Err(e) = read_csv(&mut guess_number) {
        eprintln!("读取文件失败: {}", e);
        return;
    }
    for (k, v) in guess_number {
        println!("{}: {}", k, v);
    }
}

关键修改点 #

1. 调整函数参数和main中BTreeMap的类型为BTreeMap<String, String>
2. 使用split_once替代split+collect，避免索引越界风险，同时更高效
3. 调用trim()去除字符串前后空白，再用to_string()转换为拥有所有权的String
4. 增加错误处理，在main中捕获read_csv返回的错误并打印

内容的提问来源于stack exchange，提问作者Daniel YC Lin