Rust 返回引用的返回函数

发布于11月24日

我想写一个简单的回忆录函数.

fn memoize<K: Eq + Hash, V: Clone>(mut func: impl FnMut(&K) -> V) -> impl FnMut(K) -> V {
  let mut cache = HashMap::new();
  |key| {
    cache
      .entry(key)
      .or_insert_with_key(|k| func(k))
      .clone()
  }
}

Eq + Hash的参数范围似乎合理，但Clone的返回值似乎是不必要的.理想情况下，签名应为:

fn memoize<K: Eq + Hash, V>(mut func: impl FnMut(&K) -> V) -> impl FnMut(K) -> &mut V

这需要指定返回引用的生存期(这是有意义的).

理想情况下，&mut V应该在引用函数的生命周期内一直存在(或类似的情况:&'a mut impl FnMut(K) -> &'a mut V)

由于FN特征的隐含是不稳定的，如果我想留在FN特征中(不是用大约fn call(&'a mut self, key: K) -> &'a mut V个来写我的 struct )，有没有办法在稳定 rust 蚀中做到这一点？

推荐答案

不幸的是，这是不可能的.即使手动实现FnMut，这也是不可能的，除非我们将Output设置为泛型关联类型:

pub trait GatifiedFnMut<Args: std::marker::Tuple> {
    type Output<'a>
    where
        Self: 'a;
    extern "rust-call" fn call_mut(&mut self, args: Args) -> Self::Output<'_>;
}

pub struct Memoized<K, V, F> {
    cache: HashMap<K, V>,
    func: F,
}

impl<K: Eq + Hash, V, F: FnMut(&K) -> V> GatifiedFnMut<(K,)> for Memoized<K, V, F> {
    type Output<'a> = &'a mut V
    where
        Self: 'a;
    extern "rust-call" fn call_mut(&mut self, (key,): (K,)) -> Self::Output<'_> {
        self.cache.entry(key).or_insert_with_key(&mut self.func)
    }
}

(Note:即使没有GATS也可以解决这个问题，但不能使用当前的FnOnce/FnMut/Fn特性，或者至少非常不舒服，参见下面的链接问题).

More information can be found at users.rust-lang.org.