深入理解VueComputed計算屬性原理

<div id="example">
 <p>Original message: "{{ message }}"</p>
 <p>Computed reversed message: "{{ reversedMessage }}"</p>
</div>

var vm = new Vue({
 el: '#example',
 data: {
  message: 'Hello'
 },
 computed: {
  // a computed getter
  reversedMessage: function () {
   // `this` points to the vm instance
   return this.message.split('').reverse().join('')
  }
 }
})

// src/observer/index.js

// 這里開始轉換 data 的 getter setter，原始值已存入到 __ob__ 屬性中
Object.defineProperty(obj, key, {
 enumerable: true,
 configurable: true,
 get: function reactiveGetter () {
  const value = getter ? getter.call(obj) : val
  // 判斷是否處于依賴收集狀態(tài)
  if (Dep.target) {
   // 建立依賴關系
   dep.depend()
   ...
  }
  return value
 },
 set: function reactiveSetter (newVal) {
  ...
  // 依賴發(fā)生變化，通知到計算屬性重新計算
  dep.notify()
 }
})

// src/core/instance/state.js

// 初始化計算屬性
function initComputed (vm: Component, computed: Object) {
 ...
 // 遍歷 computed 計算屬性
 for (const key in computed) {
  ...
  // 創(chuàng)建 Watcher 實例
  // create internal watcher for the computed property.
  watchers[key] = new Watcher(vm, getter || noop, noop, computedWatcherOptions)

  // 創(chuàng)建屬性 vm.reversedMessage，并將提供的函數(shù)將用作屬性 vm.reversedMessage 的 getter，
  // 最終 computed 與 data 會一起混合到 vm 下，所以當 computed 與 data 存在重名屬性時會拋出警告
  defineComputed(vm, key, userDef)
  ...
 }
}

export function defineComputed (target: any, key: string, userDef: Object | Function) {
 ...
 // 創(chuàng)建 get set 方法
 sharedPropertyDefinition.get = createComputedGetter(key)
 sharedPropertyDefinition.set = noop
 ...
 // 創(chuàng)建屬性 vm.reversedMessage，并初始化 getter setter
 Object.defineProperty(target, key, sharedPropertyDefinition)
}

function createComputedGetter (key) {
 return function computedGetter () {
  const watcher = this._computedWatchers && this._computedWatchers[key]
  if (watcher) {
   if (watcher.dirty) {
    // watcher 暴露 evaluate 方法用于取值操作
    watcher.evaluate()
   }
   // 同第1步，判斷是否處于依賴收集狀態(tài)
   if (Dep.target) {
    watcher.depend()
   }
   return watcher.value
  }
 }
}

// src/core/observer/watcher.js

// 當通過 vm.reversedMessage 獲取計算屬性時，就會進到這個 getter 方法
get () {
 // this 指的是 watcher 實例
 // 將當前 watcher 實例暫存到 Dep.target，這就表示開啟了依賴收集任務
 pushTarget(this)
 let value
 const vm = this.vm
 try {
  // 在執(zhí)行 vm.reversedMessage 的函調(diào)函數(shù)時，會觸發(fā)屬性（步驟1）和計算屬性（步驟2）的 getter
  // 在這個執(zhí)行過程中，就可以收集到 vm.reversedMessage 的依賴了
  value = this.getter.call(vm, vm)
 } catch (e) {
  if (this.user) {
   handleError(e, vm, `getter for watcher "${this.expression}"`)
  } else {
   throw e
  }
 } finally {
  if (this.deep) {
   traverse(value)
  }
  // 結束依賴收集任務
  popTarget()
  this.cleanupDeps()
 }
 return value
}

// src/core/observer/dep.js

export default class Dep {
 static target: ?Watcher;
 id: number;
 subs: Array<Watcher>;

 constructor () {
  this.id = uid++
  this.subs = []
 }

 addSub (sub: Watcher) {
  this.subs.push(sub)
 }

 removeSub (sub: Watcher) {
  remove(this.subs, sub)
 }

 depend () {
  if (Dep.target) {
   Dep.target.addDep(this)
  }
 }

 notify () {
  const subs = this.subs.slice()
  for (let i = 0, l = subs.length; i < l; i++) {
   // 更新 watcher 的值，與 watcher.evaluate() 類似，
   // 但 update 是給依賴變化時使用的，包含對 watch 的處理
   subs[i].update()
  }
 }
}

// 當首次計算 computed 屬性的值時，Dep 將會在計算期間對依賴進行收集
Dep.target = null
const targetStack = []

export function pushTarget (_target: Watcher) {
 // 在一次依賴收集期間，如果有其他依賴收集任務開始（比如：當前 computed 計算屬性嵌套其他 computed 計算屬性），
 // 那么將會把當前 target 暫存到 targetStack，先進行其他 target 的依賴收集，
 if (Dep.target) targetStack.push(Dep.target)
 Dep.target = _target
}

export function popTarget () {
 // 當嵌套的依賴收集任務完成后，將 target 恢復為上一層的 Watcher，并繼續(xù)做依賴收集
 Dep.target = targetStack.pop()
}