记忆化高阶函数通过闭包缓存计算结果,基础版用json.stringify序列化原始类型参数,增强版支持date、regexp等复杂类型并处理异常,可配置版支持lru清理和自定义key生成器。

手写一个具备记忆功能(即“记忆化”)的高阶函数,核心是缓存已计算过的输入与对应结果,避免重复运算。关键在于:用闭包保存缓存对象,用参数序列化作为缓存键(尤其处理对象/数组),并正确返回缓存值或执行原函数。
基础版本:只支持原始类型参数
适用于参数全是字符串、数字、布尔值、null、undefined 的场景,可直接用 JSON.stringify 或模板字符串拼接作 key:
function memoize(fn) {
const cache = new Map();
return function(...args) {
const key = JSON.stringify(args);
if (cache.has(key)) {
return cache.get(key);
}
const result = fn.apply(this, args);
cache.set(key, result);
return result;
};
}
- 用
Map比普通对象更安全(避免原型污染、支持任意类型 key) -
JSON.stringify对原始类型和简单嵌套对象有效,但对函数、undefined、Symbol、Date、RegExp 会丢失或转为空 - 注意:
this上下文被保留,通过fn.apply(this, args)正确绑定
增强版本:支持常见复杂类型(推荐日常使用)
用轻量级参数序列化策略,覆盖函数常用输入类型(含对象、数组、Date、RegExp 等):
function memoize(fn) {
const cache = new Map();
function serialize(arg) {
if (arg === null || typeof arg === 'boolean' || typeof arg === 'number' || typeof arg === 'string') {
return String(arg);
}
if (typeof arg === 'function') {
return arg.toString();
}
if (arg instanceof Date) {
return `Date(${arg.getTime()})`;
}
if (arg instanceof RegExp) {
return `RegExp(${arg.source},${arg.flags})`;
}
if (typeof arg === 'object') {
try {
return JSON.stringify(arg, (k, v) => {
if (v instanceof Date) return `Date(${v.getTime()})`;
if (v instanceof RegExp) return `RegExp(${v.source},${v.flags})`;
if (typeof v === 'function') return v.toString();
return v;
});
} catch {
return String(arg);
}
}
return String(arg);
}
return function(...args) {
const key = args.map(serialize).join('|||');
if (cache.has(key)) {
return cache.get(key);
}
const result = fn.apply(this, args);
cache.set(key, result);
return result;
};
}
- 对 Date、RegExp、Function 做显式处理,避免 JSON 序列化丢失信息
- 用
|||分隔参数,比纯 JSON 更易读、冲突概率更低(实际项目可用更健壮分隔符如\x00) - 捕获
JSON.stringify异常(如循环引用),降级为String(arg)
可配置版本:支持最大缓存数与自定义 key 生成器
适合需要控制内存占用或有特殊缓存策略的场景:
function memoize(fn, options = {}) {
const { maxCache = Infinity, getKey = (...args) => JSON.stringify(args) } = options;
const cache = new Map();
return function(...args) {
const key = getKey(...args);
if (cache.has(key)) {
// 命中则移到末尾(模拟 LRU 行为)
const value = cache.get(key);
cache.delete(key);
cache.set(key, value);
return value;
}
const result = fn.apply(this, args);
cache.set(key, result);
if (cache.size > maxCache) {
// 删除最老的一项(Map 保证插入顺序)
const firstKey = cache.keys().next().value;
cache.delete(firstKey);
}
return result;
};
}
// 使用示例:限制最多缓存 100 项
const expensiveCalc = memoize(compute, { maxCache: 100 });
// 自定义 key:忽略某个参数
const memoizedWithCustomKey = memoize(fetchUser, {
getKey: (id, ignoreThis) => `user-${id}`
});
- 通过
maxCache防止内存无限增长 - 提供
getKey钩子,便于按需定制缓存粒度(比如忽略调试参数、合并相似请求) - 利用 Map 的插入顺序特性实现简易 LRU 清理逻辑
注意事项与边界情况
记忆化不是万能的,使用前需确认是否适用:
- 副作用函数不能记忆化:如果原函数依赖时间、随机数、DOM 状态或发起请求,缓存会导致行为异常
-
引用相等 ≠ 值相等:两个内容相同的对象
{a:1}和{a:1}序列化后 key 相同,但若需区分实例身份(如 React 中的 ref),应改用WeakMap+ 参数引用作为 key(仅限单参数且为对象) -
异步函数需单独处理:上述版本不支持
async函数;如需记忆 Promise 结果,应缓存 Promise 实例本身(避免重复触发),而非await后的值 - 慎用于高频调用小函数:序列化开销可能超过计算本身,简单加法、取长度等无需记忆化
Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南











