You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
418 B
17 lines
418 B
3 years ago
|
// @flow
|
||
|
|
let weakMemoize = function <Arg, Return>(func: Arg => Return): Arg => Return {
|
||
|
|
// $FlowFixMe flow doesn't include all non-primitive types as allowed for weakmaps
|
||
|
|
let cache: WeakMap<Arg, Return> = new WeakMap()
|
||
|
|
return arg => {
|
||
|
|
if (cache.has(arg)) {
|
||
|
|
// $FlowFixMe
|
||
|
|
return cache.get(arg)
|
||
|
|
}
|
||
|
|
let ret = func(arg)
|
||
|
|
cache.set(arg, ret)
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export default weakMemoize
|