scala - 拉皮条 scalaz 备忘录

Question

我喜欢新的 scalaz Memo 功能，但发现它缺少两点：1）它隐藏了我需要访问的底层地图——至少是所有值的列表，以及 2）我想要一个使用 val 实现的版本我在某处读到的 scala.collection.concurrent.TrieMap 比 var Map 更可取。

我还不是一个隐含的巫师。有没有办法让这个备忘录类添加支持此功能的版本，或者我将不得不剪切/粘贴到一个独特的新类中？

Answer 1

这可以通过内置的 Memo.memo 函数来完成。Memo.memo 从函数 F => K => V 创建一个 Memo 实例。这也让您可以轻松访问底层 trie。例如：

scala> def trieMemo[A, B](trie: collection.concurrent.TrieMap[A, B]) = 
         Memo.memo[A, B](f => k => trie.getOrElseUpdate(k, f(k)))
trieMemo: [A, B](trie: scala.collection.concurrent.TrieMap[A,B])scalaz.Memo[A,B]

scala> val trie = collection.concurrent.TrieMap[Int, Int]()
trie: scala.collection.concurrent.TrieMap[Int,Int] = TrieMap()

scala> val f = trieMemo(trie)(n => n * n)
f: Int => Int = <function1>

scala> f(5)
res0: Int = 25

scala> f(10)
res1: Int = 100

scala> trie
res2: scala.collection.concurrent.TrieMap[Int,Int] = TrieMap(5 -> 25, 10 -> 100)