Rust 如何让支持所有权/借用输入的trait方法始终返回所有权类型

问题核心原因 #

你原先的代码报错是因为：对于引用类型&Vec<f64>来说，它的ToOwned trait 实现中关联类型Owned就是&Vec<f64>本身，和你预期的返回所有权类型Vec<f64>的需求不匹配。

标准实现方案 #

不需要自定义辅助trait，基于标准库的AsRef trait 就可以实现需求，核心思路是直接把要返回的所有权类型T作为trait的泛型参数，输入参数约束为所有可转换为T引用的类型：

use std::convert::AsRef;

trait Foo<T> {
    /// 输入x可以是任意能转为T引用的类型（包括T本身、&T、Box<T>等）
    fn f<X: AsRef<T>>(&self, x: X) -> f64;
    /// 始终返回所有权类型T
    fn g(&self) -> T;
}

struct Float(f64);

impl Foo<f64> for Float {
    fn f<X: AsRef<f64>>(&self, x: X) -> f64 {
        x.as_ref() + self.0
    }
    
    fn g(&self) -> f64 {
        self.0 * 2.0
    }
}

struct List(Vec<f64>);

impl Foo<Vec<f64>> for List {
    fn f<X: AsRef<Vec<f64>>>(&self, x: X) -> f64 {
        x.as_ref().iter().sum()
    }
    
    fn g(&self) -> Vec<f64> {
        self.0.iter().map(|&x| 2.0 * x).collect()
    }
}

fn main() {
    let float = Float(2.0);
    println!("{} {}", float.f(3.0), float.g());
   
    let list = List(vec![0.0, 1.0, 2.0]);
    println!("{} {:?}", list.f(&vec![1.0, 2.0]), list.g());
}

方案优势 #

• 完全基于标准库原生组件，不需要额外定义自定义辅助trait，兼容性和健壮性更强
• 输入参数自动兼容所有符合AsRef<T>约束的类型，不止支持T和&T，还覆盖Box<T>、Cow<'_, T>等常见封装类型，扩展性更好
• 显式将返回的所有权类型作为trait泛型参数，语义清晰，从根源上避免了引用类型的ToOwned默认实现带来的类型不匹配问题

如果你的f方法需要消费输入参数x的所有权，可以把AsRef<T>替换为Into<T>，即可实现传值或引用都自动转换为所有权类型T使用。

内容的提问来源于stack exchange，提问作者E. J. Winkleberry