Rust中使用Borrow trait实现兼容所有权/引用类型的trait时生命周期错误怎么解决？

你遇到的报错本质是生命周期约定不匹配：

• 你定义的EdgeFinder trait中，get_edge方法返回的&Edge遵循Rust默认的生命周期省略规则，默认和&self的生命周期绑定，也就是返回的引用存活时间不能超过self本身的借用时长。
• 但你的实现里，HashMap存储的是&'a Edge类型，从HashMap中取出的&Edge生命周期是独立的'a，要长于HashMap本身的借用时长，这和trait的生命周期约定冲突，因此编译器报错。

以下是两种适配你需求的解决方案： #

方案一：HashMap存储所有权版本的Edge（无需修改trait定义） #

如果你的业务场景允许HashMap持有Edge的所有权，这是最简单的方案，完全符合你原来的实现思路：

use std::collections::HashMap;
use std::borrow::Borrow;

pub struct Edge {
    between: (u32, u32),
    weight: u32,
}

impl Edge {
    pub fn normalize_edge(v1: u32, v2: u32) -> (u32, u32) {
        (v1.min(v2), v1.max(v2))
    }

    fn get_weight(&self) -> u32 {
        self.weight
    }
}

trait EdgeFinder {
    fn get_edge(&self, v1: u32, v2: u32) -> Option<&Edge>;
    fn get_weight(&self, v1: u32, v2: u32) -> Option<u32>;
}

// 支持所有可以Borrow到目标HashMap的类型，包括HashMap本身和&HashMap
impl<B: Borrow<HashMap<(u32, u32), Edge>>> EdgeFinder for B {
    fn get_edge(&self, v1: u32, v2: u32) -> Option<&Edge> {
        self.borrow().get(&Edge::normalize_edge(v1, v2))
    }

    fn get_weight(&self, v1: u32, v2: u32) -> Option<u32> {
        self.get_edge(v1, v2).map(|v| v.get_weight())
    }
}

// EdgeViewer不用做任何修改
pub struct EdgeViewer<T: EdgeFinder> {
    inner: T,
}

impl<T: EdgeFinder> EdgeViewer<T> {
    pub fn new(inner: T) -> Self {
        EdgeViewer { inner }
    }
}

这个方案里，从HashMap中取出的&Edge生命周期和HashMap的借用生命周期完全一致，符合trait的默认生命周期约定，同时Borrow的实现自动支持传入owned的HashMap或者引用类型的&HashMap给EdgeViewer，不会再有trait未实现的报错。

方案二：保留HashMap存储Edge引用的设计，调整trait的生命周期声明 #

如果你确实需要HashMap只存储Edge的引用而非所有权，就需要显式声明返回的Edge引用的生命周期：

use std::collections::HashMap;

pub struct Edge {
    between: (u32, u32),
    weight: u32,
}

impl Edge {
    pub fn normalize_edge(v1: u32, v2: u32) -> (u32, u32) {
        (v1.min(v2), v1.max(v2))
    }

    fn get_weight(&self) -> u32 {
        self.weight
    }
}

// 给trait添加生命周期参数，声明返回的Edge引用的生命周期
trait EdgeFinder<'a> {
    fn get_edge(&self, v1: u32, v2: u32) -> Option<&'a Edge>;
    fn get_weight(&self, v1: u32, v2: u32) -> Option<u32> {
        self.get_edge(v1, v2).map(|e| e.get_weight())
    }
}

// 给owned的HashMap实现EdgeFinder
impl<'a> EdgeFinder<'a> for HashMap<(u32, u32), &'a Edge> {
    fn get_edge(&self, v1: u32, v2: u32) -> Option<&'a Edge> {
        self.get(&Edge::normalize_edge(v1, v2)).copied()
    }
}

// 泛型实现：只要T实现了EdgeFinder，那么&T也自动实现EdgeFinder，支持引用传入
impl<'a, T: EdgeFinder<'a>> EdgeFinder<'a> for &'_ T {
    fn get_edge(&self, v1: u32, v2: u32) -> Option<&'a Edge> {
        (**self).get_edge(v1, v2)
    }
}

pub struct EdgeViewer<T> {
    inner: T,
}

impl<T> EdgeViewer<T> {
    pub fn new(inner: T) -> Self {
        EdgeViewer { inner }
    }
}

// 给EdgeViewer添加EdgeFinder相关实现时带上生命周期约束即可
impl<'a, T: EdgeFinder<'a>> EdgeViewer<T> {
    pub fn get_edge(&self, v1: u32, v2: u32) -> Option<&'a Edge> {
        self.inner.get_edge(v1, v2)
    }
}

这个方案不需要修改你的存储设计，同样支持传入owned或者引用的HashMap给EdgeViewer。

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