Rust中如何定义Trait替换结构体泛型类型且无需手动指定返回类型？

需要定义一个Trait，用于修改结构体内部属性的类型，但保持结构体本身的类型不变——比如为Foo<A>实现该Trait后，方法应返回Foo<B>而非其他类型。当前实现需要手动指定关联类型Output，希望能静态推导返回类型，无需手动定义，但尝试返回Self<TO>时报错：Self类型不允许使用类型参数。

现有实现代码 #

use std::collections::HashMap;
use std::hash::Hash;

trait StorageId: PartialEq + Eq + Hash {}

struct MyEntity<SI: StorageId> {
    avatar_image: SI,
    other_field: String
}

trait CurrentReplaceStorageId<TO: StorageId> {
    type From: StorageId;
    
    type Output;
    
    fn replace_storage_id<'a>(self, values: &'a mut HashMap<Self::From, TO>) -> Self::Output;
}

impl <F: StorageId, T: StorageId> CurrentReplaceStorageId<T> for MyEntity<F> {
    type From = F;
    
    // 必须手动指定这一行
    type Output = MyEntity<T>;
    
    fn replace_storage_id<'a>(self, values: &'a mut HashMap<Self::From, T>) -> Self::Output {
        MyEntity {
            avatar_image: values.remove(&self.avatar_image).unwrap(),
            other_field: self.other_field
        }
    }
}

尝试的报错方案 #

use std::collections::HashMap;
use std::hash::Hash;

trait StorageId: PartialEq + Eq + Hash {}

struct MyEntity<SI: StorageId> {
    avatar_image: SI,
    other_field: String
}

trait ExpectedReplaceStorageId<TO: StorageId> {
    type From: StorageId;
    
    fn replace_storage_id<'a>(self, values: &'a mut HashMap<Self::From, TO>) -> Self<TO>;
}

impl <F: StorageId, T: StorageId> ExpectedReplaceStorageId<T> for MyEntity<F> {
    type From = F;
    
    fn replace_storage_id<'a>(self, values: &'a mut HashMap<Self::From, T>) -> Self<T> {
        MyEntity {
            avatar_image: values.remove(&self.avatar_image).unwrap(),
            other_field: self.other_field
        }
    }
}

报错信息：Self类型不允许使用类型参数

解决方案 #

在Rust中，Self代表当前实现的具体实例化类型（比如这里的MyEntity<F>），它本身是一个完整类型，不能再附加类型参数。要实现自动推导返回类型，最标准的方式是使用关联类型构造器（Rust 1.51及以上版本支持），它能定义“给定一个类型参数，返回对应结构体类型”的关联关系。

优化后的实现 #

use std::collections::HashMap;
use std::hash::Hash;

trait StorageId: PartialEq + Eq + Hash {}

struct MyEntity<SI: StorageId> {
    avatar_image: SI,
    other_field: String
}

// 定义Trait，使用关联类型构造器表示替换类型后的自身
trait ReplaceStorageId {
    type From: StorageId;
    // 关联类型构造器：接受StorageId类型，返回对应的结构体类型
    type With<NewSI: StorageId>: ReplaceStorageId<From = NewSI>;

    fn replace_storage_id<NewSI: StorageId>(self, values: &mut HashMap<Self::From, NewSI>) -> Self::With<NewSI>;
}

impl<F: StorageId> ReplaceStorageId for MyEntity<F> {
    type From = F;
    // 指定构造器返回替换泛型后的结构体类型
    type With<NewSI: StorageId> = MyEntity<NewSI>;

    fn replace_storage_id<NewSI: StorageId>(self, values: &mut HashMap<Self::From, NewSI>) -> Self::With<NewSI> {
        MyEntity {
            avatar_image: values.remove(&self.avatar_image).unwrap(),
            other_field: self.other_field
        }
    }
}

关键说明 #

• 关联类型构造器type With<NewSI: StorageId>允许我们在Trait中定义“类型转换规则”，实现时只需要指定With<NewSI>对应MyEntity<NewSI>，后续调用方法时编译器会自动推导返回类型。
• 这种设计不仅适用于MyEntity，任何单泛型参数的结构体都可以按此模式实现ReplaceStorageId，无需重复手动指定返回类型。

内容的提问来源于stack exchange，提问作者Ramesh Kithsiri HettiArachchi