循环中的Rust生命周期问题

时间:2017-05-22 04:07:16

标签: rust borrow-checker

如何在不进行数组复制或每次迭代多次调用b()的情况下编译此示例 - b()必须执行一些昂贵的解析?

这不是我写的完整代码,但它说明了我遇到的问题。在这里,Test正在尝试执行某种流式解析工作。 c()是解析函数,解析成功时返回Someb()是一个函数,当c()无法使用可用数据解析时,它会尝试从流中读取更多数据。返回的值是包含已解析范围的self.v的切片。

struct Test {
    v: [u8; 10],
    index: u8,
}

impl Test {
    fn b(&mut self) {
        self.index = 1
    }

    fn c(i: &[u8]) -> Option<&[u8]> {
        Some(i)
    }

    fn a(&mut self) -> &[u8] {
        loop {
            self.b();

            match Test::c(&self.v) {
                Some(r) => return r,
                _ => continue,
            }
        }
    }
}

fn main() {
    let mut q = Test {
        v: [0; 10],
        index: 0,
    };

    q.a();
}

编译时,会产生以下借用检查错误:

error[E0502]: cannot borrow `*self` as mutable because `self.v` is also 
borrowed as immutable
  --> <anon>:17:13
   |
17 |             self.b();
   |             ^^^^ mutable borrow occurs here
18 | 
19 |             match Test::c(&self.v) {
   |                            ------ immutable borrow occurs here
...
24 |     }
   |     - immutable borrow ends here

如果我将a()更改为:

fn a(&mut self) -> Option<&[u8]> {
    loop {
        self.b();

        if let None = Test::c(&self.v) {
            continue
        }

        if let Some(r) = Test::c(&self.v) {
            return Some(r);
        } else {
            unreachable!();
        }
    }
}

然后它运行,但显然有两次调用解析函数c()的缺点。

我有点明白,在返回值取决于它时更改self是不安全的,但是,我不明白为什么self.v的不可变借用在下一次迭代中仍然存在,我们试图再次致电b()

1 个答案:

答案 0 :(得分:4)

现在&#34; Rustc不能&#34;交易&#34;有条件借款退货&#34;。请参阅此comment from Gankro on issue 21906

如果只有一个执行路径终止循环,它就不能为借用分配正确的生命周期。

我可以建议这种解决方法,但我不确定它是否是最佳的:

fn c(i: &[u8]) -> Option<(usize, usize)> {
    Some((0, i.len()))
}

fn a(&mut self) -> &[u8] {
    let parse_result;
    loop {
        self.b();

        match Test::c(&self.v) {
            Some(r) => {
                parse_result = r;
                break;
            }
            _ => {}
        }
    }
    let (start, end) = parse_result;
    &self.v[start..end]
}

您可以使用数组索引构造解析结果,并将它们转换为循环外部的引用。

另一种选择是诉诸unsafe来解耦生命。我不是安全使用不安全的专家,所以要注意别人的评论。

fn a(&mut self) -> &[u8] {
    loop {
        self.b();

        match Test::c(&self.v) {
            Some(r) => return unsafe{ 
                // should be safe. It decouples lifetime of 
                // &self.v and lifetime of returned value,
                // while lifetime of returned value still
                // cannot outlive self
                ::std::slice::from_raw_parts(r.as_ptr(), r.len())
            },
            _ => continue,
        }
    }
}