![2dab35ded2e94c93343c529717ab280e.png](https://i-blog.csdnimg.cn/blog_migrate/5ba905b551f6328af4e5a39891d5ef26.jpeg)
作者:ssh
转发链接:https://mp.weixin.qq.com/s/msXG28gG2rXNRzZz2HgR-w
前言
各位使用 react 技术栈的小伙伴都不可避免的接触过redux + react-redux的这套组合,众所周知 redux 是一个非常精简的库,它和 react 是没有做任何结合的,甚至可以在 vue 项目中使用。
redux 的核心状态管理实现其实就几行代码
function createStore(reducer) { let currentState let subscribers = [] function dispatch(action) { currentState = reducer(currentState, action); subscribers.forEach(s => s()) } function getState() { return currentState; } function subscribe(subscriber) { subscribers.push(subscriber) return function unsubscribe() { ... } } dispatch({ type: 'INIT' }); return { dispatch, getState, };}
它就是利用闭包管理了 state 等变量,然后在 dispatch 的时候通过用户定义 reducer 拿到新状态赋值给 state,再把外部通过 subscribe 的订阅给触发一下。
那 redux 的实现简单了,react-redux 的实现肯定就需要相对复杂,它需要考虑如何和 react 的渲染结合起来,如何优化性能。
目标
- 本文目标是尽可能简短的实现react-reduxv7 中的 hook 用法部分Provider,useSelector, useDispatch方法。(不实现connect方法)
- 可能会和官方版本的一些复杂实现不一样,但是保证主要的流程一致。
- 用 TypeScript 实现,并且能获得完善的类型提示。
预览
![29c75d0c7f9b31d15e1ec9fddbcdc0af.gif](https://i-blog.csdnimg.cn/blog_migrate/6c7bcb550af44d17296dcb8499230a4e.gif)
redux gif.gif
预览地址:
https://sl1673495.github.io/tiny-react-redux
源码地址:
https://github.com/sl1673495/tiny-react-redux
性能
说到性能这个点,自从 React Hook 推出以后,有了useContext和useReducer这些方便的 api,新的状态管理库如同雨后春笋般的冒了出来,其中的很多就是利用了Context做状态的向下传递。
举一个最简单的状态管理的例子
export const StoreContext = React.createContext();function App({ children }) { const [state, setState] = useState({}); return ( {children} );}function Son() { const { state } = useContext(StoreContext); return
state是{state.xxx}
;}
利用 useState 或者 useContext,可以很轻松的在所有组件之间通过 Context 共享状态。
但是这种模式的缺点在于 Context 会带来一定的性能问题,下面是 React 官方文档中的描述:
![af8280506d074dedb0dab8ffdb301922.png](https://i-blog.csdnimg.cn/blog_migrate/3b3d5f67e1fe2c1c5c42ae377520ca14.jpeg)
Context性能问题
想像这样一个场景,在刚刚所描述的 Context 状态管理模式下,我们的全局状态中有count和message两个状态分别给通过StoreContext.Provider向下传递
- Counter计数器组件使用了count
- Chatroom聊天室组件使用了message
而在计数器组件通过 Context 中拿到的 setState 触发了count改变的时候,
由于聊天室组件也利用useContext消费了用于状态管理的 StoreContext,所以聊天室组件也会被强制重新渲染,这就造成了性能浪费。
虽然这种情况可以用useMemo进行优化,但是手动优化和管理依赖必然会带来一定程度的心智负担,而在不手动优化的情况下,肯定无法达到上面动图中的重渲染优化。
那么react-redux作为社区知名的状态管理库,肯定被很多大型项目所使用,大型项目里的状态可能分散在各个模块下,它是怎么解决上述的性能缺陷的呢?接着往下看吧。
缺陷示例
在我之前写的类 vuex 语法的状态管理库react-vuex-hook[1]中,就会有这样的问题。因为它就是用了Context + useReducer的模式。
你可以直接在 在线示例[2] 这里,在左侧菜单栏选择需要优化的场景,即可看到上述性能问题的重现,优化方案也已经写在文档底部。
这也是为什么我觉得Context + useReducer的模式更适合在小型模块之间共享状态,而不是在全局。
使用
本文的项目就上述性能场景提炼而成,由
- 聊天室组件,用了 store 中的count
- 计数器组件,用了 store 中的message
- 控制台组件,用来监控组件的重新渲染。
redux 的定义
redux 的使用很传统,跟着官方文档对于 TypeScript 的指导走起来,并且把类型定义和 store 都 export 出去。
import { createStore } from 'redux';type AddAction = { type: 'add';};type ChatAction = { type: 'chat'; payload: string;};type LogAction = { type: 'log'; payload: string;};const initState = { message: 'Hello', logs: [] as string[],};export type ActionType = AddAction | ChatAction | LogAction;export type State = typeof initState;function reducer(state: State, action: ActionType): State { switch (action.type) { case 'add': return { ...state, count: state.count + 1, }; case 'chat': return { ...state, message: action.payload, }; case 'log': return { ...state, logs: [action.payload, ...state.logs], }; default: return initState; }}export const store = createStore(reducer);
在组件中使用
import React, { useState, useCallback } from 'react';import { Card, Button, Input } from 'antd';import { Provider, useSelector, useDispatch } from '../src';import { store, State, ActionType } from './store';import './index.css';import 'antd/dist/antd.css';function Count() { const count = useSelector((state: State) => state.count); const dispatch = useDispatch(); // 同步的add const add = useCallback(() => dispatch({ type: 'add' }), []); dispatch({ type: 'log', payload: '计数器组件重新渲染', }); return (
计数器
store中的count现在是 {count}
add
);}export default () => { return ( );};
可以看到,我们用Provider组件里包裹了Count组件,并且把 redux 的 store 传递了下去
在子组件里,通过useDispatch可以拿到 redux 的 dispatch, 通过useSelector可以访问到 store,拿到其中任意的返回值。
实现
用最简短的方式实现代码,探究 react-redux 为什么能在count发生改变的时候不让使用了message的组件重新渲染。
实现 Context
利用官方 api 构建 context,并且提供一个自定义 hook: useReduxContext去访问这个 context,对于忘了用 Provider 包裹的情况进行一些错误提示:
对于不熟悉自定义 hook 的小伙伴,可以看我之前写的这篇文章:使用 React Hooks + 自定义 Hook 封装一步一步打造一个完善的小型应用。[3]
import React, { useContext } from "react";import { Store } from "redux";interface ContextType { store: Store;}export const Context = (React.createContext null);export function useReduxContext() { const contextValue = useContext(Context); if (!contextValue) { throw new Error( "could not find react-redux context value; please ensure the component is wrapped in a " ); } return contextValue;}
实现 Provider
import React, { FC } from "react";import { Store } from "redux";import { Context } from "./Context";interface ProviderProps { store: Store;}export const Provider: FC = ({ store, children }) => { return {children};};
实现 useDispatch
这里就是简单的把 dispatch 返回出去,通过泛型传递让外部使用的时候可以获得类型提示。
泛型推导不熟悉的小伙伴可以看一下之前这篇:进阶实现智能类型推导的简化版 Vuex[4]
import { useReduxContext } from './Context';import { Dispatch, Action } from 'redux';export function useDispatch() { const { store } = useReduxContext(); return store.dispatch as Dispatch;}
实现 useSelector
这里才是重点,这个 api 有两个参数。
- selector: 定义如何从 state 中取值,如state => state.count
- equalityFn: 定义如何判断渲染之间值是否有改变。
在性能章节也提到过,大型应用中必须做到只有自己使用的状态改变了,才去重新渲染,那么equalityFn就是判断是否渲染的关键了。
关键流程(初始化):
- 根据传入的 selector 从 redux 的 store 中取值。
- 定义一个latestSelectedState保存上一次 selector 返回的值。
- 定义一个checkForceUpdate方法用来控制当状态发生改变的时候,让当前组件的强制渲染。
- 利用store.subscribe订阅一次 redux 的 store,下次 redux 的 store 发生变化执行checkForceUpdate。
关键流程(更新)
- 当用户使用dispatch触发了 redux store 的变动后,store 会触发checkForceUpdate方法。
- checkForceUpdate中,从latestSelectedState拿到上一次 selector 的返回值,再利用 selector(store)拿到最新的值,两者利用equalityFn进行比较。
- 根据比较,判断是否需要强制渲染组件。
有了这个思路,就来实现代码吧:
import { useReducer, useRef, useEffect } from 'react';import { useReduxContext } from './Context';type Selector = (state: State) => Selected;type EqualityFn = (a: Selected, b: Selected) => boolean;// 默认比较的方法const defaultEqualityFn = (a: T, b: T) => a === b;export function useSelector( selector: Selector, equalityFn: EqualityFn = defaultEqualityFn,) { const { store } = useReduxContext(); // 强制让当前组件渲染的方法。 const [, forceRender] = useReducer(s => s + 1, 0); // 存储上一次selector的返回值。 const latestSelectedState = useRef(); // 根据用户传入的selector,从store中拿到用户想要的值。 const selectedState = selector(store.getState()); // 检查是否需要强制更新 function checkForUpdates() { // 从store中拿到最新的值 const newSelectedState = selector(store.getState()); // 如果比较相等,就啥也不做 if (equalityFn(newSelectedState, latestSelectedState.current)) { return; } // 否则更新ref中保存的上一次渲染的值 // 然后强制渲染 latestSelectedState.current = newSelectedState; forceRender(); } // 组件第一次渲染后 执行订阅store的逻辑 useEffect(() => { // 重点,去订阅redux store的变化 // 在用户调用了dispatch后,执行checkForUpdates const unsubscribe = store.subscribe(checkForUpdates); // 组件被销毁后 需要调用unsubscribe停止订阅 return unsubscribe; }, []); return selectedState;}
总结
本文涉及到的源码地址:https://github.com/sl1673495/tiny-react-redux
原版的 react-redux 的实现肯定比这里的简化版要复杂的多,它要考虑 class 组件的使用,以及更多的优化以及边界情况。
从简化版的实现入手,我们可以更清晰的得到整个流程脉络,如果你想进一步的学习源码,也可以考虑多花点时间去看官方源码并且单步调试。
推荐React 学习相关文章
《手把手深入教你5个技巧编写更好的React代码【实践】》
《React 函数式组件性能优化知识点指南汇总》
《13个精选的React JS框架》
《深入浅出画图讲解React Diff原理【实践】》
《【React深入】React事件机制》
《Vue 3.0 Beta 和React 开发者分别杠上了》
《手把手深入Redux react-redux中间件设计及原理(上)【实践】》
《手把手深入Redux react-redux中间件设计及原理(下)【实践】》
《前端框架用vue还是react?清晰对比两者差异》
《为了学好 React Hooks, 我解析了 Vue Composition API》
《【React 高级进阶】探索 store 设计、从零实现 react-redux》
《写React Hooks前必读》
《深入浅出掌握React 与 React Native这两个框架》
《可靠React组件设计的7个准则之SRP》
《React Router v6 新特性及迁移指南》
《用React Hooks做一个搜索栏》
《你需要的 React + TypeScript 50 条规范和经验》
《手把手教你绕开React useEffect的陷阱》
《浅析 React / Vue 跨端渲染原理与实现》
《React 开发必须知道的 34 个技巧【近1W字】》
《三张图详细解说React组件的生命周期》
《手把手教你深入浅出实现Vue3 & React Hooks新UI Modal弹窗》
《手把手教你搭建一个React TS 项目模板》
《全平台(Vue/React/微信小程序)任意角度旋图片裁剪组件》
《40行代码把Vue3的响应式集成进React做状态管理》
《手把手教你深入浅出React 迷惑的问题点【完整版】》
作者:ssh
转发链接:https://mp.weixin.qq.com/s/msXG28gG2rXNRzZz2HgR-w