如何编写可接收T类型或&T类型参数并返回对应T类型的Rust函数？

要实现一个既能接受类型T又能接受&T，并分别返回原始数据或引用克隆的函数，可以通过自定义trait的方式实现，具体步骤如下：

1. 定义自定义转换trait #

首先创建一个trait，用来统一处理“获取所有权类型T”的逻辑：

trait IntoOwned<T> {
    fn into_owned(self) -> T;
}

2. 为目标类型实现trait #

分别为T和&T实现这个trait，对应两种输入场景：

• 当输入是T本身时，直接返回自身，无需克隆
• 当输入是&T时，调用clone()返回克隆后的T（要求T实现Clone trait）

// 处理T类型输入
impl<T> IntoOwned<T> for T {
    fn into_owned(self) -> T {
        self
    }
}

// 处理&T类型输入
impl<T: Clone> IntoOwned<T> for &T {
    fn into_owned(self) -> T {
        self.clone()
    }
}

3. 实现通用函数 #

编写泛型函数foo，接受任何实现了IntoOwned<T>的类型作为参数，调用trait方法完成转换：

fn foo<T>(input: impl IntoOwned<T>) -> T {
    input.into_owned()
}

使用示例 #

下面是一个完整的使用例子，验证两种输入场景的行为：

#[derive(Clone, Debug)]
struct MyStruct(i32);

fn main() {
    let original = MyStruct(42);
    let reference = &original;
    
    // 传入T类型，转移所有权，original后续不可再使用
    let owned_result = foo(original);
    // 传入&T类型，克隆数据，reference仍可正常使用
    let cloned_result = foo(reference);
    
    println!("{:?}", owned_result); // 输出: MyStruct(42)
    println!("{:?}", cloned_result); // 输出: MyStruct(42)
}

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