本文是一篇逐行粒度的vuex源碼分析,帶你一步一步去實現一個vuex,不一樣於市面上其餘的源碼分析,本文不會從上帝視角去談vuex的設計,而是將vuex的功能一個個拆解,變成簡單易懂的幾個部分,很是適合初學者閱讀。vue
爲了實現經過Vue.use()方法引入vuex,須要爲vuex定義一個install方法。vuex中的intall方法主要做用是將store實例注入到每個vue組件中,具體實現方式以下git
export function install (_Vue) {
// 避免重複安裝
if (Vue && Vue === _Vue) {
// 開發環境報錯
console.warn("duplicate install");
}
Vue = _Vue;
// 開始註冊全局mixin
applyMixin(Vue);
}
複製代碼
以上代碼中經過定義一個全局變量Vue保存當前的引入的Vue來避免重複安裝,而後經過apllyMixin實現將store注入到各個實例中去github
export default function (Vue) {
// 獲取vue版本
const version = Number(Vue.version.split(".")[0]);
// 根據版本選擇註冊方式
if (version >= 2) {
// 版本大於2在mixin中執行初始化函數
Vue.mixin({ beforeCreate: vuexInit });
} else {
// 低版本,將初始化方法放在options.init中執行
const _init = Vue.prototype._init;
Vue.prototype._init = function (options = {}) {
options.init = options.init
? [vuexInit].concat(options.init)
: vuexInit;
_init();
};
}
// 初始化函數:將store做爲屬性注入到全部組件中
function vuexInit () {
// 根組件
if (this.$options && this.$options.store) {
this.$store = typeof this.$options.store === "function"
? this.$options.store()
: this.$options.store;
} else if (this.$options.parent && this.$options.parent.$store) { // 非根組件
this.$store = this.$options.parent.$store;
}
}
}
複製代碼
首先看這段代碼核心邏輯實現的關鍵函數vuexInit,該函數首先判斷this.$options選項(該選項在根實例實例化時傳入new Vue(options:Object))vuex
new Vue({
store
})
複製代碼
中是否包含store屬性,若是有,則將實例的this.options.store,若是沒有則指向this.store。 此時咱們在install執行後,經過在實例化根組件時把store傳入options就能將全部子組件的$store屬性都指向這個store了。 此外須要注意的時applyMixin執行時首先會判斷當前Vue的版本號,版本2以上經過mixin混入的方式在全部組件實例化的時候執行vueInit,而 版本2如下則經過options.init中插入執行的方式注入。如下時安裝函數的幾點總結api
commit實際上就是一個比較簡單的發佈-訂閱模式的實現,不過這個過程當中會涉及module的實現,state與getters之間響應式的實現方式,併爲以後介紹actions能夠作一些鋪墊數組
首先回顧下commit的使用promise
// 實例化store
const store = new Vuex.Store({
state: { count: 1 },
mutations: {
add (state, number) {
state.count += number;
}
}
});
複製代碼
實例化store時,參數中的mutation就是事件隊列中的事件,每一個事件傳入兩個參數,分別時state和payload,每一個事件實現的都是根據payload改變state的值數據結構
<template>
<div>
count:{{state.count}}
<button @click="add">add</button>
</div>
</template>
<script>
export default {
name: "app",
created () {
console.log(this);
},
computed: {
state () {
return this.$store.state;
}
},
methods: {
add () {
this.$store.commit("add", 2);
}
}
};
</script>
<style scoped>
</style>
複製代碼
咱們在組件中經過commit觸發相應類型的mutation並傳入一個payload,此時state會實時發生變化app
首先來看爲了實現commit咱們在構造函數中須要作些什麼異步
export class Store {
constructor (options = {}) {
// 聲明屬性
this._mutations = Object.create(null);
this._modules = new ModuleCollection(options);
// 聲明發布函數
const store = this;
const { commit } = this;
this.commit = function (_type, _payload, _options) {
commit.call(store, _type, _payload, _options);
};
const state = this._modules.root.state;
// 安裝根模塊
this.installModule(this, state, [], this._modules.root);
// 註冊數據相應功能的實例
this.resetStoreVm(this, state);
}
複製代碼
首先是三個實例屬性_mutations是發佈訂閱模式中的事件隊列,_modules屬性用來封裝傳入的options:{state, getters, mutations, actions} 爲其提供一些基礎的操做方法,commit方法用來觸發事件隊列中相應的事件;而後咱們會在installModule 中註冊事件隊列,在resetStoreVm中實現一個響應式的state。
在實例化store時咱們會傳入一個對象參數,這裏麪包含state,mutations,actions,getters,modules等數據項咱們須要對這些數據項進行封裝,並暴露一個這個些數據項的操做方法,這就是Module類的做用,另外在vuex中有模塊的劃分,須要對這些modules進行管理,由此衍生出了ModuleCollection類,本節先專一於commit的實現對於模塊劃分會放在後面討論,對於直接傳入的state,mutations,actions,getters,在vuex中會先經過Module類進行包裝,而後註冊在ModuleCollection的root屬性中
export default class Module {
constructor (rawModule, runtime) {
const rawState = rawModule.state;
this.runtime = runtime;// 1.todo:runtime的做用是啥
this._rawModule = rawModule;
this.state = typeof rawState === "function" ? rawState() : rawState;
}
// 遍歷mumation,執行函數
forEachMutation (fn) {
if (this._rawModule.mutations) {
forEachValue(this._rawModule.mutations, fn);
}
}
}
export function forEachValue (obj, fn) {
Object.keys(obj).forEach((key) => fn(obj[key], key));
}
複製代碼
構造函數中傳入的參數rawModule就是{state,mutations,actions,getters}對象,在Module類中定義兩個屬性_rawModule用於存放傳入的rawModule,forEachMutation實現mutations的遍歷執行,將mutation對象的value,key傳入fn並執行,接下去將這個module掛在modulecollection的root屬性上
export default class ModuleCollection {
constructor (rawRootModule) {
// 註冊根module,入參:path,module,runtime
this.register([], rawRootModule, false);
}
// 1.todo runtime的做用?
register (path, rawRootModule, runtime) {
const module = new Module(rawRootModule, runtime);
this.root = module;
}
}
複製代碼
通過這樣一系列的封裝,this._modules屬性就是下面這樣的數據結構
因爲mutations中保存的全部事件都是爲了按必定規則改變state,因此咱們要先介紹下store是如何進行state的管理的 尤爲是如何經過state的改變響應式的改變getters中的值,在構造函數中提到過一個方法resetStoreVm,在這個函數中 會實現state和getters的響應式關係
resetStoreVm (store, state) {
const oldVm = store._vm;
// 註冊
store._vm = new Vue({
data: {
$$state: state
}
});
// 註銷舊實例
if (oldVm) {
Vue.nextTick(() => {
oldVm.destroy();
});
}
}
複製代碼
這個函數傳入兩個參數,分別爲實例自己和state,首先註冊一個vue實例保存在store實例屬性_vm上,其中data數據項中定義了$$state屬性指向state,後面會介紹將getters分解並放在computed數據項中這樣很好的利用Vue原有的數據響應系統實現響應式的state,而且賦新值以後會把老的實例註銷。
對於state的包裝實際還差一步,咱們日常訪問state的時候是直接經過store.state訪問的,若是不作處理如今咱們只能經過store._vm.data.$$state來訪問,實際vuex經過class的get,set屬性實現state的訪問和更新的
export class Store {
get state () {
return this._vm._data.$$state;
}
set state (v) {
if (process.env.NODE_ENV !== "production") {
console.error("user store.replaceState()");
}
}
}
複製代碼
值得注意的是,咱們不能直接對state進行賦值,而要經過store.replaceState賦值,不然將會報錯
接下去終於要步入commit原理的核心了,發佈-訂閱模式包含兩個步驟,事件訂閱和事件發佈,首先來談談vuex是如何實現訂閱過程的
export class Store {
constructor (options = {}) {
// 聲明屬性
this._mutations = Object.create(null);// 爲何不直接賦值null
this._modules = new ModuleCollection(options);
const state = this._modules.root.state;
// 安裝根模塊
this.installModule(this, state, [], this._modules.root);
}
installModule (store, state, path, module) {
// 註冊mutation事件隊列
const local = this.makeLocalContext(store, path);
module.forEachMutation((mutation, key) => {
this.registerMutation(store, key, mutation, local);
});
}
// 註冊mutation
registerMutation (store, type, handler, local) {
const entry = this._mutations[type] || (this._mutations[type] = []);
entry.push(function WrappedMutationHandler (payload) {
handler.call(store, local.state, payload);
});
}
}
複製代碼
咱們只截取相關的部分代碼,其中兩個關鍵的方法installModule和registerMutation,咱們在此處會省略一些關於模塊封裝的部分,此處的local能夠簡單的理解爲一個{state,getters}對象,事件註冊的大體過程就是遍歷mutation並將mutation進行包裝後push進指定類型的事件隊列,首先經過Moulde類的實例方法forEachMutation對mutation進行遍歷,並執行registerMutation進行事件的註冊,在registerMutation中生成一個this._mutations指定類型的事件隊列,註冊事件後的this._mutations的數據結構以下
根據事件註冊後this._mutations的結構,咱們能夠很輕鬆的實現事件發佈,找到指定類型的事件隊列,遍歷這個隊列,傳入參數並執行。
// 觸發對應type的mutation
commit (_type, _payload, _options) {
// 獲取參數
const {
type,
payload
} = unifyObjectStyle(_type, _payload, _options);
const entry = this._mutations[type];
// 遍歷觸發事件隊列
entry.forEach(function commitIterator (handler) {
handler(payload);
});
}
複製代碼
可是須要注意的是,首先須要對參數進行下處理,就是unifyObjectStyle乾的事情
// 入參規則:type能夠是帶type屬性的對象,也能夠是字符串
function unifyObjectStyle (type, payload, options) {
if (isObject(type)) {
payload = type;
options = payload;
type = type.type;
}
return { type, payload, options };
}
複製代碼
其實實現了type能夠爲字符串,也能夠爲對象,當爲對象是,內部使用的type就是type.type,而第二個 參數就變成了type,第三個參數變成了payload。
到此關於commit的原理已經介紹完畢,全部的代碼見分支 github.com/miracle9312…
定義一個action
add ({ commit }, number) {
return new Promise((resolve, reject) => {
setTimeout(() => {
const pow = 2;
commit("add", Math.pow(number, pow));
resolve(number);
}, 1000);
});
}
複製代碼
觸發action
this.$store.dispatch("add", 4).then((data) => {
console.log(data);
});
複製代碼
有時咱們須要觸發一個異步執行的事件,好比接口請求等,可是若是依賴mutatoin這種同步執行的事件隊列,咱們沒法 獲取執行的最終狀態。此時咱們須要找到一種解決方案實現如下兩個目標
經過這兩個目標,咱們能夠大體推算該如何實現了,只要保證定義的全部事件都返回一個promise,再將這些promise 放在一個隊列中,經過promise.all去執行,返會一個最終狀態的promise,這樣既能保證事件之間的執行順序,也能 捕獲最終的執行狀態。
首先咱們定義一個實例屬性_actions,用於存放事件隊列
constructor (options = {}) {
// ...
this._actions = Object.create(null);
// ...
}
複製代碼
接着在module類中定義一個實例方法forEachActions,用於遍歷執行actions
export default class Module {
// ...
forEachAction (fn) {
if (this._rawModule.actions) {
forEachValue(this._rawModule.actions, fn);
}
}
// ...
}
複製代碼
而後在installModule時期去遍歷actions,註冊事件隊列
installModule (store, state, path, module) {
// ...
module.forEachAction((action, key) => {
this.registerAction(store, key, action, local);
});
// ...
}
複製代碼
註冊
registerAction (store, type, handler, local) {
const entry = this._actions[type] || (this._actions[type] = []);
entry.push(function WrappedActionHandler (payload, cb) {
let res = handler.call(store, {
dispatch: local.dispatch,
commit: local.commit,
state: local.state,
rootState: store.state
}, payload, cb);
// 默認action中返回promise,若是不是則將返回值包裝在promise中
if (!isPromise(res)) {
res = Promise.resolve(res);
}
return res;
});
}
複製代碼
註冊方法中包含四個參數,store表明store實例,type表明action類型,handler是action函數。首先判斷是否已存在該類型acion的事件隊列,若是不存在則須要初始化爲數組。而後將該事件推入指定類型的事件隊列。須要注意的兩點,第一,action函數訪問到的第一個參數爲一個context對象,第二,事件返回的值始終是一個promise。
dispatch (_type, _payload) {
const {
type,
payload
} = unifyObjectStyle(_type, _payload);
// ??todo 爲何是一個事件隊列,什麼時候會出現一個key對應多個action
const entry = this._actions[type];
// 返回promise,dispatch().then()接收的值爲數組或者某個值
return entry.length > 1
? Promise.all(entry.map((handler) => handler(payload)))
: entry[0](payload);
}
複製代碼
首先獲取相應類型的事件隊列,而後傳入參數執行,返回一個promise,當事件隊列中包含的事件個數大於1時 將返回的promise保存在一個數組中,而後經過Pomise.all觸發,當事件隊列中的事件只有一個時直接返回promise 這樣咱們就能夠經過dispatch(type, payload).then(data=>{})獲得異步執行的結果,此外事件隊列中的事件 觸發經過promise.all實現,兩個目標都已經達成。
在store實例化時咱們定義以下幾個選項:
const store = new Vuex.Store({
state: { count: 1 },
getters: {
square (state, getters) {
return Math.pow(state.count, 2);
}
},
mutations: {
add (state, number) {
state.count += number;
}
}
});
複製代碼
首先咱們在store中定義一個state,getters和mutations,其中state中包含一個count,初始值爲1,getters中定義一個square,該值返回爲count的平方,在mutations中定義一個add事件,當觸發add時count會增長number。
接着咱們在頁面中使用這個store:
<template>
<div>
<div>count:{{state.count}}</div>
<div>getterCount:{{getters.square}}</div>
<button @click="add">add</button>
</div>
</template>
<script>
export default {
name: "app",
created () {
console.log(this);
},
computed: {
state () {
return this.$store.state;
},
getters () {
return this.$store.getters;
}
},
methods: {
add () {
this.$store.commit("add", 2);
}
}
};
</script>
<style scoped>
</style>
複製代碼
執行的結果是,咱們每次觸發add事件時,state.count會相應增2,而getter始終時state.count的平方。這不禁得讓咱們想起了vue中的響應式系統,data和computed之間的關係,貌似一模一樣,實際上vuex就是利用vue中的響應式系統實現的。
首先定義一個實例屬性_wappedGetters用來存放getters
export class Store {
constructor (options = {}) {
// ...
this._wrappedGetters = Object.create(null);
// ...
}
}
複製代碼
在modules中定義一個遍歷執行getters的實例方法,並在installModule方法中註冊getters,並將getters存放至_wrappedGetters屬性中
installModule (store, state, path, module) {
// ...
module.forEachGetters((getter, key) => {
this.registerGetter(store, key, getter, local);
});
// ...
}
複製代碼
registerGetter (store, type, rawGetters, local) {
// 處理getter重名
if (this._wrappedGetters[type]) {
console.error("duplicate getter");
}
// 設置_wrappedGetters,用於
this._wrappedGetters[type] = function wrappedGetterHandlers (store) {
return rawGetters(
local.state,
local.getters,
store.state,
store.getters
);
};
}
複製代碼
須要注意的是,vuex中不能定義兩個相同類型的getter,在註冊時,咱們將一個返回選項getters執行結果的函數,傳入的參數爲store實例,選項中的getters接受四個參數分別爲做用域下和store實例中的state和getters關於local的問題在以後module原理的時候再作介紹,在這次的實現中local和store中的參數都是一致的。
以後咱們須要將全部的getters在resetStoreVm時期注入computed,而且在訪問getters中的某個屬性時將其代理到store.vm中的相應屬性
// 註冊響應式實例
resetStoreVm (store, state) {
// 將store.getters[key]指向store._vm[key],computed賦值
forEachValue(wrappedGetters, function (fn, key) {
computed[key] = () => fn(store);
});
// 註冊
store._vm = new Vue({
data: {
$$state: state
},
computed
});
// 註銷舊實例
if (oldVm) {
Vue.nextTick(() => {
oldVm.destroy();
});
}
}
複製代碼
在resetStroreVm時期,遍歷wrappedGetters,並將getters包裝在一個具備相同key的computed中再將這個computed注入到store._vm實例中。
resetStoreVm (store, state) {
store.getters = {};
forEachValue(wrappedGetters, function (fn, key) {
// ...
Object.defineProperty(store.getters, key, {
get: () => store._vm[key],
enumerable: true
});
});
// ...
}
複製代碼
而後將store.getters中的屬性指向store._vm中對應的屬性,也就是store.computed中對應的屬性這樣,當store._vm中data.$$state(store.state)發生變化時,引用state的getter也會實時計算以上就是getters可以響應式變化的原理 具體代碼見 github.com/miracle9312…
helpers.js中向外暴露了四個方法,分別爲mapState,mapGetters,mapMutations和mapAction。這四個輔助方法 幫助開發者在組件中快速的引用本身定義的state,getters,mutations和actions。首先了解其用法再深刻其原理
const store = new Vuex.Store({
state: { count: 1 },
getters: {
square (state, getters) {
return Math.pow(state.count, 2);
}
},
mutations: {
add (state, number) {
state.count += number;
}
},
actions: {
add ({ commit }, number) {
return new Promise((resolve, reject) => {
setTimeout(() => {
const pow = 2;
commit("add", Math.pow(number, pow));
resolve(number);
}, 1000);
});
}
}
});
複製代碼
以上是咱們定義的store
<template>
<div>
<div>count:{{count}}</div>
<div>getterCount:{{square}}</div>
<button @click="mutAdd(1)">mutAdd</button>
<button @click="actAdd(1)">actAdd</button>
</div>
</template>
<script>
import vuex from "./vuex/src";
export default {
name: "app",
computed: {
...vuex.mapState(["count"]),
...vuex.mapGetters(["square"])
},
methods: {
...vuex.mapMutations({ mutAdd: "add" }),
...vuex.mapActions({ actAdd: "add" })
}
};
</script>
<style scoped>
</style>
複製代碼
而後經過mapXXX的方式將store引入組件並使用。觀察這幾個方法的引用方式,能夠知道這幾個方法最終都會返回一個 對象,對象中全部的值都是一個函數,再經過展開運算符把這些方法分別注入到computed和methods屬性中。對於mapState 和mapGetters而言,返回對象中的函數,執行後會返回傳入參數對應的值(return store.state[key];或者return store.getters[key]), 而對於mapMutations和mapActions而言,返回對象中的函數,將執行commit([key],payload),或者dispatch([key],payload) 這就是這幾個方法的簡單原理,接下去將一個個分析vuex中的實現
export const mapState = function (states) {
// 定義一個返回結果map
const res = {};
// 規範化state
normalizeMap(states).forEach(({ key, val }) => {
// 賦值
res[key] = function mappedState () {
const state = this.$store.state;
const getters = this.$store.getters;
return typeof val === "function"
? val.call(this, state, getters)
: state[val];
};
});
// 返回結果
return res;
};
複製代碼
首先看mapsState最終的返回值res是一個對象,傳入的參數是咱們想要map出來的幾個屬性,mapState能夠傳入一個字符串數組或者是對象數組,字符串數組中包含的是引用的屬性,對象數組包含的是使用值與引用的映射,這兩種形式的傳參,咱們須要經過normalizeMap進行規範化,統一返回一個對象數組
function normalizeMap (map) {
return Array.isArray(map)
? map.map(key => ({ key, val: key }))
: Object.keys(map).map(key => ({ key, val: map[key] }))
}
複製代碼
normalizeMap函數首先判斷傳入的值是否爲數組,如果,則返回一個key和val都爲數組元素的對象數組,若是不是數組,則判斷傳入值爲一個對象,接着遍歷該對象,返回一個以對象鍵值爲key和val值的對象數組。此時經過normalizeMap以後的map都將是一個對象數組。
接着遍歷規範化以後的數組,對返回值對象進行賦值,賦值函數執行後返回state對應key的值若是傳入值爲一個函數,則將getters和state做爲參數傳入並執行,最終返回該對象,這樣在computed屬性中展開後就能直接經過key來引用對應state的值了。
mapGetters與mapState的實現原理基本一致
export const mapGetters = function (getters) {
const res = {};
normalizeMap(getters)
.forEach(({ key, val }) => {
res[key] = function mappedGetter () {
return this.$store.getters[val];
};
});
return res;
};
複製代碼
export const mapActions = function (actions) {
const res = {};
normalizeMap(actions)
.forEach(({ key, val }) => {
res[key] = function (...args) {
const dispatch = this.$store.dispatch;
return typeof val === "function"
? val.apply(this, [dispatch].concat(args))
: dispatch.apply(this, [val].concat(args));
};
});
return res;
};
複製代碼
mapActions執行後也將返回一個對象,對象的key用於組件中引用,對象中value爲一個函數,該函數傳參是dispatch執行時的payload,其中val若是不是一個函數,則判斷其爲actionType經過dispath(actionType,payload)來觸發對應的action若是傳入的參數爲一個函數則將dispatch和payload做爲參數傳入並執行,這樣能夠實如今mapAction時組合調用多個action,或者自定義一些其餘行爲。最終返回該對象,在組件的methods屬性中展開後,能夠經過調用key對應的函數來觸發action。
mapMutation的實現原理與mapActions大同小異
export const mapMutations = function (mutations) {
const res = {};
normalizeMap(mutations)
.forEach(({ key, val }) => {
res[key] = function mappedMutation (...args) {
const commit = this.$store.commit;
return typeof val === "function"
? val.apply(this, [commit].concat(args))
: commit.apply(this, [val].concat(args));
};
});
return res;
};
複製代碼
爲了方便進行store中不一樣功能的切分,在vuex中能夠將不一樣功能組裝成一個單獨的模塊,模塊內部能夠單獨管理state,也能夠訪問到全局狀態。
// main.js
const store = new Vuex.Store({
state: {},
getters: {},
mutations: {},
actions: {},
modules: {
a: {
namespaced: true,
state: { countA: 9 },
getters: {
sqrt (state) {
return Math.sqrt(state.countA);
}
},
mutations: {
miner (state, payload) {
state.countA -= payload;
}
},
actions: {
miner (context) {
console.log(context);
}
}
}
}
});
複製代碼
//app.vue
<template>
<div>
<div>moduleSqrt:{{sqrt}}</div>
<div>moduleCount:{{countA}}</div>
<button @click="miner(1)">modMutAdd</button>
</div>
</template>
<script>
import vuex from "./vuex/src";
export default {
name: "app",
created () {
console.log(this.$store);
},
computed: {
...vuex.mapGetters("a", ["sqrt"]),
...vuex.mapState("a", ["countA"])
},
methods: {
...vuex.mapMutations("a", ["miner"])
}
};
</script>
<style scoped>
</style>
複製代碼
上述代碼中,咱們定義了一個key爲a的module,將其namespaced設置成了true,對於namespace=false的模塊,它將自動繼承父模塊的命名空間。對於模塊a,他有如下幾點特性
根據以上特性,能夠將以後的module的實現分爲幾個部分
vuex最後構造出的module是這樣的一種嵌套的結構
{
state: {},
getters: {},
mutations: {},
actions: {},
modules: {
a: {
namespaced: true,
state: {},
getters: {},
mutations: {},
actions: {}
},
b: {
namespaced: true,
state: {},
getters: {},
mutations: {},
actions: {}
}
}
}
複製代碼
咱們會在store的構造函數中將這個對象做爲ModuleCollection實例化的參數
export class Store {
constructor (options = {}) {
this._modules = new ModuleCollection(options);
}
}
複製代碼
全部的嵌套結構的構造都在ModuleCollection實例化的過程當中進行
// module-collection.js
export default class ModuleCollection {
constructor (rawRootModule) {
// 註冊根module,入參:path,module,runtime
this.register([], rawRootModule, false);
}
// 根據路徑獲取模塊,從root開始搜索
get (path) {
return path.reduce((module, key) => module.getChild(key), this.root);
}
// 1.todo runtime的做用?
register (path, rawModule, runtime = true) {
// 生成module
const newModule = new Module(rawModule, runtime);
if (path.length === 0) { // 根模塊,註冊在root上
this.root = newModule;
} else { // 非根模塊,獲取父模塊,掛載
const parent = this.get(path.slice(0, -1));
parent.addChild(path[path.length - 1], newModule);
}
// 模塊上是否含有子模塊,有則註冊子模塊
if (rawModule.modules) {
forEachValue(rawModule.modules, (newRawModule, key) => {
this.register(path.concat(key), newRawModule, runtime);
});
}
}
}
複製代碼
// module.js
export default class Module {
addChild (key, module) {
this._children[key] = module;
}
}
複製代碼
實例化時首先會執行register函數,在register函數中根據傳入的rawModule建立一個Module的實例而後根據註冊的路徑判斷是否爲根模塊,若是是,則將該module實例掛載在root屬性上,若是不是則經過get方法找到該模塊的父模塊,將其經過模塊的addChild方法掛載在父模塊的_children屬性上,最後判斷該模塊是否含有嵌套模塊,若是有則遍歷嵌套模塊,遞歸執行register方法,這樣就能構造如上圖所示的嵌套模塊結構了。有了以上這樣的結構,咱們能夠用reduce方法經過path來獲取指定路徑下的模塊,也能夠用遞歸的方式對全部的模塊進行統一的操做,大大方便了模塊的管理。
有了基本的模塊結構後,下面的問題就是如何進行模塊做用域的封裝了,讓每一個模塊有本身的state而且對於這個state有本身管理這個state的方法,而且咱們但願這些方法也可以訪問到全局的一些屬性。
總結一下如今咱們要作的事情,
// module
{
state: {},
getters: {}
...
modules:{
n1:{
namespaced: true,
getters: {
g(state, rootState) {
state.s // => state.n1.s
rootState.s // => state.s
}
},
mutations: {
m(state) {
state.s // => state.n1.s
}
},
actions: {
a({state, getters, commit, dispatch}) {
commit("m"); // => mutations["n1/m"]
dispatch("a1"); // => actions["n1/a1"]
getters.g // => getters["n1/g"]
},
a1(){}
}
}
}
}
複製代碼
在namespaced=true的模塊中,訪問到的state,getters都是自模塊內部的state和getters,只有rootState,以及rootGetters指向根模塊的state和getters;另外,在模塊中commit觸發的都是子模塊內部的mutations,dispatch觸發的都是子模塊內部的actions。在vuex中經過路徑匹配去實現這種封裝。
//state
{
"s": "any"
"n1": {
"s": "any",
"n2": {
"s": "any"
}
}
}
// getters
{
"g": function () {},
"n1/g": function () {},
"n1/n2/g": function () {}
}
// mutations
{
"m": function () {},
"n1/m": function () {},
"n1/n2/m": function () {}
}
// actions
{
"a": function () {},
"n1/a": function () {},
"n1/n2/a": function () {}
}
複製代碼
vuex中要構造這樣一種數據結構,去存儲各個數據項,而後將context中的commit方法重寫,將commit(type)代理至namespaceType以實現commit方法的封裝,相似的dispatch也是經過這種方式進行封裝,而getters則是實現了一個getterProxy,將key代理至store.getters[namespace+key]上,而後在context中的getters替換成該getterProxy,而state則是利用了以上這種數據結構,直接找到對應path的state賦給context.state,這樣經過context訪問到的都是模塊內部的數據了。
接着來看看代碼實現
installModule (store, state, path, module, hot) {
const isRoot = !path.length;
// 獲取namespace
const namespace = store._modules.getNamespace(path);
}
複製代碼
全部數據項的構造,以及context的構造都在store.js的installModule方法中,首先經過傳入的path獲取namespace
// 根據路徑返回namespace
getNamespace (path) {
let module = this.root;
return path.reduce((namespace, key) => {
module = module.getChild(key);
return namespace + (module.namespaced ? `${key}/` : "");
}, "");
}
複製代碼
獲取namespace的方法是ModuleCollections的一個實例方法,它會逐層訪問modules,判斷namespaced屬性,若爲true則將path[index]拼在namespace上 這樣就得到了完整的namespace
以後是嵌套結構state的實現
installModule (store, state, path, module, hot) {
// 構造嵌套state
if (!isRoot && !hot) {
const moduleName = path[path.length - 1];
const parentState = getNestedState(state, path.slice(0, -1));
Vue.set(parentState, moduleName, module.state);
}
}
複製代碼
首先根據出path獲取state上對應的parentState,此處入參state就是store.state
function getNestedState (state, path) {
return path.length
? path.reduce((state, key) => state[key], state)
: state
}
複製代碼
其中的getNestState,用於根據路徑獲取相應的state,在獲取parentState以後,將module.state掛載在parentState[moduleName]上。 這樣就構造了一個如上說所述的嵌套state結構。
在獲得namespace以後咱們須要將傳入的getters,mutations,actions根據namespace去構造了
installModule (store, state, path, module, hot) {
module.forEachMutation((mutation, key) => {
const namespacdType = namespace + key;
this.registerMutation(store, namespacdType, mutation, local);
});
module.forEachAction((action, key) => {
const type = action.root ? type : namespace + key;
const handler = action.handler || action;
this.registerAction(store, type, handler, local);
});
module.forEachGetters((getter, key) => {
const namespacedType = namespace + key
this.registerGetter(store, namespacedType, getter, local);
});
}
複製代碼
getters,mutations,actions的構造有着幾乎同樣的方式,只不過度別掛載在store._getters,store._mutations,stors._actions上而已,所以咱們值分析mutations的構造過程。首先是forEachMutation遍歷module中的mutations對象,而後經過ergisterMustions註冊到以namespace+key的 key上
function registerMutation (store, type, handler, local) {
const entry = store._mutations[type] || (store._mutations[type] = [])
entry.push(function wrappedMutationHandler (payload) {
handler.call(store, local.state, payload)// mutation中第一個參數是state,第二個參數是payload
})
}
複製代碼
實際上會存放在store._mutations[namespace+key]上。
經過上述操做,咱們已經完成了封裝的一半,接下來咱們還要爲每一個module實現一個context,在這個context裏面有state,getters,commit和actions, 可是這裏的state,getters只能訪問module裏面的state和getters,而commit和actions也只能觸達到module內部的state和getters
installModule (store, state, path, module, hot) {
// 註冊mutation事件隊列
const local = module.context = makeLocalContext(store, namespace, path);
}
複製代碼
咱們會在installModule裏面去實現這個context,而後將組裝完的context分別賦給local和module.context, 而這個local在會在register的時候傳遞給getters,mutations, actions做爲參數
function makeLocalContext (store, namespace, path) {
const noNamespace = namespace === "";
const local = {
dispatch: noNamespace
? store.dispatch
: (_type, _payload, _options) => {
const args = unifyObjectStyle(_type, _payload, _options);
let { type } = args;
const { payload, options } = args;
if (!options || !options.root) {
type = namespace + type;
}
store.dispatch(type, payload, options);
},
commit: noNamespace
? store.commit
: (_type, _payload, _options) => {
const args = unifyObjectStyle(_type, _payload, _options);
let { type } = args;
const { payload, options } = args;
if (!options || !options.root) {
type = namespace + type;
}
store.commit(type, payload, options);
}
};
return local;
}
複製代碼
首先看context中的commit和dispatch方法的實現二者實現方式大同小異,咱們只分析commit,首先經過namespace判斷是否爲封裝模塊,若是是則返回一個匿名函數,該匿名函數首先進行參數的規範化,以後會調用store.dispatch,而此時的調用會將傳入的type進行偷換換成namespace+type,因此咱們在封裝的module中執行的commit[type]實際上都是調用store._mutations[namespace+type]的事件隊列
function makeLocalContext (store, namespace, path) {
const noNamespace = namespace === "";
Object.defineProperties(local, {
state: {
get: () => getNestedState(store.state, path)
},
getters: {
get: noNamespace
? () => store.getters
: () => makeLocalGetters(store, namespace)
}
});
return local;
}
複製代碼
而後是state,經過local.state訪問到的都是將path傳入getNestedState獲取到的state,實際上就是module內的state,而getters則是經過代理的方式實現訪問內部getters的
function makeLocalGetters (store, namespace) {
const gettersProxy = {}
const splitPos = namespace.length
Object.keys(store.getters).forEach(type => {
// skip if the target getter is not match this namespace
if (type.slice(0, splitPos) !== namespace) return
// extract local getter type
const localType = type.slice(splitPos)
// Add a port to the getters proxy.
// Define as getter property because
// we do not want to evaluate the getters in this time.
Object.defineProperty(gettersProxy, localType, {
get: () => store.getters[type],
enumerable: true
})
})
return gettersProxy
}
複製代碼
首先聲明一個代理對象gettersProxy,以後遍歷store.getters,判斷是否爲namespace的路徑全匹配,若是是,則將gettersProxy 的localType屬性代理至store.getters[type],而後將gettersProxy返回,這樣經過local.getters訪問的localType實際上 就是stores.getters[namespace+type]了。 如下是一個小小的總結
獲取路徑對應的命名空間(namespaced=true時拼上)->state拼接到store.state上使其成爲一個基於path的嵌套結構->註冊localContext
註冊localContext