如何使用conservative_impl_trait 返回对迭代器的引用？

Question

我有一个petgraph::Graph 结构，我通过给每个节点权重一个parent_edge_idx 强加了一个树结构，这是从其父节点连接到自身的边的Option<EdgeIdx>。

我需要遍历节点的子节点。我需要连接边的边权和子节点的权重。

我想将该迭代分解为一个帮助函数，该函数返回对Iterator<Item = (EdgeIdx, NodeIdx)> 的引用。我想免费做这个；因为我必须借用self.search_tree才能做到这一点，所以迭代器只在self的生命周期内有效。

1. 这是一个合理的函数吗？
2. 可以写这个函数吗？

任何门控功能都可以；我每晚都在。

fn children<'a>(
    &'a mut self,
    node_idx: NodeIdx,
) -> &'a impl Iterator<Item = (EdgeIdx, NodeIdx)> {
    &self.search_tree.neighbors(node_idx).map(|child_idx| {
        let node = self.search_tree.node_weight(child_idx).unwrap();
        let edge_idx = node.parent_edge_idx.unwrap();
        (edge_idx, child_idx)
    })
}

Answer 1

in this question 已经介绍了如何返回迭代器。

1. 注意你don't need to return a reference：你想直接返回一个迭代器value，所以如果我们删除方法体和返回类型中的第一个&，那就更接近了我们需要的。

2. 我们将使用impl Iterator，这样我们就不必准确地命名实际的迭代器类型。请注意（下面的代码）我们need to use the impl Iterator<..> + 'a syntax，这意味着（匿名）迭代器包含至少在'a 的生命周期内有效的引用。

3. 我们不能在这里使用&mut self！请注意，我们需要借用两次self.search_tree：一次用于.neighbors() 迭代器，一次用于map 闭包中使用的self.search_tree。多次借用与可变引用不兼容。

4. 我们将move作为闭包的捕获模式，这样它就可以直接捕获self引用，而不是通过引用（这很重要，因为我们可以返回迭代器和闭包。

5. Petgraph 专用，但我们将 g.node_weight(node_index).unwrap() 替换为等效的 &g[node_index]，但后者更易于阅读。

这里是您的代码的复制品，但对 1-5 进行了修改以使其编译：

#![feature(conservative_impl_trait)]
extern crate petgraph;

use petgraph::Graph;
use petgraph::graph::{NodeIndex, EdgeIndex};

struct Foo {
    search_tree: Graph<Node, i32>,
}

struct Node {
    parent_edge_idx: Option<EdgeIndex>,
}

impl Foo {
    fn children<'a>(&'a self, node_idx: NodeIndex)
        -> impl Iterator<Item = (EdgeIndex, NodeIndex)> + 'a
    {
        self.search_tree.neighbors(node_idx).map(move |child_idx| {
            let node = &self.search_tree[child_idx];
            let edge_idx = node.parent_edge_idx.unwrap();
            (edge_idx, child_idx)
        })
    }
}