04 React Hook 定时器

前段时间学习了 React 新发布的 Hook 新功能,学完立马就爱上了这个新 API,用起来感觉比 class 组件爽多了。但 hook 虽然看似简单,但是要能熟练运用还是得花上一段时间。

随机间隔计数器

目标:实现一个计数器,通过输入框输入计数器每次计数的时间(ms),按输入的时间间隔每次增一

class 版本

如果熟悉 React class 组件模式,这个功能实现不难。但是这样有两个弊端:

  1. 会有大量模板代码,我们不得不通过多个生命周期来实现一个功能

  2. 同一个功能逻辑代码却分布在不同地方,比如创建定时器和清除定时器放在两个不同的生命周期

import React from 'react'
class ClassVersion extends React.Component {
constructor(props) {
super(props)
this.state = {
count: 0,
delay: 1000
}
}
componentDidMount() {
this.timer = setInterval(this.tick, this.state.delay)
}
componentDidUpdate(prevProps, prevState) {
if (prevState.delay !== this.state.delay) {
clearInterval(this.timer)
this.timer = setInterval(this.tick, this.state.delay)
}
}
componentWillUnmount() {
clearInterval(this.timer)
}
handelDelayChange = e => {
this.setState({
delay: +e.target.value
})
}
tick = () => {
this.setState({
count: this.state.count + 1
})
}
render() {
const { count, delay } = this.state
return (
<>
<h1>Class 版本定时器</h1>
<Input onChange={this.handelDelayChange} defaultValue={delay} />
<div>
Count: {count} Delay: {delay}ms
</div>
</>
)
}
}

Hook 版本

通过 Hook 实现计数器,代码更加已读,而且功能逻辑代码集中,创建计数器和清空计数器的代码集中一起。

import React, { useState, useEffect } from 'react'
function HookVersion() {
const [count, setCount] = useState(0)
const [delay, setDelay] = useState(1000)
useEffect(() => {
const timer = setInterval(() => {
setCount(c => c + 1)
}, delay)
return () => clearInterval(timer)
}, [delay])
const handelDelayChange = e => setDelay(+e.target.value)
return (
<>
<h1>Hook 版本定时器</h1>
<Input onChange={handelDelayChange} defaultValue={delay} />
<div>
Count: {count} Delay: {delay}ms
</div>
</>
)
}

上面的 Hook 代码能够正常运行,可以注意到我们在每次计数器新增的时候调用的是 setCount(c => c + 1),传入的参数是一个函数 c => c + 1,也就是接收之前的值然后每次增一,而不是 setCount(count + 1),可以明确的是如果传入 count + 1 是无法正常工作的,count 会被固定为 0,所以计数器增加到 1 的时候就是停止不动,每次都是计数为 1。

count 被固定的原因是在 delay 不发生改变的情况下 effect 并不会重复执行,定时器每次 setCount 读取到的都是初始值。

虽然通过传入函数而不是固定值可以解决 count 被固定的问题,但是却无法读取每次渲染时期的 props。如何解决呢?可以通过在每次计数的时候不改变定时器,但是动态指向定时器的回调。

useRef() 返回了一个字面量,持有一个可变的 current 属性,在每一次渲染之间共享。可以将 Ref Hook 看作是一个容器,.current 属性可以指向任何值,类似于 class 组件的 this。通过使用 ref 来保存每次定时器回调函数。

React 组件的 props 和 state 会变化时,都会被重新渲染,并且把之前的渲染结果“忘记”的一干二净。两次渲染之间,是互不相干的。

useEffect() Hook 同样会“遗忘”之前的结果。它清理上一个 effect 并且设置新的 effect。新的 effect 获取到了新的 props 和 state。

但是 setInterval() 不会 “忘记”。 它会一直引用着旧的 props 和 state,除非把它换了。但是只要把它换了,就不得不重新设置时间。

但是通过 ref 我们可以做到只更换定时器的回调而不改变定时器的时间:

  1. 设置计时器 setInterval(fn, delay),其中 fn 调用 cb 回调。

  2. 第一次渲染,设置回调为 cb1

  3. 第二次渲染,设置回调为 cb2

import React, { useState, useEffect, useRef } from 'react'
function HookVersion() {
const [count, setCount] = useState(0)
const [delay, setDelay] = useState(1000)
useInterval(() => {
setCount(count + 1)
}, delay)
function handelDelayChange(e) {
setDelay(+e.target.value)
}
return (
<>
<h1>Hook 版本定时器</h1>
<Input onChange={handelDelayChange} defaultValue={delay} />
<div>
Count: {count} Delay: {delay}ms
</div>
</>
)
}
function useInterval(cb, delay) {
const ref = useRef()
useEffect(() => {
ref.current = cb
})
useEffect(() => {
const timer = setInterval(() => ref.current(), delay)
return () => clearInterval(timer)
}, [delay])
}

这里获取提取出可复用的定时器 Hook 会更加优雅,而且 useIntervalsetInterval 使用方式一样:

setInterval(fn, delay)
useInterval(fn, delay)