Rust函数返回局部值引用报错：无法定位引用位置

问题描述 #

编写Rust函数encode_initial_configs时触发E0515错误，编译器提示返回值引用了局部临时值，但无法定位引用来源。

函数代码 #

fn encode_initial_configs<'a, TSym, NTSym>(
    alpha: impl IntoIterator<Item = TSym> + 'a,
    grammar: &'a OPGrammar<TSym, NTSym>,
    chunk_size: usize,
) -> impl Iterator<Item = GPUParseConfig> + 'a
where
    TSym: Eq + std::hash::Hash + Clone,
    NTSym: Eq + Clone,
{
    assert_ne!(chunk_size, 0, "`chunk_size` must be non-zero.");

    let mut alpha_chunks = grammar
        .encode_iterator(alpha)
        .chunks(chunk_size)
        .into_iter()
        .map(|chunk| chunk.collect_vec())
        .fuse();
    let curr_chunk = alpha_chunks.next();
    InitialConfigs {
        alpha_chunks,
        last_sym: 0,
        curr_chunk,
    }
}

编译器报错 #

error[E0515]: cannot return value referencing temporary value
  --> src/par_parse.rs:77:5
   |
70 |       let mut alpha_chunks = grammar
   |  ____________________________-
71 | |         .encode_iterator(alpha)
72 | |         .chunks(chunk_size)
   | |___________________________- temporary value created here
...
77 | /     InitialConfigs {
78 | |         alpha_chunks,
79 | |         last_sym: 0,
80 | |         curr_chunk,
81 | |     }
   | |_____^ returns a value referencing data owned by the current function
   |
   = help: use `.collect()` to allocate the iterator

For more information about this error, try `rustc --explain E0515`.

补充：encode_iterator方法代码 #

pub fn encode_iterator<'a>(
    &'a self,
    s: impl IntoIterator<Item = TSym> + 'a,
) -> impl Iterator<Item = u32> + 'a {
    s.into_iter().map(|sym| self.encode_term(sym))
}

问题原因 #

错误的核心在于你使用的chunks迭代器适配器（推测是itertools库提供的）：

• 它返回的Chunks迭代器及其产生的Chunk元素，本质上依赖于对上游迭代器的引用，而上游迭代器是链式调用中创建的临时值。
• 虽然你通过map(|chunk| chunk.collect_vec())将每个块转换为Vec，但整个迭代器链仍然持有对临时值的引用。当你把alpha_chunks放入InitialConfigs返回时，就会触发E0515——返回值引用了函数内部的临时数据，这些数据会在函数结束时被销毁，导致悬垂引用。

解决方法 #

要修复这个问题，需要让迭代器持有数据的所有权而非引用。最直接的方式是先将所有块完整收集到堆上，再创建持有所有权的迭代器：

修改后的代码 #

fn encode_initial_configs<'a, TSym, NTSym>(
    alpha: impl IntoIterator<Item = TSym> + 'a,
    grammar: &'a OPGrammar<TSym, NTSym>,
    chunk_size: usize,
) -> impl Iterator<Item = GPUParseConfig> + 'a
where
    TSym: Eq + std::hash::Hash + Clone,
    NTSym: Eq + Clone,
{
    assert_ne!(chunk_size, 0, "`chunk_size` must be non-zero.");

    // 先将所有块收集到Vec中，确保所有权被持有
    let chunks: Vec<Vec<u32>> = grammar
        .encode_iterator(alpha)
        .chunks(chunk_size)
        .into_iter()
        .map(|chunk| chunk.collect_vec())
        .collect();
    
    let mut alpha_chunks = chunks.into_iter().fuse();
    let curr_chunk = alpha_chunks.next();
    InitialConfigs {
        alpha_chunks,
        last_sym: 0,
        curr_chunk,
    }
}

代码说明 #

1. .collect()的作用：遍历整个迭代器链，将所有块转换为Vec<Vec<u32>>，此时所有数据都被分配在堆上，所有权完全属于chunks变量。
2. into_iter()创建所有权迭代器：调用chunks.into_iter()会生成一个直接持有Vec<Vec<u32>>所有权的迭代器，不再依赖任何临时值或引用。
3. 消除悬垂引用：alpha_chunks和curr_chunk都不再引用函数内部的临时数据，完全符合生命周期约束，E0515错误会被修复。

如果需要处理超大数据流、不想提前遍历整个迭代器，可以手动实现分块逻辑，确保每一块的所有权被独立持有，避免引用上游临时迭代器。

内容的提问来源于stack exchange，提问作者Michele De Pascalis