防抖（debounce）是前端经常用到的一个工具函数，也是我在面试中必问的一个问题。团队内部推广React hooks以后，我在面试中也加入了相关的题目。如何实现一个useDebounce这个看起来很基础的问题，实际操作起来却让很多背代码的小伙伴漏出马脚。
问题的安排往往是这样的：
- 什么是防抖、节流，分别解释一下？
- 在白纸上手写一个防抖or节流函数，自己任选（限时4分钟）
- react hooks有了解吗？上机实现一个useDebounce、useThrottle
- tyepscript有了解吗？用ts再来写一遍
- ……
围绕一个主题不断切换考察点，这样一轮下来，轻松又流畅，同时可以试探出很多信息。
实际情况是，很多候选人在第3题就卡住了，不得不说很可惜。
场景还原
写一个防抖函数
一个经典的防抖函数可能是这样的：
function debounce(fn, ms) {
let timer;
return function(...args) {
if (timer) {
clearTimeout(timer)
}
timer = setTimeout(() => {
fn(...args)
timer = null;
}, ms);
}
}
复制代码改成react hooks
先提供测试用例：
export default function() {
const [counter, setCounter] = useState(0);
const handleClick = useDebounce(function() {
setCounter(counter + 1)
}, 1000)
return <div style={{ padding: 30 }}>
<Button
onClick={handleClick}
>click</Button>
<div>{counter}</div>
</div>
}
复制代码很多小伙伴会想当然的就改成这样：
function useDebounce(fn, time) {
return debounce(fn, time);
}
复制代码简单、优雅，还复用了刚才的代码，测试一下，看起来并没有什么问题：
1-7292504.gif
但是这个代码如果放上生产环境，你会被用户锤死。
真的吗？
换个用例来试一下：
export default function() {
const [counter1, setCounter1] = useState(0);
const [counter2, setCounter2] = useState(0);
const handleClick = useDebounce(function() {
console.count('click1')
setCounter1(counter1 + 1)
}, 500)
useEffect(function() {
const t = setInterval(() => {
setCounter2(x => x + 1)
}, 500);
return clearInterval.bind(undefined, t)
}, [])
return <div style={{ padding: 30 }}>
<Button
onClick={function() {
handleClick()
}}
>click</Button>
<div>{counter1}</div>
<div>{counter2}</div>
</div>
}
复制代码2-7292504.gif
当引入一个自动累加counter2就开始出问题了。这时很多候选人就开始懵了，有的候选人会尝试分析原因。只有深刻理解react hooks在重渲染时的工作原理才能快速定位到问题（事实上出错不要紧，能够快速定位问题的小伙伴才是我们苦苦寻找的）。
有的候选人开启胡乱调试大法，慌忙修改setCounter1：
const handleClick = useDebounce(function() {
console.count('click1')
setCounter1(x => x + 1)
}, 500)
复制代码当然结果依然错误，而且暴漏了自己对react hooks特性不够熟悉的问题……
有的候选人猜到是重渲染缓存的问题，于是写成这样：
function useDebounce(fn, delay) {
return useCallback(debounce(fn, delay), [])
}
复制代码在配合setCounter1(x => x + 1)修改的情况下，可以得到正确的结果。但并没有正确解决问题。依然是错误的。有兴趣的读者可以复现一下这个现象，思考一下为什么，欢迎留言讨论。
问题出在哪里？
我们在useDebounce里面加个log
function useDebounce(fn, time) {
console.log('usedebounce')
return debounce(fn, time);
}
复制代码3-7292504.gif
控制台开始疯狂的输出log。看到这里，很多读者就明白了。如果是前面表现稍好的候选人，我可以提示到此。
每次组件重新渲染，都会执行一遍所有的hooks，这样debounce高阶函数里面的timer就不能起到缓存的作用（每次重渲染都被置空）。timer不可靠，debounce的核心就被破坏了。
如何调整？
修复这个问题可以有很多办法。比如利用React组件的缓存机制：
function useDebounce(fn, delay, dep = []) {
const { current } = useRef({ fn, timer: null });
useEffect(function () {
current.fn = fn;
}, [fn]);
return useCallback(function f(...args) {
if (current.timer) {
clearTimeout(current.timer);
}
current.timer = setTimeout(() => {
current.fn.call(this, ...args);
}, delay);
}, dep)
}
复制代码就可以实现一个可靠的useDebounce。
同理我们直接给出useThrottle的代码：
function useThrottle(fn, delay, dep = []) {
const { current } = useRef({ fn, timer: null });
useEffect(function () {
current.fn = fn;
}, [fn]);
return useCallback(function f(...args) {
if (!current.timer) {
current.timer = setTimeout(() => {
delete current.timer;
}, delay);
current.fn.call(this, ...args);
}
}, dep);
}