Yew框架中如何在消息内传递子树引用实现节点选中

首先明确结论：在Yew消息中传递节点引用的方案完全不可行，也绝对不推荐用unsafe绕过编译检查实现这个需求。

为什么传引用会报错 #

Yew的消息会被存在组件内部的调度队列里，等待update生命周期执行时处理，因此消息类型必须满足'static生命周期约束——也就是消息里携带的数据必须能安全存活到程序运行结束，不能依赖临时的借用。
你在view方法里拿到的&Tree只是当前渲染轮次下的临时借用，view执行结束后这个借用就会失效。如果强行把这个引用存到消息里，等消息被处理的时候，引用指向的内存可能已经被释放（比如后续修改树结构替换、删除节点），就会产生悬垂指针导致内存安全问题。Rust编译器报生命周期错误，本质就是在阻止你写这种必然有安全隐患的代码，不是语法漏写的问题。

至于用unsafe实现的思路完全不可取：unsafe只是绕过编译器的生命周期检查，不会帮你解决引用悬垂的问题，只要后续树结构发生变动，裸指针就会指向无效内存，轻则页面直接崩溃，重则出现无法排查的逻辑错误，属于典型的滥用unsafe场景。

正确实现思路 #

你之前看到数组/向量结构传索引的方案，本质是传递可以唯一定位目标数据的标识，这个思路完全可以套用到树结构上——不需要连续内存的下标，只要给每个节点分配一个全局唯一的ID，把ID存在消息里传递，处理消息时通过ID找到对应节点修改选中状态即可，和数组传索引的逻辑完全一致。

具体修改步骤 #

• 调整消息定义，用节点ID替代引用：

pub enum Msg {
    SelectTree(usize) // 携带选中节点的唯一ID
}

• 给节点注解结构体加ID字段，用静态原子计数器生成全局唯一ID，不需要手动维护：

use std::sync::atomic::{AtomicUsize, Ordering};
static NODE_ID_COUNTER: AtomicUsize = AtomicUsize::new(0);

#[derive(Debug, Clone)]
pub struct Annot {
    id: usize, // 节点唯一标识
    is_selected: bool
}

impl Default for Annot {
    fn default() -> Annot { 
        Annot { 
            id: NODE_ID_COUNTER.fetch_add(1, Ordering::Relaxed),
            is_selected: false 
        } 
    }
}

• 渲染节点时，把当前节点的ID捕获到点击回调里，点击时直接发送携带对应ID的消息：

// 叶子节点渲染逻辑
Tree::Leaf {ann} => {
    let cls = if ann.is_selected { "selected" } else { "" };
    let node_id = ann.id;
    html! { 
        <li> 
            <a class={cls} 
                onclick={ctx.link().callback(move |_| Msg::SelectTree(node_id))}>
            </a> 
        </li> 
    }
}

// 中间节点渲染逻辑
Tree::Node {ann, x, l, r} => {
    let cls = if ann.is_selected { "selected" } else { "" };
    let node_id = ann.id;
    html! {
        <li>
            <a class={cls} 
                onclick={ctx.link().callback(move |_| Msg::SelectTree(node_id))}>
                { x }
            </a>
            <ul>
                { l.view(ctx) }
                { r.view(ctx) }
            </ul>
        </li>
    }
}

• 给Tree实现一个辅助方法，用来递归更新选中状态：遍历所有节点，把匹配ID的节点设为选中，其余节点取消选中即可。

impl Tree {
    // 其余原有方法保留，新增set_selected方法
    fn set_selected(&mut self, target_id: usize) {
        match self {
            Tree::Leaf {ann} => {
                ann.is_selected = ann.id == target_id;
            }
            Tree::Node {ann, l, r, ..} => {
                ann.is_selected = ann.id == target_id;
                l.set_selected(target_id);
                r.set_selected(target_id);
            }
        }
    }
}

• 最后在组件的update逻辑里调用这个方法即可：

fn update(&mut self, ctx: &Context<Self>, msg: Self::Message) -> bool {
    match msg {
        Msg::SelectTree(id) => {
            self.tree.set_selected(id);
            true
        }
    }
}

这种实现完全符合Rust的安全规则，没有任何生命周期问题，也不需要用到unsafe，逻辑和你之前看到的数组传索引方案本质一致，只是把数组下标换成了自定义的节点ID而已。

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