0%

React-Redux 源码粗读

Overview

首先看一张图:

React-Redux

可以看到入口文件里面引入了这几个模块:

  • Provider
  • connectAdvanced
  • connect

我们就来依次看一下这几个模块

Provider

store 挂在在了 context 上面,给予了子组件访问 store 的能力。

connectAdvanced

一个用来生成 connect 的高阶函数,引用了以下模块:

1
2
3
4
5
6
import hoistStatics from "hoist-non-react-statics";
import invariant from "invariant";
import { Component, createElement } from "react";

import Subscription from "../utils/Subscription";
import { storeShape, subscriptionShape } from "../utils/PropTypes";

Subscription

首先看一下 Subscription 模块。

createListenerCollection

实现一个订阅发布模式,提供 clearnotifysubscribe 方法。分别用来清除当前队列,触发事件,订阅方法。

Subscription

订阅类,订阅 onStateChange 事件,同时确保当父级也有事件订阅时,事件触发时,先触发父级事件,然后再触发当前组件事件。

makeSelectorStateful(sourceSelector,store):selector

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
function makeSelectorStateful(sourceSelector, store) {
// wrap the selector in an object that tracks its results between runs.
const selector = {
run: function runComponentSelector(props) {
try {
const nextProps = sourceSelector(store.getState(), props);
if (nextProps !== selector.props || selector.error) {
selector.shouldComponentUpdate = true;
selector.props = nextProps;
selector.error = null;
}
} catch (error) {
selector.shouldComponentUpdate = true;
selector.error = error;
}
},
};

return selector;
}

selector 用来通过 stateprop 来计算出下一个状态的 props,然后传递给连接的组件。
此函数是对 selector 的封装,在计算 props 的基础上,并记录上一次计算的结果。

connectAdvanced(selectorFactory,{options}):wrapWithConnect(WrappedComponent)

connectAdvanced 的功能主要是根据传入的 selectorFactory, options 返回用于生成 Connect 组件一个高阶组件。

connect

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class Connect extends Component {
constructor(props, context) {
super(props, context)
...
this.initSelector()
this.initSubscription()
}
...
initSelector() {
const sourceSelector = selectorFactory(this.store.dispatch, selectorFactoryOptions)
this.selector = makeSelectorStateful(sourceSelector, this.store)
this.selector.run(this.props)
}

initSubscription() {
if (!shouldHandleStateChanges) return

// parentSub's source should match where store came from: props vs. context. A component
// connected to the store via props shouldn't use subscription from context, or vice versa.
const parentSub = (this.propsMode ? this.props : this.context)[subscriptionKey]
this.subscription = new Subscription(this.store, parentSub, this.onStateChange.bind(this))

// `notifyNestedSubs` is duplicated to handle the case where the component is unmounted in
// the middle of the notification loop, where `this.subscription` will then be null. An
// extra null check every change can be avoided by copying the method onto `this` and then
// replacing it with a no-op on unmount. This can probably be avoided if Subscription's
// listeners logic is changed to not call listeners that have been unsubscribed in the
// middle of the notification loop.
this.notifyNestedSubs = this.subscription.notifyNestedSubs.bind(this.subscription)
}
...
render() {
const selector = this.selector
selector.shouldComponentUpdate = false

if (selector.error) {
throw selector.error
} else {
return createElement(WrappedComponent, this.addExtraProps(selector.props))
}
}
}

定义在 connectAdvanced 内部的组件,通过 context 获取 store 中数据,通过 initSelector 方法初始化一个包含初始 storeselector, 并通过 initSubscription 方法订阅 store 的变化。
此组件的 render 方法只是一个代理,用来把通过 selector 计算出来的 props 传递给需要包裹的组件。

connect

connect 引入了以下模块:

1
2
3
4
5
6
import connectAdvanced from "../components/connectAdvanced";
import shallowEqual from "../utils/shallowEqual";
import defaultMapDispatchToPropsFactories from "./mapDispatchToProps";
import defaultMapStateToPropsFactories from "./mapStateToProps";
import defaultMergePropsFactories from "./mergeProps";
import defaultSelectorFactory from "./selectorFactory";

只是在 connectAdvanced 的基础上,传入了 selectorFactorymapStateToPropsFactories, mapDispatchToProps, mapDispatchToPropsFactories 作为 connectAdvancedoptions 参数。

defaultSelectorFactory

定义了默认 selector 的行为,用来从传入的 state 中计算下一个状态的 props.

defaultMap xxx To xxx Factories

这类模块用来生成映射到 props 方法的工厂函数,它们的代码如下形式:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import { bindActionCreators } from "redux";
import { wrapMapToPropsConstant, wrapMapToPropsFunc } from "./wrapMapToProps";

export function whenMapDispatchToPropsIsFunction(mapDispatchToProps) {
return typeof mapDispatchToProps === "function"
? wrapMapToPropsFunc(mapDispatchToProps, "mapDispatchToProps")
: undefined;
}

export function whenMapDispatchToPropsIsMissing(mapDispatchToProps) {
return !mapDispatchToProps
? wrapMapToPropsConstant((dispatch) => ({ dispatch }))
: undefined;
}

export function whenMapDispatchToPropsIsObject(mapDispatchToProps) {
return mapDispatchToProps && typeof mapDispatchToProps === "object"
? wrapMapToPropsConstant((dispatch) =>
bindActionCreators(mapDispatchToProps, dispatch)
)
: undefined;
}

可以看到,主要引用了 wrapMapToProps 这个模块。

wrapMapToProps

wrapMapToPropsFunc

这个方法是一个通过把 mapToProps 函数包裹起来作为一个代理函数,用来做一下几件事情:

  • 检测 mapToProps 函数的调用是否依赖与 props, 用来供 selectorFactory 来决定在 props 变化的时候是否需要重新调用。
  • 在第一次调用的时候,如果 mapToProp 执行后返回另一个函数,那么处理它,并且把这个返回的函数作为新的 mapToProps 用来处理后续的调用。
  • 当第一次调用的时候,验证调用的结果是否是一个字面量对象。为了能够警告开发者,他们的 mapToProps 函数没有返回一个合法的结果。

总结

react-redux 就是一个用来绑定 redux 到 react 的一个库。如果不用 react-redux 的话,直接在 react 里面去用 redux 也是可以的,只是过于麻烦。需要手动监听 store 的变化,并手动执行组件的 render 方法。
react-redux 提供了一个高阶组件 connect 来帮我们做这件事情,而且帮我们做了一些优化。我们需要的只是,提供 mapDispatchToPropsmapStateToProps 给需要监听 store 的组件就可以了。

如果觉得此文章对你有帮助,可以请我喝咖啡O(∩_∩)O