Mobx 源码简记
整体会写得比较乱,同时也比较简单,和读书笔记差不多,基本是边读边写。见谅~
主要三大部分Atom
、Observable
、Derivation
Atom
Mobx的原子类,能够被观察和通知变化,observableValue继承于Atom。observableValue ---> Atom
同时里面有几个比较重要的属性与方法。
-
属性
- observers,用于存放这个被原子类被谁观察了,是一个set结构
- diffValue,后续更新依赖的时候要用这个来判断
-
方法
- reportObserved,调用全局的reportObserved函数,通知自身被观察了
- reportChanged,调用全局的propagateChanged函数,通知自身发生变化了
Observable
Observable是一个工厂函数,让数据变得可观察。这个东西需要和上述的Atom建立联系,即将具体的值与Atom联系起来。从而打通自身能够被观察,同时能通知变化的整个流程。
三种可被观察的数据类型:对象,数组,Map,下面简单介绍如何实现。假如都不是,就会提示用户调用observable.box,使其拥有get,set方法,即上述说的observableValue数据类型。
部分代码如下:
fucntion Observable(v) { // 如果已被观察,直接返回 if (isObservable(v)) return v // 根据其类型分别调用observable.object、observable.array、observable.map const res = isPlainObject(v) ? observable.object(v, arg2, arg3) : Array.isArray(v) ? observable.array(v, arg2) : isES6Map(v) ? observable.map(v, arg2) : v // 返回被观察过的东西 if (res !== v) return res // 都不是,提示用户调用observable.box(value) fail( process.env.NODE_ENV !== "production" && `The provided value could not be converted into an observable. If you want just create an observable reference to the object use 'observable.box(value)'` )}
重点是observable.object、observable.array、observable.map三者的实现,下面是讨论关于对象的实现方式
-
对象(observable.object)
- 先创建一个base对象,我们称为adm对象。同时,根据这个base对象创建一个proxy,会通过该proxy将会对原对象的各种值进行代理,而adm[$mobx]指向该一个新建的ObservableObjectAdministration数据类型
- 对传进来的props(即需要被观察的对象),会先寻找是否有get属性(即计算属性),有的话会创建一个计算属性代理,并和其余的属性一起挂载在该proxy上
-
有计算属性时,会新建一个既有observableValue也有derivation属性的computedValue类,存放到adm[$mobx].values里面,key就是computed的key
- 同时会拿到它的get函数,作为这个derivation的监听函数,进行初始化监听
- 并通过Object.defineProperty设置了该属性的get和set属性
- 其余的属性,会新建一个observableValue,存放到adm[$mobx].values里面,并通过Object.defineProperty设置了该属性的get和set属性
-
然后,重点是创建proxy时的handler对象的get和set函数,在有新属性访问时或改变值时会调用get和set函数
- 访问新属性时,get函数会读取adm[$mobx],如果没有,会通过has方法,建立一个**observableValue**,并放到adm[\$mobx].pendingKeys中(还不知道有什么用)
- 设置新属性时,会新建一个observableValue存放进去adm[$mobx].values中,同时,通过Object.defineProperty设置了该属性的get和set属性
重点:
(observableValue简称为oV,Object.defineProperty简称为Od)
- 上面说的所有通过Od定义后的set会调用已存放的oV的set,get会调用已存放的oV的get
- 第一点说过oV继承于Atom,所以oV的set会调用reportChanged,oV的get会调用reportObserved
这样子,整个对象属性的监听流程就建立起来了
Reaction
Reaction(反应)是一类特殊的Derivation,可以注册响应函数,使之在条件满足时自动执行。使用如下:
// new Reaction(name, onInvalidate)const reaction = new Reaction('name', () => { // do something,即响应函数,发生副作用的地方 console.log('excuted!')})const ob = observable.object({ name: 'laji', key: '9527'})reaction.track(() => { // 注册需要被追踪的变量,这里访问了已经被观察的ob对象,所以当ob.name或ob.key发生改变时,上面的响应函数会执行 console.log(`my name is ${ob.name}`) console.log(`${ob.key} hey hey hey!`)})ob.name = 'mike' // 'excuted!'
让我们分析一下源码实现,主要有几个重点:
- 初始化Reaction类时,会将onInvalidate函数存储起来
-
在调用track函数时,这个是重点,会调用trackDerivedFunction(this, fn, undefined)
- trackDerivedFunction这个函数,就是依赖收集,即注册需要被追踪的变量,它会做几件事情,看下面注释
export function trackDerivedFunction
(derivation: IDerivation, f: () => T, context: any) { // 将该 Derivation 的 dependenciesState 和当前所有依赖的 lowestObserverState 设为最新的状态 changeDependenciesStateTo0(derivation) // 建立一个该derivation新的newObserving数组,里面存放的是谁被该derivation注册依赖了 derivation.newObserving = new Array(derivation.observing.length + 100) // 记录新的依赖的数量 derivation.unboundDepsCount = 0 // 每次执行都分配一个全局的 uid derivation.runId = ++globalState.runId // 重点,将当前的derivation分配为全局的globalState.trackingDerivation,这样被观察的 Observable 在其 reportObserved 方法中就能获取到该 Derivation const prevTracking = globalState.trackingDerivation globalState.trackingDerivation = derivation let result // 下面运行存入track的函数,触发被观察变量的get方法 if (globalState.disableErrorBoundaries === true) { result = f.call(context) } else { try { result = f.call(context) } catch (e) { result = new CaughtException(e) } } globalState.trackingDerivation = prevTracking // 比较新旧依赖,更新依赖 bindDependencies(derivation) return result}
可以看到,重点有两个,一个是将当前的derivation分配为全局的globalState.trackingDerivation,一个是下面的更新依赖过程。
-
接下来,我们看看触发了被观察变量的get方法,会是怎样的,上面说过,调用get方法会执行reportObserved函数
export function reportObserved(observable: IObservable): boolean { // 拿到刚才被设置到全局的derivation const derivation = globalState.trackingDerivation if (derivation !== null) { if (derivation.runId !== observable.lastAccessedBy) { observable.lastAccessedBy = derivation.runId // 这行是重点,将被观察的变量,放到derivation.newObserving数组中,因此,derivation里就存放了这次访问中被观察的变量 derivation.newObserving![derivation.unboundDepsCount++] = observable if (!observable.isBeingObserved) { observable.isBeingObserved = true observable.onBecomeObserved() } } return true } else if (observable.observers.size === 0 && globalState.inBatch > 0) { queueForUnobservation(observable) } return false}
-
之后是bindDependencies函数的执行。这里面有两点,不做代码解读了:
- 一是主要是比较derivation的新旧observing(存放被观察变量的数组),防止重复记录,同时去除已过期的被观察变量
- 二是,observable(被观察的变量)的observers(是一个Set结构)更新里面存放的derivation,即记录自身被谁观察了,在后面调用reportChanged时,触发响应函数
被观察的变量发生变化时
此时会调用observable的set函数,然后调用reportChanged,最终会调用一个叫做propagateChanged函数。
export function propagateChanged(observable: IObservable) { // 已经在运行了,直接返回 if (observable.lowestObserverState === IDerivationState.STALE) return observable.lowestObserverState = IDerivationState.STALE // 上面说过,observable(被观察的变量)的observers存放着derivation // 这里就是执行每个derivation的onBecomeStale函数 observable.observers.forEach(d => { if (d.dependenciesState === IDerivationState.UP_TO_DATE) { if (d.isTracing !== TraceMode.NONE) { logTraceInfo(d, observable) } d.onBecomeStale() } d.dependenciesState = IDerivationState.STALE })}
onBecomeStale最终会调用derivation里的schedule函数,里面做了两件事:
- 把自身推进全局的globalState.pendingReactions数组
-
执行runReactions函数
- 该函数就核心就做一件事情,遍历globalState.pendingReactions数组,执行里面每个derivation的runReaction函数
- runReaction最终会调用derivation自身的onInvalidate,即响应函数
至此,整个mobx的数据观察与响应流程就都一一解释完整了(autorun,autorunAsync,when等函数都是基于Reaction来实现的,就不作过多解读了)
Mobx-React源码简记
既然mobx都说了,那就把mobx-react也分析一下吧。其实很简单,只要理解了Reaction与Observable,就很容易明白mobx-react的实现了。
mobx-react的实现主要也是两点
- 通过provide和inject,将已经被观察过的observerableStore集中起来并按需分配到所需要的组件中
- 被observer的组件,改写其render函数,使其可以响应变化
第一点比较简单,实现一个hoc,把observerableStore添加到context上,然后被inject的组件就可以拿到所需的observerableStore
我们重点看下第二点,实现第二点的主要逻辑,在observer.js
里面的makeComponentReactive函数中,看下面简化版的重点解析
// makeComponentReactivefunction makeComponentReactive(render) { if (isUsingStaticRendering === true) return render.call(this) // 改造后的render函数 function reactiveRender() { // 防止重复执行响应函数,因为componentWillReact有可能有副作用 isRenderingPending = false // render函数执行后返回的jsx let rendering = undefined // 注册需要被追踪的变量 reaction.track(() => { if (isDevtoolsEnabled) { this.__$mobRenderStart = Date.now() } try { // _allowStateChanges是安全地执行原来的render函数,假如在action外有更改变量的行为,会报错 // 重点是这个,因为render函数被执行了,所以假如里面有被observe过的变量,就能被追踪,更新到依赖该reaction的依赖列表里面 rendering = _allowStateChanges(false, baseRender) } catch (e) { exception = e } if (isDevtoolsEnabled) { this.__$mobRenderEnd = Date.now() } }) return rendering } // ....省略一些代码 // 新建一个Reaction,注册响应函数 const reaction = new Reaction(`${initialName}#${rootNodeID}.render()`, () => { if (!isRenderingPending) { // 正在执行响应函数 isRenderingPending = true // 这里就是执行新的componentWillReact生命周期的地方 if (typeof this.componentWillReact === "function") this.componentWillReact() if (this.__$mobxIsUnmounted !== true) { let hasError = true try { setHiddenProp(this, isForcingUpdateKey, true) // 也是重点,通过forceUpdate,更新组件 if (!this[skipRenderKey]) Component.prototype.forceUpdate.call(this) hasError = false } finally { setHiddenProp(this, isForcingUpdateKey, false) if (hasError) reaction.dispose() } } } }) // 改写原来的render reaction.reactComponent = this reactiveRender[mobxAdminProperty] = reaction this.render = reactiveRender return reactiveRender.call(this)}
可以见到,通过建立一个Reaction,实现了render函数里的被观察的变量收集及响应函数注册。而且在通过forceUpdate重新更新组件后,render函数会被重新执行一遍,从而实时更新被观察的变量。整体的实现还是巧妙的。
除此之外,还有一些生命周期的优化,对props、state也进行监听等操作,在这里就不一一解读了