41 lines
816 B
Go
41 lines
816 B
Go
// Copyright © 2016 Alan A. A. Donovan & Brian W. Kernighan.
|
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
|
|
|
// See page 272.
|
|
|
|
//!+
|
|
|
|
// Package memo provides a concurrency-unsafe
|
|
// memoization of a function of type Func.
|
|
package memo
|
|
|
|
// A Memo caches the results of calling a Func.
|
|
type Memo struct {
|
|
f Func
|
|
cache map[string]result
|
|
}
|
|
|
|
// Func is the type of the function to memoize.
|
|
type Func func(key string) (interface{}, error)
|
|
|
|
type result struct {
|
|
value interface{}
|
|
err error
|
|
}
|
|
|
|
func New(f Func) *Memo {
|
|
return &Memo{f: f, cache: make(map[string]result)}
|
|
}
|
|
|
|
// NOTE: not concurrency-safe!
|
|
func (memo *Memo) Get(key string) (interface{}, error) {
|
|
res, ok := memo.cache[key]
|
|
if !ok {
|
|
res.value, res.err = memo.f(key)
|
|
memo.cache[key] = res
|
|
}
|
|
return res.value, res.err
|
|
}
|
|
|
|
//!-
|