rust - 我可以有条件地提供特征函数的默认实现吗？

Question

我有以下特点:

trait MyTrait {
    type A;
    type B;

    fn foo(a: Self::A) -> Self::B;

    fn bar(&self);
}

bar 等其他函数必须始终由特征的用户实现。

我想给 foo 一个默认实现，但只有当类型 A = B 时。

伪 Rust 代码:

impl??? MyTrait where Self::A = Self::B ??? {
    fn foo(a: Self::A) -> Self::B {
        a
    }
}

这是可能的:

struct S1 {}

impl MyTrait for S1 {
    type A = u32;
    type B = f32;

    // `A` is different from `B`, so I have to implement `foo`
    fn foo(a: u32) -> f32 {
        a as f32
    }

    fn bar(&self) {
        println!("S1::bar");
    }
}

struct S2 {}

impl MyTrait for S2 {
    type A = u32;
    type B = u32;

    // `A` is the same as `B`, so I don't have to implement `foo`,
    // it uses the default impl

    fn bar(&self) {
        println!("S2::bar");
    }
}

这在 Rust 中可能吗？

Answer 1

您可以通过引入冗余类型参数在特征定义本身中提供默认实现:

trait MyTrait {
    type A;
    type B;

    fn foo<T>(a: Self::A) -> Self::B
    where
        Self: MyTrait<A = T, B = T>,
    {
        a
    }
}

可以为个别类型覆盖此默认实现。但是，专用版本将从特征上的 foo() 定义继承特征绑定(bind)，因此您只能在 A == B 时实际调用方法:

struct S1;

impl MyTrait for S1 {
    type A = u32;
    type B = f32;

    fn foo<T>(a: Self::A) -> Self::B {
        a as f32
    }
}

struct S2;

impl MyTrait for S2 {
    type A = u32;
    type B = u32;
}

fn main() {
    S1::foo(42);  // Fails with compiler error
    S2::foo(42);  // Works fine
}

Rust 也有一个 unstable impl specialization feature ，但我不认为它可以用来实现你想要的。