新聞中心
useSetState
管理 object 類型 state 的 Hooks,用法與 class 組件的 this.setState 基本一致。

先來了解一下可變數(shù)據(jù)和不可變數(shù)據(jù)的含義和區(qū)別如下:
- 可變數(shù)據(jù)(mutable)即一個數(shù)據(jù)被創(chuàng)建之后,可以隨時進(jìn)行修改,修改之后會影響到原值。
- 不可變數(shù)據(jù)(Immutable) 就是一旦創(chuàng)建,就不能再被更改的數(shù)據(jù)。對Immutable? 對象的任何修改或添加刪除操作都會返回一個新的Immutable 對象。
我們知道,React Function Components 中的 State 是不可變數(shù)據(jù)。所以我們經(jīng)常需要寫類似如下的代碼:
setObj((prev) => ({
...prev,
name: 'Gopal',
others: {
...prev.others,
age: '27',
}
}));通過 useSetState,可以省去對象擴(kuò)展運(yùn)算符操作這個步驟,即:
setObj((prev) => ({
name: 'Gopal',
others: {
age: '27',
}
}));其內(nèi)部實(shí)現(xiàn)也比較簡單,如下所示:
- 調(diào)用設(shè)置值方法的時候,會根據(jù)傳入的值是否為函數(shù)。如果是函數(shù),則入?yún)榕f狀態(tài),輸出新的狀態(tài)。否則直接作為新狀態(tài)。這個符合 setState 的使用方法。
- 使用對象拓展運(yùn)算符,返回新的對象,保證原有數(shù)據(jù)不可變。
const useSetState =>(
initialState: S | (() => S),
): [S, SetState] => {
const [state, setState] = useState(initialState);
// 合并操作,并返回一個全新的值
const setMergeState = useCallback((patch) => {
setState((prevState) => {
// 新狀態(tài)
const newState = isFunction(patch) ? patch(prevState) : patch;
// 也可以通過類似 Object.assign 的方式合并
// 對象拓展運(yùn)算符,返回新的對象,保證原有數(shù)據(jù)不可變
return newState ? { ...prevState, ...newState } : prevState;
});
}, []);
return [state, setMergeState];
};
可以看到,其實(shí)就是將對象拓展運(yùn)算符的操作封裝到內(nèi)部。
還有其他更優(yōu)雅的方式?我們可以使用 use-immer[1]
useImmer(initialState) 非常類似于 useState。該函數(shù)返回一個元組,元組的第一個值是當(dāng)前狀態(tài),第二個是 updater 函數(shù),它接受一個 immer producer 函數(shù)或一個值作為參數(shù)。
使用如下:
const [person, updatePerson] = useImmer({
name: "Michel",
age: 33
});
function updateName(name) {
updatePerson(draft => {
draft.name = name;
});
}
function becomeOlder() {
updatePerson(draft => {
draft.age++;
});
}當(dāng)向更新函數(shù)傳遞一個函數(shù)的時候,draft 參數(shù)可以自由地改變,直到 producer 函數(shù)結(jié)束,所做的改變將是不可變的,并成為下一個狀態(tài)。這更符合我們的使用習(xí)慣,可以通過 draft.xx.yy 的方式更新我們對象的值。
useBoolean 和 useToggle
這兩個都是特殊情況下的值管理。
useBoolean,優(yōu)雅的管理 boolean 狀態(tài)的 Hook。
useToggle,用于在兩個狀態(tài)值間切換的 Hook。
實(shí)際上,useBoolean 又是 useToggle 的一個特殊使用場景。
先看 useToggle。
- 這里使用了 typescript 函數(shù)重載聲明入?yún)⒑统鰠㈩愋?,根?jù)不同的入?yún)祷夭煌慕Y(jié)果。比如第一個入?yún)?boolean 布爾值,則返回一個元組,第一項為 boolean 值,第二個為更新函數(shù)。優(yōu)先級從上到下依次變低。
- 入?yún)⒖赡苡袃蓚€值,第一個為默認(rèn)值(認(rèn)為是左值),第二個是取反之后的值(認(rèn)為是右值),可以不傳,不傳的時候,則直接根據(jù)默認(rèn)值取反!defaultValue。
- toggle 函數(shù)。切換值,也就是上面的左值和右值的轉(zhuǎn)換。
- set。直接設(shè)置值。
- setLeft。設(shè)置默認(rèn)值(左值)。
- setRight。如果傳入了 reverseValue, 則設(shè)置為 reverseValue。否則設(shè)置為 defautValue 的取反值。
// TS 函數(shù)重載的使用
function useToggle(): [boolean, Actions ];
function useToggle(defaultValue: T): [T, Actions ];
function useToggle(defaultValue: T, reverseValue: U): [T | U, Actions ];
function useToggle(
// 默認(rèn)值
defaultValue: D = false as unknown as D,
// 取反
reverseValue?: R,
) {
const [state, setState] = useState(defaultValue);
const actions = useMemo(() => {
const reverseValueOrigin = (reverseValue === undefined ? !defaultValue : reverseValue) as D | R;
// 切換 state
const toggle = () => setState((s) => (s === defaultValue ? reverseValueOrigin : defaultValue));
// 修改 state
const set = (value: D | R) => setState(value);
// 設(shè)置為 defaultValue
const setLeft = () => setState(defaultValue);
// 如果傳入了 reverseValue, 則設(shè)置為 reverseValue。 否則設(shè)置為 defautValue 的反值
const setRight = () => setState(reverseValueOrigin);
return {
toggle,
set,
setLeft,
setRight,
};
// useToggle ignore value change
// }, [defaultValue, reverseValue]);
}, []);
return [state, actions];
}
而 useBoolean 是對 useToggle 的一個使用。如下,比較簡單,不細(xì)說
export default function useBoolean(defaultValue = false): [boolean, Actions] {
const [state, { toggle, set }] = useToggle(defaultValue);
const actions: Actions = useMemo(() => {
const setTrue = () => set(true);
const setFalse = () => set(false);
return {
toggle,
set: (v) => set(!!v),
setTrue,
setFalse,
};
}, []);
return [state, actions];
}usePrevious
保存上一次狀態(tài)的 Hook。
其原理,是每次狀態(tài)變更的時候,比較值有沒有發(fā)生變化,變更狀態(tài):
- 維護(hù)兩個狀態(tài) prevRef(保存上一次的狀態(tài))和 curRef(保存當(dāng)前狀態(tài))。
- 狀態(tài)變更的時候,使用 shouldUpdate 判斷是否發(fā)生變化,默認(rèn)通過Object.is 判斷。開發(fā)者可以自定義 shouldUpdate 函數(shù),并決定什么時候記錄上一次狀態(tài)。
- 狀態(tài)發(fā)生變化,更新 prevRef 的值為上一個 curRef,并更新 curRef 為當(dāng)前的狀態(tài)。
const defaultShouldUpdate =(a?: T, b?: T) => !Object.is(a, b);
function usePrevious(
state: T,
shouldUpdate: ShouldUpdateFunc= defaultShouldUpdate,
): T | undefined {
// 使用了 useRef 的特性,一直保持引用不變
// 保存上一次值
const prevRef = useRef();
// 當(dāng)前值
const curRef = useRef();
// 自定義是否更新上一次的值
if (shouldUpdate(curRef.current, state)) {
prevRef.current = curRef.current;
curRef.current = state;
}
return prevRef.current;
}
useRafState
只在 requestAnimationFrame callback 時更新 state,一般用于性能優(yōu)化。
window.requestAnimationFrame() 告訴瀏覽器——你希望執(zhí)行一個動畫,并且要求瀏覽器在下次重繪之前調(diào)用指定的回調(diào)函數(shù)更新動畫。該方法需要傳入一個回調(diào)函數(shù)作為參數(shù),該回調(diào)函數(shù)會在瀏覽器下一次重繪之前執(zhí)行。
假如你的操作是比較頻繁的,就可以通過這個 hook 進(jìn)行性能優(yōu)化。
- 重點(diǎn)看 setRafState 方法,它執(zhí)行的時候,會取消上一次的 setRafState 操作。重新通過 requestAnimationFrame 去控制 setState 的執(zhí)行時機(jī)。
- 另外在頁面卸載的時候,會直接取消操作,避免內(nèi)存泄露。
function useRafState(initialState?: S | (() => S)) {
const ref = useRef(0);
const [state, setState] = useState(initialState);
const setRafState = useCallback((value: S | ((prevState: S) => S)) => {
cancelAnimationFrame(ref.current);
ref.current = requestAnimationFrame(() => {
setState(value);
});
}, []);
// unMount 的時候,去除監(jiān)聽
useUnmount(() => {
cancelAnimationFrame(ref.current);
});
return [state, setRafState] as const;
}
useSafeState
用法與 React.useState 完全一樣,但是在組件卸載后異步回調(diào)內(nèi)的 setState 不再執(zhí)行,避免因組件卸載后更新狀態(tài)而導(dǎo)致的內(nèi)存泄漏。
代碼如下:
在更新的時候,通過 useUnmountedRef 判斷如果組件卸載,則停止更新。
function useSafeState(initialState?: S | (() => S)) {
// 判斷是否卸載
const unmountedRef = useUnmountedRef();
const [state, setState] = useState(initialState);
const setCurrentState = useCallback((currentState) => {
// 如果組件卸載,則停止更新
if (unmountedRef.current) return;
setState(currentState);
}, []);
return [state, setCurrentState] as const;
}
useUnmountedRef 這個我們之前提過,簡單回顧下,其實(shí)就是在 hook 的返回值中標(biāo)記組件為已卸載。
const useUnmountedRef = () => {
const unmountedRef = useRef(false);
useEffect(() => {
unmountedRef.current = false;
// 如果已經(jīng)卸載,則會執(zhí)行 return 中的邏輯
return () => {
unmountedRef.current = true;
};
}, []);
return unmountedRef;
};useGetState
給 React.useState 增加了一個 getter 方法,以獲取當(dāng)前最新值。
其實(shí)現(xiàn)如下:
其實(shí)就是通過 useRef 記錄最新的 state 的值,并暴露一個 getState 方法獲取到最新的。
function useGetState(initialState?: S) {
const [state, setState] = useState(initialState);
// useRef 返回一個可變的 ref 對象,其 .current 屬性被初始化為傳入的參數(shù)(initialValue)。返回的 ref 對象在組件的整個生命周期內(nèi)持續(xù)存在。
// 使用 useRef 處理 state
const stateRef = useRef(state);
stateRef.current = state;
const getState = useCallback(() => stateRef.current, []);
return [state, setState, getState];
}
這在某一些情況下,可以避免 React 的閉包陷阱。如官網(wǎng)例子:
const [count, setCount, getCount] = useGetState(0);
useEffect(() => {
const interval = setInterval(() => {
console.log('interval count', getCount());
}, 3000);
return () => {
clearInterval(interval);
};
}, []);
假如這里不使用 getCount(),而是直接使用 count,是獲取不到最新的值的。
總結(jié)與思考
React 的 function Component 的狀態(tài)管理還是比較靈活,我們可以針對一些場景進(jìn)行封裝和優(yōu)化,從而更優(yōu)雅的管理我們的 state 狀態(tài),希望 ahooks 這些封裝能對你有所幫助。
參考資料
[1]use-immer: https://github.com/immerjs/use-immer
新聞名稱:這些Hook更優(yōu)雅地管理你的狀態(tài)
網(wǎng)站鏈接:http://fisionsoft.com.cn/article/dhdpcej.html


咨詢
建站咨詢
