generics - 闭包作为 Rust 结构中的一种类型

Question

我正在尝试在 Rust 中创建这样的结构:

pub struct Struct<T, F>
    where T: Eq,
          T: Hash,
          F: Fn() -> T
{
    hashMap: HashMap<T, F>,
    value: T,
}

我的构造函数如下所示:

pub fn new(init_value: T) -> Struct<T, F> {
    Struct {
        hashMap: HashMap::new(),
        value: init_state,
    }
}

然而，当尝试实例化类时，使用 let a = Struct::<MyEnum>::new(MyEnum::Init); ，编译器提示泛型需要两个参数(expected 2 type arguments, found 1)

我看到了here此代码有效:

fn call_with_one<F>(some_closure: F) -> i32
    where F: Fn(i32) -> i32 {

    some_closure(1)
}

let answer = call_with_one(|x| x + 2);

我想问题出在我的模板实例化中有另一个泛型，但我该怎么做呢？

Answer 1

Struct::new 没有任何依赖于 F 的参数，因此编译器无法推断它应该为 F< 使用什么类型。如果您稍后调用了一个使用 F 的方法，那么编译器将使用该信息来确定 Struct 的具体类型。例如:

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

pub struct Struct<T, F>
    where T: Eq,
          T: Hash,
          F: Fn() -> T,
{
    hash_map: HashMap<T, F>,
    value: T,
}

impl<T, F> Struct<T, F>
    where T: Eq,
          T: Hash,
          F: Fn() -> T,
{
    pub fn new(init_value: T) -> Struct<T, F> {
        Struct {
            hash_map: HashMap::new(),
            value: init_value,
        }
    }

    pub fn set_fn(&mut self, value: T, func: F) {
        self.hash_map.insert(value, func);
    }
}

fn main() {
    let mut a = Struct::new(0);
    a.set_fn(0, || 1); // the closure here provides the type for `F`
}

不过这有一个问题。如果我们用不同的闭包第二次调用 set_fn:

fn main() {
    let mut a = Struct::new(0);
    a.set_fn(0, || 1);
    a.set_fn(1, || 2);
}

然后我们得到一个编译器错误:

error[E0308]: mismatched types
  --> <anon>:33:17
   |
33 |     a.set_fn(1, || 2);
   |                 ^^^^ expected closure, found a different closure
   |
   = note: expected type `[closure@<anon>:32:17: 32:21]`
   = note:    found type `[closure@<anon>:33:17: 33:21]`
note: no two closures, even if identical, have the same type
  --> <anon>:33:17
   |
33 |     a.set_fn(1, || 2);
   |                 ^^^^
help: consider boxing your closure and/or using it as a trait object
  --> <anon>:33:17
   |
33 |     a.set_fn(1, || 2);
   |                 ^^^^

正如编译器所提到的，每个闭包表达式都定义了一个全新的类型并求值为该类型。但是，通过按照您的方式定义 Struct，您将强制 HashMap 中的所有函数具有相同的类型。这真的是你想要的吗？

如果你想将 T 的不同值映射到可能不同类型的闭包，那么你需要使用特征对象而不是泛型，正如编译器所建议的那样。如果您希望结构拥有闭包，则必须在对象类型周围使用 Box。

pub struct Struct<T>
    where T: Eq,
          T: Hash,
{
    hash_map: HashMap<T, Box<Fn() -> T + 'static>>,
    value: T,
}

set_fn 看起来像这样:

pub fn set_fn<F: Fn() -> T + 'static>(&mut self, value: T, func: F) {
    self.hash_map.insert(value, Box::new(func));
}