Vue?2源碼閱讀?Provide?Inject?依賴注入詳解

更新時(shí)間：2022年08月26日 10:12:49 作者：MiyueFE

這篇文章主要為大家介紹了Vue?2源碼閱讀?Provide?Inject?依賴注入詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

Provide/Inject 初始化

1. initInjections 依賴初始化

該步驟其實(shí)發(fā)生在 initState 之前，但是由于 provide/inject 一般是配合使用，所以這里調(diào)整了一下順序。

該函數(shù)的定義與過(guò)程都比較簡(jiǎn)單：

export function initInjections(vm: Component) {
  const result = resolveInject(vm.$options.inject, vm)
  if (result) {
    toggleObserving(false)
    Object.keys(result).forEach(key => {
      if (__DEV__) {
        defineReactive(vm, key, result[key], () => warn(''))
      } else {
        defineReactive(vm, key, result[key])
      }
    })
    toggleObserving(true)
  }
}
export function resolveInject(inject: any, vm: Component): Record<string, any> | undefined | null {
  if (inject) {
    const result = Object.create(null)
    const keys = hasSymbol ? Reflect.ownKeys(inject) : Object.keys(inject)

    for (let i = 0; i < keys.length; i++) {
      const key = keys[i]
      if (key === '__ob__') continue
      const provideKey = inject[key].from
      if (provideKey in vm._provided) {
        result[key] = vm._provided[provideKey]
      } else if ('default' in inject[key]) {
        const provideDefault = inject[key].default
        result[key] = isFunction(provideDefault) ? provideDefault.call(vm) : provideDefault
      } else if (__DEV__) {
        warn('')
      }
    }
    return result
  }
}

在 initInjections 函數(shù)中，只是遍歷了 options.inject 配置的依賴數(shù)據(jù)，并關(guān)閉了依賴數(shù)據(jù)的響應(yīng)式依賴收集，最后通過(guò) defineReactive 將對(duì)應(yīng)的數(shù)據(jù)掛載到實(shí)例 vm 上，以便后面能直接訪問(wèn)。

這就是官方提示的為什么 provide/inject 的數(shù)據(jù)不是響應(yīng)式的了。

而 resolveInject 函數(shù)就是用來(lái)對(duì)組件的 inject 依賴數(shù)據(jù)進(jìn)行處理，并返回一個(gè)沒(méi)有多余原型鏈的對(duì)象。

在官方文檔中，inject 接收一個(gè)字符串?dāng)?shù)組或者一個(gè) key 為 string 的對(duì)象，而作為對(duì)象時(shí)則必須有 from 字段來(lái)表示依賴數(shù)據(jù)的獲取指向，另外也接收一個(gè) default 屬性作為降級(jí)時(shí)使用的默認(rèn)值。

但是，在 mergeOptions 之后，會(huì)將 options.inject 轉(zhuǎn)為標(biāo)準(zhǔn)對(duì)象格式。

并且這里并沒(méi)有對(duì)注入數(shù)據(jù) provide[key] 進(jìn)行處理，而是直接賦值；所以才有：如果你傳入了一個(gè)可監(jiān)聽(tīng)的對(duì)象，那么其對(duì)象的 property 還是可響應(yīng)的。

resolveInject() 函數(shù)就是解析標(biāo)準(zhǔn)格式 inject 配置，并將上層組件的 provide 的值或者 default 默認(rèn)值綁定到函數(shù)返回對(duì)象中；如果這兩個(gè)都沒(méi)有，則會(huì)提示錯(cuò)誤信息 “injection xx not found”

2. initProvide 注入數(shù)據(jù)初始化

初始化注入數(shù)據(jù)的過(guò)程也很簡(jiǎn)單，整個(gè)過(guò)程其實(shí)與 initInjection 類似。其函數(shù)定義如下：

export function initProvide(vm: Component) {
  const provideOption = vm.$options.provide
  if (provideOption) {
    const provided = isFunction(provideOption) ? provideOption.call(vm) : provideOption
    if (!isObject(provided)) {
      return
    }
    const source = resolveProvided(vm)
    
    const keys = hasSymbol ? Reflect.ownKeys(provided) : Object.keys(provided)
    for (let i = 0; i < keys.length; i++) {
      const key = keys[i]
      Object.defineProperty(
        source,
        key,
        Object.getOwnPropertyDescriptor(provided, key)!
      )
    }
  }
}
export function resolveProvided(vm: Component): Record<string, any> {
  const existing = vm._provided
  const parentProvides = vm.$parent && vm.$parent._provided
  if (parentProvides === existing) {
    return (vm._provided = Object.create(parentProvides))
  } else {
    return existing
  }
}

官方文檔中對(duì) provide 配置項(xiàng)的說(shuō)明是，可以是一個(gè)對(duì)象或者一個(gè)返回對(duì)象的函數(shù)。

所以這里首先判斷了 options.provide 的類型并獲取到了結(jié)果，如果結(jié)果不是對(duì)象則會(huì)直接退出。
然后，則是初始化 provide 的數(shù)據(jù)。

此時(shí)會(huì)將當(dāng)前實(shí)例的 provided 數(shù)據(jù)與父組件實(shí)例的 provided 進(jìn)行比較，如果相同，則返回一個(gè) 以父組件實(shí)例 provided 數(shù)據(jù)為原型創(chuàng)建的對(duì)象，否則直接返回當(dāng)前實(shí)例的 provided 數(shù)據(jù)。

因?yàn)槊恳粋€(gè)實(shí)例都會(huì)進(jìn)行與父組件實(shí)例的注入數(shù)據(jù)比較，所以才能多層級(jí)傳遞