当arena不足时如何重构循环数据使用且不在类型签名引入Rc/RefCell

需求背景 #

Rust 开发中遇到循环数据引用场景，现有 arena 方案不足以满足需求，同时不希望在类型签名中引入 Rc/RefCell 造成类型污染。期望实现节点间通过邮箱通信的逻辑，且必须保留 node_action 方法的入参签名（传入不可变输入引用和可变输出引用）。

原始实现代码 #

struct Node {
    outbound_mailboxes: Vec<u32>,       // 节点间通信内存
    inbound_links: Vec<(usize, usize)>, // (源节点索引, 源节点发件箱索引)
}

fn node_action(inputs: Vec<&u32>, outputs: Vec<&mut u32>) {  // 需保留该签名
    println!("{:?}, {:?}", inputs, outputs);
    // 业务逻辑：读取输入，写入输出
}

fn main() {
    // 3个节点双向连接：0--1、0--2
    let mut nodes = vec![
        Node { outbound_mailboxes: vec![0, 1], inbound_links: vec![(1, 0), (2, 0)] },
        Node { outbound_mailboxes: vec![2],    inbound_links: vec![(0, 0)] },
        Node { outbound_mailboxes: vec![3],    inbound_links: vec![(0, 1)] },
    ];

    // 触发节点0的动作
    let mut inputs: Vec<&u32> = Vec::new();
    for (i, j) in &nodes[0].inbound_links {
        inputs.push(&nodes[*i].outbound_mailboxes[*j]);
    }
    let outputs: Vec<&mut u32> = nodes[0].outbound_mailboxes.iter_mut().map(|m| m).collect();
    node_action(inputs, outputs);
}

编译报错信息 #

error[E0502]: cannot borrow `nodes` as mutable because it is also borrowed as immutable
  --> src/main.rs:24:34
   |
22 |         inputs.push(&nodes[*i].outbound_mailboxes[*j]);
   |                      ----- 不可变借用发生于此
23 |     }
24 |     let outputs: Vec<&mut u32> = nodes[0].outbound_mailboxes.iter_mut().map(|m| m).collect();
   |                                  ^^^^^ 可变借用发生于此
25 |     node_action(inputs, outputs);
   |                 ------ 不可变借用后续在此处使用

解决思路 #

核心是利用 Rust 切片的 split_at_mut 方法，将目标节点和其他源节点的借用拆分，让借用检查器可以识别出不同节点的引用不存在重叠，避免同时对整个 nodes 数组存在不可变和可变借用的冲突。

最终实现代码 #

struct Node {
    outbound_mailboxes: Vec<u32>,       // 节点间通信内存
    inbound_links: Vec<(usize, usize)>, // (源节点索引, 源节点发件箱索引)
}

fn node_action(inputs: &[&u32], outputs: &[&mut u32]) {
    println!("{:?}, {:?}", inputs, outputs)
}

fn trigger_action(nodes: &mut [Node], target_node_idx: usize) {
    assert!(target_node_idx < nodes.len());
    // 拆分出目标节点左侧的所有节点和剩余切片
    let (left_nodes, rest) = nodes.split_at_mut(target_node_idx);
    // 从剩余切片拆分出目标节点和右侧的所有节点
    let (target_node, right_nodes) = rest.split_first_mut().unwrap();
    // 安全获取目标节点发件箱的可变引用作为输出
    let outputs: Vec<&mut u32> = target_node.outbound_mailboxes.iter_mut().collect();
    let mut inputs: Vec<&u32> = Vec::new();
    for (i, j) in &target_node.inbound_links {
        // 禁止节点自引用，避免借用冲突
        assert_ne!(*i, target_node_idx);
        // 从左/右侧节点切片中读取源节点数据，不会和目标节点的可变借用冲突
        let node = {
            if *i < target_node_idx { &left_nodes[*i] }
            else { &right_nodes[*i - (target_node_idx + 1)] }
        };
        inputs.push(&node.outbound_mailboxes[*j]);
    }
    node_action(&inputs, &outputs)
}

fn main() {
    // 3个节点双向连接：0--1、0--2
    let mut nodes = vec![
        Node { outbound_mailboxes: vec![0, 1], inbound_links: vec![(1, 0), (2, 0)] },
        Node { outbound_mailboxes: vec![2],    inbound_links: vec![(0, 0)] },
        Node { outbound_mailboxes: vec![3],    inbound_links: vec![(0, 1)] },
    ];

    trigger_action(&mut nodes, 0); // 输出 "[2, 3], [0, 1]"
    trigger_action(&mut nodes, 1); // 输出 "[0], [2]"
    trigger_action(&mut nodes, 2); // 输出 "[0], [1]"
}

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