在一个 action 被分发(dispatch) 后,useSelector() 默认对 select 函数的返回值进行引用比较 ===,并且仅在返回值改变时触发重渲染。但是,不同于 connect(),useSelector()并不会阻止父组件重渲染导致的子组件重渲染的行为,即使组件的 props 没有发生改变。
useSelector 源码分析
import { useContext, useEffect, useReducer, useRef, } from 'react';
import StoreContext from './context'; type EqualityFn
export default function useSelector<T, Selected extends unknown>( selector: (state: T) => Selected, equalityFn?: EqualityFn
const store = useContext(StoreContext);
// 注:react-redux@8-beta 中使用 React18提供的useSyncExternalStoreapi 来做强制更新。
// 17版本之前采用下面的方式进行强制render
const [, forceRender] = useReducer((s) => s + 1, 0);
// 保存最近一次从store中获取的state
const latestStoreState = useRef
// 保存最近一次通过selector函数返回的值 - 也就是useSelector 返回的值
const latestSelectedState = useRef
useEffect(() => {
// checkUpdate - 判断是否强制更新;
function checkUpdate() {
const newState = store.getState();
// state 没有改变,直接返回;发生在 reducer 中default的情况;
// 其他情况使用immer,都会返回一个新的state对象(本质还是浅拷贝)
if (newState === latestStoreState) return;
const newSelectedState = selector(newState);
// 默认的 equalityFn 进行的是 === 判断;
// 所以默认的equalityFn函数的 === 决定了,
// 1 我们不能通过selector函数去返回创建一个新的对象进行返回
// 2 利用浅拷贝的特性,尽量返回state上最小的粒度,保证当前selector返回的值没有改变时不会执行强制更新
if (!equalityFn) equalityFn = (a, b) => a === b;
// 对比新旧SelectedState 数据 是否全等,如果不等 则进行强制更新
if (!equalityFn(newSelectedState, latestSelectedState.current)) {
latestSelectedState.current = newSelectedState;
latestStoreState.current = newState;
forceRender();
}
}
// 执行dispatch({type:XXX}) ,会调用 store.subscribe进行监听
// 执行checkUpdate函数
const unsubscribe = store.subscribe(checkUpdate);
return () => unsubscribe();
}, [store]);
// useSelector 返回最新的seletor函数返回值
return latestSelectedState.current;
}
不同引用的探究
// 写法一
const { project,pages } = useSelector((state: IRootState) => state.prototype);
// 写法二
const project = useSelector((state: IRootState) => state.prototype.project);
const project = useSelector((state: IRootState) => state.prototype.project);
// 写法三 (传递一个equalityFn 函数进行判断 )
const { project,pages } = useSelector((state: IRootState) => state.prototype,equalityFn);
结论
1.写法一,不可取。因为reducer 源码中,会缓存上一次返回的oldSelectdState,和新获取的selectdState 进行全等判断。如果直接返回一个模块,一旦这个模块中任意值发生变化,整个模块值都会改变,即使我们使用到的project 和 pages 数据没有发生改变,也会让该组件执行重新更新。
2.写法二,可取。因为每一个useSelector 的 seletor注册函数返回值都是模块内最小粒度。如果这个值没有发生改变,就不会执行更新。多个useSelector 如果其中有多处需要更新的话,react中会进行批量更新,也只会强制更新一次,对性能不会有影响。
3.写法三,如果想要返回一个对象,可以传递一个equalityFn 进行深度比较,不采用 === 比较方式。缺点是,如果数据量很大或者嵌套很深,深度比较会有性能问题。