如何在React中优雅的使用Interval(轮询)

在前端开发中,经常会使用轮询(setInterval),比如后端异步数据处理,需要定时查询最新状态。但是在用React Hook进行轮询操作时,可能会发现setInterval没有那么轻松驾驭,今天笔者就来谈谈在项目开发中是如何解决setInterval调用问题的,以及如何更加优雅的使用setInterval

问题的引入

先从一个简单的例子开始,为了便于叙述,本文中的案例用一个计数定时器来演示。

import React, { useEffect, useState } from "react";

export default function IntervalExp() {
  const [count, setCount] = useState(0);
  useEffect(() => {
    const timer = setInterval(() => {
      setCount(count + 1);
    }, 1000);
    return () => clearInterval(timer);
  }, []);
  return (
    

当前计数:{count}

); }

首先使用useState定义了一个count变量,然后在useEffect中,定义了一个名为timer的定时器,并在定时器中执行count+1操作,并在组件卸载时清除定时器。

理想状态下,count会执行+1操作,并不断的递增。但实际并非如此,count在变为1以后,将不再有任何变化。原因很简单,useEffect中由于没有将依赖的count对象添加到依赖对象数组中,所以它每次拿到的都是老的count对象,也就是0。

方法一:添加依赖数组

import React, { useEffect, useState } from "react";

export default function IntervalExp() {
  const [count, setCount] = useState(0);
  useEffect(() => {
    const timer = setInterval(() => {
      setCount(count + 1);
    }, 1000);
    console.log("更新了", timer);
    return () => clearInterval(timer);
  }, [count]);
  return (
    

当前计数{count}

); }

当把count对象加入到依赖数组以后,可以发现定时器现在可以正常工作了。但是注意这里有个坑,在return的时候,即组件卸载的时候,一定要做清理操作,否则你的定时器会执行的越来越快,因为新的定时器会不断生成,但老的定时器却没有清理

但是这种方式完美吗?并不然,如果定时器操作的数据包含父组件传递的props,或者是其他的state,都需要加到依赖数组中,这样做不仅不美观,而且容易出错。同时,这种方式还有个问题,就是定时器要在每次变化时要重新生成,这必然也会有很高的性能损耗

方法二:不添加依赖数组的方式(useRef)

useRef是官方的hook,使用useRef定义的对象有个current对象,是可以存储数据的,而且存储的数据可以被修改,并在组件的每一次渲染中,都能从current中拿到最新的数据。基于ref的这一特性,实现一个名为useInterval的自定义hook。

import { useEffect, useRef } from "react";

export const useInterval = (cb: Function, time = 1000) => {
  const cbRef = useRef();
  useEffect(() => {
    cbRef.current = cb;
  });
  useEffect(() => {
    const callback = () => {
      cbRef.current?.();
    };
    const timer = setInterval(() => {
      callback();
    }, time);
    return () => clearInterval(timer);
  }, []);
};

在这个自定义hook中,有回调函数和轮询时间两个参数。使用useEffect把最新的回调函数赋值给ref.current,这样在第二个useEffect中就能从ref.current上拿到最新的callback,然后在定时器中执行它。
在项目中如何使用呢?

useInterval(() => {
    setCount(count + 1);
  }, 1000);

只需引入自定义hook,并按照上面的格式调用即可。

方法三:更高级的办法(useReducer)☆☆☆

回头看第一个例子,为什么在useEffect中不添加count就无法实现想要的定时器效果呢,说白了是因为读取了state的数据,而又因为闭包原因拿不到最新的count数据,所以导致interval操作失败。其实借助useReducer就可以在不读取count的情况更新count数据。

import React, { useEffect, useReducer } from "react";

function reducer(state: { count: number }) {
  return { count: state.count + 1 };
}
export default function IntervalExp() {
  const [state, dispatch] = useReducer(reducer, { count: 0 });
  useEffect(() => {
    setInterval(() => {
      dispatch();
    }, 1000);
  }, []);

  return (
    

当前计数{state.count}

); }

在这个案例中,使用useReducer定义了一个简单的count操作方法,在interval中,通过调用dispatch方法,成功更新了count数据。useReducer在需要操作多个state的复杂业务逻辑场景下可以使用,虽然定义起来麻烦,但是可以实现将组件中的业务逻辑抽离出来,写出更加易于维护的代码,而且在目前这个场景中,useReducer比上面两个方式处理的更加优雅,也是本文推荐的方式。

总结

hook是React中非常有魅力的一个发明,灵活使用hook可以写出更有品质的代码。作者写本文的目的也是因为在实际开发中遇到了这一问题,因此希望本文可以帮助到其他开发者。

参考文章:usestate中的回调函数_React Hooks 中使用 setInterval 的若干方法

你可能感兴趣的