Question

在尝试重载调用语法时，我引入了一个简单的缓存，可以缓存昂贵计算的结果。我对使用一段语法感到有点困惑。我将在问题出现之前逐步介绍代码。

缓存旨在像这样使用：

fn fib(x: i32) -> i32 {
    if x < 2 { x } else { fib(x-1) + fib(x-2) }
}

fn main() {
    let mut cfib = Cache::new(fib);

    // Loop that repeats computation and extracts it from the cache
    // the second time.
    for x in 1..200 {
        let val = 5 * x % 40;
        println!("fibc({}) = {}", val, cfib(val));
    }
}

我们首先使用序言来启用尚未稳定的功能：

#![feature(fn_traits, unboxed_closures)]

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

我们将缓存作为具有HashMap的结构和计算新值的函数引入。

struct Cache<T, R> {
    cache: HashMap<T, R>,
    func: fn(T) -> R,
}

impl<T, R> Cache<T, R>
    where T: Eq + Hash + Copy,
          R: Copy
{
    fn new(func: fn(T) -> R) -> Cache<T, R> {
        Cache { cache: HashMap::new(), func: func }
    }

    fn compute(&mut self, x: T) -> R {
        let func = self.func;
        let do_insert = || (func)(x);
        *self.cache.entry(x).or_insert_with(do_insert)
    }
}

我创建了FnMut特征的实现，因为缓存需要是可变的。

impl<T, R> FnMut<(T,)> for Cache<T, R>
    where T: Eq + Hash + Copy,
          R: Copy
{
    extern "rust-call" fn call_mut(&mut self, args: (T,))
        -> Self::Output
    {
        let (arg,) = args;
        self.compute(arg)
    }
}

即使我发现语法FnMut<(T,)>很奇怪，但这很好且安全，并且很清楚地表达了意图。由于我需要定义函数的返回类型，我想将开头写为：

impl<T, R> FnMut<(T,), Output=R> for Cache<T, R>
    where T: Eq + Hash + Copy,
          R: Copy
{}

但是失败并出现错误：

error[E0229]: associated type bindings are not allowed here
  --> src/main.rs:55:24
   |
55 | impl<T, R> FnMut<(T,), Output=R> for Cache<T, R>
   |                        ^^^^^^^^ associate type not allowed here

我必须像这样实施FnOnce：

impl<T, R> FnOnce<(T,)> for Cache<T,R>
    where T: Eq + Hash + Copy,
          R: Copy
{
    type Output = R;

    extern "rust-call" fn call_once(self, _arg: (T,))
        -> Self::Output
    {
        unimplemented!()
    }
}

这是毫无意义的，因为call_once将永远不会被调用，而Associated Types似乎应该是可能的。但是，它失败并出现错误，那里不允许关联类型。

Rust Compiler Error Index提到语法Fn(T) -> R并且还说Fn<(T,), Output=U>应该有效，但即使我使用每晚的Rust编译器，我也无法使其工作。

由于希望在编译时捕获尽可能多的错误，因此最好避免在FnOnce中创建“未实现”的函数，因为这将在运行时而不是编译时失败。

是否可以仅实现FnMut并以某种方式提供函数的返回类型？

Answer 1

这是毫无意义的，因为永远不会调用call_once

这不取决于你的决定;这取决于来电者。他们可能决定在FnOnce上下文中调用缓存。

好消息是FnOnce的完全合理的实施 - 只需委托FnMut实施：

impl<T, R> FnOnce<(T,)> for Cache<T,R>
    where T: Eq + Hash + Copy,
          R: Copy
{
    type Output = R;

    extern "rust-call" fn call_once(mut self, arg: (T,))
        -> Self::Output
    {
        self.call_mut(arg)
    }
}

这就是编译器自动实现这些特性的原因;如果合适，它还会将FnMut委托给Fn。

另见

是否有可能避免对“FnOnce”进行无意义的定义？

1 个答案: