30秒学会 React 片段 – useTimeout

A hook that implements setTimeout in a declarative manner.

• Create a custom hook that takes a callback and a delay.
• Use the React.useRef() hook to create a ref for the callback function.
• Use the React.useEffect() hook to remember the latest callback.
• Use the React.useEffect() hook to set up the timeout and clean up.

代码实现

const useTimeout = (callback, delay) => {
  const savedCallback = React.useRef();

  React.useEffect(() => {
    savedCallback.current = callback;
  }, [callback]);

  React.useEffect(() => {
    function tick() {
      savedCallback.current();
    }
    if (delay !== null) {
      let id = setTimeout(tick, delay);
      return () => clearTimeout(id);
    }
  }, [delay]);
};

使用样例

const OneSecondTimer = props => {
  const [seconds, setSeconds] = React.useState(0);
  useTimeout(() => {
    setSeconds(seconds + 1);
  }, 1000);

  return <p>{seconds}</p>;
};

ReactDOM.render(<OneSecondTimer />, document.getElementById('root'));