如何定义函数的类型参数(或其关联类型)的函数局部类型别名?

时间:2019-02-06 16:52:44

标签: rust type-alias

我有一个泛型函数foo,具有一些复杂的特征界限:

use std::ops::Index;

// This trait is just as an example
trait Float {
    const PI: Self;
    fn from_f32(v: f32) -> Self;
}
// impl Float for f32, f64 ...

fn foo<C>(container: &C)
where
    C: Index<u32>,
    <C as Index<u32>>::Output: Float,
{
    // ...
}

我现在需要在函数内使用一堆类型<C as Index<u32>>::Output(例如,通过::PI或说::from_f32(3.0)来获得π)。但是,这种类型很难手动输入,并且使整个代码非常冗长且难以阅读。 (注意:在我的真实代码中,实际类型更长甚至更难看。)

为解决此问题,我尝试创建一个函数本地类型别名:

// Inside of `foo`:
type Floaty = <C as Index<u32>>::Output;

但这会导致此错误:

error[E0401]: can't use type parameters from outer function
  --> src/lib.rs:16:20
   |
10 | fn foo<C>(container: &C)
   |    --- - type variable from outer function
   |    |
   |    try adding a local type parameter in this method instead
...
16 |     type Floaty = <C as Index<u32>>::Output;
   |                    ^ use of type variable from outer function

因此,就像其他项目一样,type别名也将被视为无论它们是否在函数中。没有什么好主意,我试图编写一个扩展为以下类型的宏:

// Inside of `foo`:
macro_rules! Floaty {
    () => { <C as Index<u32>>::Output };
}

Floaty!()::PI;    // errors

虽然我在此方面取得了部分成功(Floaty!()在某些类型上下文中有效),但最后一行错误如下:

error: expected one of `.`, `;`, `?`, `}`, or an operator, found `::`
  --> src/lib.rs:20:14
   |
20 |     Floaty!()::PI;    // errors
   |              ^^ expected one of `.`, `;`, `?`, `}`, or an operator here

error[E0575]: expected method or associated constant, found associated type `Index::Output`
  --> src/lib.rs:17:17
   |
17 |         () => { <C as Index<u32>>::Output };
   |                 ^^^^^^^^^^^^^^^^^^^^^^^^^
...
20 |     Floaty!()::PI;    // errors
   |     --------- in this macro invocation
   |
   = note: can't use a type alias as a constructor

我的尝试都没有完全奏效。 是否可以避免每次写出全名?

2 个答案:

答案 0 :(得分:4)

我看到的唯一方法是将类型作为另一个类型参数添加到函数中。

fn foo<F, C>(container: &C)
where
    F: Float,
    C: Index<u32, Output = F>,
{
    let pi = F::PI;
    // ...
}

这通常不会引起类型推断的问题,因为对于给定的F只有一种类型C有效(至少在此示例中如此),但是它确实有一定的用途嘈杂,因为要指定类型F,还必须为C放置一个占位符,反之亦然。

答案 1 :(得分:3)

柴油具有类似的“问题”,并且已经通过defining non-function-local type aliases解决了。我喜欢此解决方案,因为您也可以使用别名来清理特征范围:

type Floaty<C> = <C as Index<u32>>::Output;

fn foo<C>(container: &C)
where
    C: Index<u32>,
    Floaty<C>: Float,
{
    let p = Floaty::<C>::PI;
    // ...
}

请注意,您必须更改特征Float以要求它为Sized才能真正运行此代码。