Rust中使用petgraph的NodeIndex封装函数时编译错误求助

我是Rust新手，使用petgraph crate开发，将DiGraph作为自定义结构体MyGraph的成员，封装add_node和update_edge函数供外部调用时遇到编译错误，代码如下：

use petgraph::graph::DiGraph;
use petgraph::adj::NodeIndex;

#[derive(Default)]
struct MyNode {}

#[derive(Default)]
struct MyGraph {
    g: DiGraph<MyNode,()>
}

impl MyGraph {
    pub fn add_node(&mut self, node: MyNode) -> NodeIndex {
        self.g.add_node(node) // 编译失败：返回值类型为u32，与NodeIndex不匹配
    }
    pub fn update_edge(&mut self, a: NodeIndex, b: NodeIndex) {
        self.g.update_edge(a, b, ()); // 编译失败：期望NodeIndex，实际传入u32
    }
}
fn main() {
    let mut my_graph=MyGraph::default();
    let node1=my_graph.add_node(MyNode::default());
    let node2=my_graph.add_node(MyNode::default());
    my_graph.update_edge(node1,node2);
}

编译器报错信息片段：

...
error[E0308]: arguments to this function are incorrect
   --> src\main.rs:17:16
    |
17  |         self.g.update_edge(a, b, ());
    |                ^^^^^^^^^^^
    |
note: expected struct `NodeIndex`, found `u32`
   --> src\main.rs:17:28
    |
17  |         self.g.update_edge(a, b, ());
    |                            ^
    = note: expected struct `NodeIndex`
                 found type `u32`
...

请问哪里出错了？正确的参数类型和返回类型应该是什么？

错误原因 #

你导入的NodeIndex路径错误——petgraph::adj::NodeIndex是petgraph内部用于邻接表的旧类型（本质是u32的别名），而DiGraph使用的是petgraph::graph::NodeIndex，这两个是完全不同的类型，导致了类型不匹配。

修正方案 #

将NodeIndex的导入路径改为petgraph::graph::NodeIndex，同时确保函数的返回值和参数类型与DiGraph的方法保持一致：

use petgraph::graph::{DiGraph, NodeIndex}; // 修正导入路径

#[derive(Default)]
struct MyNode {}

#[derive(Default)]
struct MyGraph {
    g: DiGraph<MyNode,()>
}

impl MyGraph {
    pub fn add_node(&mut self, node: MyNode) -> NodeIndex {
        self.g.add_node(node) // 现在返回的类型与声明一致
    }
    pub fn update_edge(&mut self, a: NodeIndex, b: NodeIndex) {
        self.g.update_edge(a, b, ()); // 参数类型匹配，编译通过
    }
}
fn main() {
    let mut my_graph=MyGraph::default();
    let node1=my_graph.add_node(MyNode::default());
    let node2=my_graph.add_node(MyNode::default());
    my_graph.update_edge(node1,node2);
}

补充说明 #

• petgraph::graph::NodeIndex是DiGraph等图结构的官方节点索引类型，它内部包含了图的标识信息，避免不同图之间的索引混淆。
• 如果你需要使用邻接表相关的API，才需要引入petgraph::adj模块的类型，但对于常规的图操作，使用graph模块下的类型即可。

内容的提问来源于stack exchange，提问作者Udo Eisenbarth