Question

我正在尝试将闭包存储为HashMap值。如果我按值传递闭包arg，一切都很好：

use std::collections::hash_map::HashMap;

fn main() {
    let mut cmds: HashMap<String, Box<FnMut(String)->()>>
        = HashMap::new();

    cmds.insert("ping".to_string(), Box::new(|&mut:s| { println!("{}", s); }));

    match cmds.get_mut("ping") {
        Some(f) => f("pong".to_string()),
        _ => ()
    }
}

（playpen）

但是如果我想要一个带参考arg的闭包，那么事情就会向南发展：

use std::collections::hash_map::HashMap;

fn main() {
    let mut cmds: HashMap<String, Box<FnMut(&str)->()>>
        = HashMap::new();

    cmds.insert("ping".to_string(), Box::new(|&mut:s| { println!("{}", s); }));

    match cmds.get_mut("ping") {
        Some(f) => f("pong"),
        _ => ()
    }
}


<anon>:8:37: 8:78 error: type mismatch: the type `closure[<anon>:8:46: 8:77]` implements the trait `core::ops::FnMut(_)`, but the trait `for<'r> core::ops::FnMut(&'r str)` is required (expected concrete lifetime, found bound lifetime parameter )
<anon>:8     cmds.insert("ping".to_string(), Box::new(|&mut:s| { println!("{}", s); }));
                                             ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
<anon>:8:37: 8:78 note: required for the cast to the object type `for<'r> core::ops::FnMut(&'r str)`
<anon>:8     cmds.insert("ping".to_string(), Box::new(|&mut:s| { println!("{}", s); }));
                                             ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
error: aborting due to previous error

（playpen）

我读了How to rewrite code to new unboxed closures的答案，并尝试将地图构建分解为自己的函数，以便有一个地方可以挂起where子句，但没有骰子：

use std::collections::hash_map::HashMap;

fn mk_map<F>() -> HashMap<String, (String, Box<F>)>
    where F: for<'a> FnMut(&'a str) -> ()
{
    let mut cmds: HashMap<String, (String, Box<F>)> = HashMap::new();
    cmds.insert("ping".to_string(), ("ping".to_string(), Box::new(|&mut:s| { println!("{}", s); })));
    cmds
}   

fn main() {
    let cmds = mk_map();
    match cmds.get_mut("ping") {
        Some(&mut (_, ref mut f)) => f("pong"),
        _ => println!("invalid command")
    }
}


<anon>:8:58: 8:99 error: mismatched types: expected `Box<F>`, found `Box<closure[<anon>:8:67: 8:98]>` (expected type parameter, found closure)
<anon>:8     cmds.insert("ping".to_string(), ("ping".to_string(), Box::new(|&mut:s| { println!("{}", s); })));
                                                                  ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

（playpen）

这样做的正确方法是什么？

Answer 1

我的解决方案：

#![allow(unstable)]
use std::collections::hash_map::HashMap;

// #1 returning a trait object   
fn mk_map<'a>() -> HashMap<String, (String, Box<FnMut(&str) + 'a>)> {
    let mut cmds : HashMap<_, (_, Box<FnMut(&str)>)> = HashMap::new();

    cmds.insert("ping".to_string(), ("ping".to_string(), 
        Box::new(|&mut: s: &str| { println!("{}", s); })));
    // #2                  ^-- give a little help to the compiler here
    cmds
}   

fn main() {
    let mut cmds = mk_map();
    // minor change: cmds needs to be mutable
    match cmds.get_mut("ping") {
        Some(&mut (_, ref mut f)) => f("pong"),
        _ => println!("invalid command")
    }
}

成分：

返回特质对象
在闭包的参数类型上为编译器提供一些帮助：Box::new(|&mut: s: &str|

说实话，我并不是100％肯定＃2的原因（我的意思是，至少要把它留下来应该给出一个更容易理解的错误信息）。可能是rustc的一个问题。

在＃1上，我几乎可以肯定它是必需的，因为你不能为从函数返回的闭包命名具体的返回类型（它是由编译器动态创建的匿名类型），所以Trait对象现在应该是返回闭包的唯一方法。

<强>附录回复评论：

假设你有几种类型的trait Foo {}：

trait Foo {}
impl Foo for u32 {}
impl Foo for Vec<f32> {}

如果你编写一个像mk_map一样的函数（让我们称之为make_foo），我评论说实现它很难。我们来看看：

fn mk_foo<F>() -> Box<F> where F: Foo {
    unimplemented!()
}

mk_foo的签名说我应该能够用任何实现Foo的类型调用该函数。所以这一切都应该是有效的：

   let a: Box<Vec<f32>> = mk_foo::<Vec<f32>>();
   let b: Box<u32> = mk_foo::<u32>();

即。写入的函数不返回特征对象。它有希望返回一个Box，其中包含调用者选择的任何具体类型。这就是为什么实际实现这个功能并不容易的原因。它应该知道如何从零开始创建几种类型。

在HashMap中存储带有引用arg的未装箱闭包

1 个答案: