React Hooks與setInterval

前言

Hooks出來已經有段時間了，相信大家都用過段時間了，有沒有小夥伴們遇到坑呢，我這邊就有個setInterval的坑，和小夥伴們分享下解決方案。

前言

寫個count每秒自增的定時器，如下寫法結果，界面上count爲1？

function Counter() {
  let [count, setCount] = useState(0);
  useEffect(() => {
    let id = setInterval(() => {
      setCount(count + 1);
    }, 1000);
    return () => clearInterval(id);
  }, []);
  return <h1>{count}</h1>;
}

https://codesandbox.io/embed/hooks-setinterval-error-w4qu6

如果某些特定值在兩次重渲染之間沒有發生變化，你可以通知 React 跳過對 effect 的調用。就是將第二個參數改成[],類似於更接近類組件的 componentDidMount 和 componentWillUnmount 生命週期，只執行一次。 effect的第二個參數中傳入的值就是 它更改的話， effect也會重新執行一遍的值。

因爲Effect的第二個參數爲[]，沒有依賴,Effect只會執行一次。setInterval中拿到的count永遠是0,界面會一直顯示1,如下所示：

function Counter() {
  let [count, setCount] = useState(0);
  useEffect(() => {
    let id = setInterval(() => {
      setCount(0 + 1);
    }, 1000);
    return () => clearInterval(id);
  }, []);
 return <h1>{count}</h1>;
}

那有些小夥伴會說，如果我們直接往第二個參數加count呢

function Counter() {
//... 
useEffect(() => {
    let id = setInterval(() => {
      setCount(count + 1);
    }, 1000);
    return () => clearInterval(id);
  }, [count]);
//...
}

這樣效果是對的，但是性能不好。每當count更改了，useEffect就會渲染一次,定時器也會不停的被新增與移除。如下所示：

//第一次
function Counter() {
//... 
useEffect(() => {
    let id = setInterval(() => {
      setCount(0 + 1);
    }, 1000);
    return () => clearInterval(id);
  }, [0]);
//...
}
//第二次
function Counter() {
//... 
useEffect(() => {
    let id = setInterval(() => {
      setCount(1 + 1);
    }, 1000);
    return () => clearInterval(id);
  }, [1]);
//...
//第N次
}

那到底要怎麼做纔能有保障性能，定時器只監聽一次，又使定時器起作用呢？

方案一、函數式更新

useState 中的set方法可接收函數，該函數將接收先前的state，並返回一個更新後的值。這樣定時器每次拿到的是最新的值。

function Counter() {
let [count, setCount] = useState(0);
useEffect(() => {
    let id = setInterval(() => {
      setCount(v => {
        return v + 1;
      });
    }, 1000);
    return () => clearInterval(id);
  }, []);
return <h1>{count}</h1>;
}

https://codesandbox.io/embed/hooks-setinterval-usestate-grres

方案二、用useRef

useRef返回一個可變的ref對象,返回的ref對象在組件的整個生命週期內保持不變。
將定時器函數提取出來，每次定時器觸發時，都能取到最新到count.

function Counter() {
  let [count, setCount] = useState(0);
  const myRef = useRef(null);
  myRef.current = () => {
    setCount(count + 1);
  };
  useEffect(() => {
    let id = setInterval(() => {
      myRef.current();
    }, 1000);
    return () => clearInterval(id);
  }, []);
  return <h1>{count}</h1>;
}

https://codesandbox.io/embed/hooks-setinterval-useref-cgif3

思考：爲什麼不直接setInterval(myRef.current, 1000)這樣寫不行呢,還要包個方法返回？

function Counter() {
  let [count, setCount] = useState(0);
  const myRef = useRef(null);
  myRef.current = () => {
    setCount(count + 1);
  };
  useEffect(() => {
    let id = setInterval(myRef.current, 1000);
    return () => clearInterval(id);
  }, []);
 return <h1>{count}</h1>;
}

https://codesandbox.io/embed/hooks-setinterval-useref-error-52dm0

下面的例子可以很好的解釋。假如把myRef.current爲cur變量,定時器的第一個參數爲interval變量，cur變量更改，interval的取的還是之前賦值的值。

var cur=()=>{var count=0;console.log(count)};
var interval=cur;
var cur=()=>{var count=1;console.log(count)};
interval();//0

var cur=()=>{var count=0;console.log(count)};
var interval=()=>{cur()};
var cur=()=>{var count=1;console.log(count)};
interval();//1

方案三、自定義hook

可以寫個自定義hook，方便重複使用。

function useInterval(fun) {
  const myRef = useRef(null);
  useEffect(() => {
    myRef.current = fun;
  }, [fun]);
  useEffect(() => {
    let id = setInterval(() => {
      myRef.current();
    }, 1000);
    return () => clearInterval(id);
  }, []);
}

function Counter() {
  let [count, setCount] = useState(0);
  useInterval(() => {
    setCount(count + 1);
  });
  return <h1>{count}</h1>;
}

https://codesandbox.io/embed/hooks-setinterval-ownhooks-0tpxe

方案四、用useReducer

將count變量存入reducer中，使用useReducer更新count

function reducer(state, action) {
  switch (action.type) {
    case "increment":
      return state + 1;
    default:
      throw new Error();
  }
}

function Counter() {
  const [state, dispatch] = useReducer(reducer, 0);
  useEffect(() => {
    setInterval(() => {
      dispatch({ type: "increment" });
    }, 1000);
  }, []);
  return <h1>{state}</h1>;
}

https://codesandbox.io/embed/hooks-setinterval-usereducer-2byrm

還有什麼好的方案歡迎小夥伴們留言評論～～

Happy coding .. :)

相關鏈接

https://raoenhui.github.io/react/2019/11/07/hooksSetinterval/index.html

https://zh-hans.reactjs.org/docs/hooks-effect.html

https://zh-hans.reactjs.org/docs/hooks-reference.html

https://overreacted.io/making-setinterval-declarative-with-react-hooks/