用150行程式碼實現Vuex 80%的功能

殷榮檜發表於2019-02-12

行程Vue

作者：殷榮檜@騰訊

本文地址,歡迎檢視

本文github倉庫程式碼地址，歡迎star，謝謝。

如果你對自己用少量程式碼實現各個框架感興趣，那下面這些你都可以一看：

build-your-own-react

build-your-own-flux

build-your-own-redux

export class Store {
    constructor(options = {}, Vue) {
        this.options = options;
        Vue.mixin({ beforeCreate: vuexInit });
    }
    get state () {
        return this.options.state;
    }
}
function vuexInit () {
    const options = this.$options
    if (options.store) {
        // 元件內部設定了store,則優先使用元件內部的store
        this.$store = typeof options.store === 'function'
        ? options.store()
        : options.store
    } else if (options.parent && options.parent.$store) {
        // 元件內部沒有設定store,則從根App.vue下繼承$store方法
        this.$store = options.parent.$store
    }
}
複製程式碼

介面程式碼如下：

<script>
export default {
  name: 'app',
  created() {
    console.log('列印出this.$store.state.count的結果',this.$store.state.count);
  },
}
</script>
複製程式碼

執行結果: 成功列印出this.$store.state.count的值為0

二. vuex中的getter方法的實現

main.js程式碼如下：

let store = new Vuex.Store({
    state: {
        count: 0
    },
    getters: {
        getStatePlusOne(state) {
            return state.count + 1
        }
    }

}, Vue);

new Vue({
    store,
    render: h => h(App),
}).$mount('#app')
複製程式碼

store.js的程式碼如下：

export class Store {
    constructor(options = {}, Vue) {
        this.options = options;
        this.getters = {}
        Vue.mixin({ beforeCreate: vuexInit });
        forEachValue(options.getters, (getterFn, getterName) => {
            registerGetter(this, getterName, getterFn);
        })
    }
    get state() {
        return this.options.state;
    }
}

function registerGetter(store, getterName, getterFn) {
    Object.defineProperty(store.getters, getterName, {
        get: () => {
            return getterFn(store.state)
        }
    })
}

// 將物件中的每一個值放入到傳入的函式中作為引數執行
function forEachValue(obj, fn) {
    Object.keys(obj).forEach(key => fn(obj[key], key));
}

function vuexInit() {
    const options = this.$options
    if (options.store) {
        // 元件內部設定了store,則優先使用元件內部的store
        this.$store = typeof options.store === 'function' ?
            options.store() :
            options.store
    } else if (options.parent && options.parent.$store) {
        // 元件內部沒有設定store,則從根App.vue下繼承$store方法
        this.$store = options.parent.$store
    }
}
複製程式碼

介面程式碼如下：

<script>
export default {
  name: 'app',
  created() {
    console.log('列印出this.$store.getters.getStatePlusOne的結果',this.$store.getters.getStatePlusOne);
  },
}
</script>
複製程式碼

執行結果：成功列印出this.$store.getters.getStatePlusOne的值為1

三. mutation和commit方法的實現

main.js程式碼如下：

let store = new Vuex.Store({
    state: {
        count: 0
    },
    mutations: {
        incrementFive(state) {
            // console.log('初始state', JSON.stringify(state));
            state.count = state.count + 5;
        }
    },
    getters: {
        getStatePlusOne(state) {
            return state.count + 1
        }
    }

}, Vue);
複製程式碼

store.js的程式碼如下：

export class Store {
    constructor(options = {}, Vue) {
        Vue.mixin({ beforeCreate: vuexInit })
        this.options = options;
        this.getters = {};
        this.mutations = {};
        const { commit } = this;
        this.commit = (type) => {
            return commit.call(this, type);
        }
        forEachValue(options.getters, (getterFn, getterName) => {
            registerGetter(this, getterName, getterFn);
        });

        forEachValue(options.mutations, (mutationFn, mutationName) => {
            registerMutation(this, mutationName, mutationFn)
        });

        this._vm = new Vue({
            data: {
                state: options.state
            }
        });
    }

    get state() {
        // return this.options.state; // 無法完成頁面中的雙向繫結，所以改用this._vm的形式
        return this._vm._data.state;
    }
    commit(type) {
        this.mutations[type]();
    }
}

function registerMutation(store, mutationName, mutationFn) {
    store.mutations[mutationName] = () => {
        mutationFn.call(store, store.state);
    }
}


function registerGetter(store, getterName, getterFn) {
    Object.defineProperty(store.getters, getterName, {
        get: () => {
            return getterFn(store.state)
        }
    })
}

// 將物件中的每一個值放入到傳入的函式中作為引數執行
function forEachValue(obj, fn) {
    Object.keys(obj).forEach(key => fn(obj[key], key));
}

function vuexInit() {
    const options = this.$options
    if (options.store) {
        // 元件內部設定了store,則優先使用元件內部的store
        this.$store = typeof options.store === 'function' ?
            options.store() :
            options.store
    } else if (options.parent && options.parent.$store) {
        // 元件內部沒有設定store,則從根App.vue下繼承$store方法
        this.$store = options.parent.$store
    }
}
複製程式碼

介面程式碼如下：

<script>
export default {
  name: 'app',
  created() {
    console.log('列印出this.$store.getters.getStatePlusOne的結果',this.$store.getters.getStatePlusOne);
  },
  mounted() {
    setTimeout(() => {
      this.$store.commit('incrementFive');
      console.log('store state自增5後的結果', this.$store.state.count);
    }, 2000);
  },
  computed: {
    count() {
      return this.$store.state.count;
    }
  }
}
</script>
複製程式碼

執行結果：成功在2秒之後輸出count自增5後的結果5

四. actions和dispatch方法的實現

main.js程式碼如下：

let store = new Vuex.Store({
    state: {
        count: 0
    },
    actions: {
        countPlusSix(context) {
            context.commit('plusSix');
        }
    },
    mutations: {
        incrementFive(state) {
            // console.log('初始state', JSON.stringify(state));
            state.count = state.count + 5;
        },
        plusSix(state) {
            state.count = state.count + 6;
        }
    },
    getters: {
        getStatePlusOne(state) {
            return state.count + 1
        }
    }

}, Vue);
複製程式碼

store.js的程式碼如下：

export class Store {
    constructor(options = {}, Vue) {
        Vue.mixin({ beforeCreate: vuexInit })
        this.options = options;
        this.getters = {};
        this.mutations = {};
        this.actions = {};
        const { dispatch, commit } = this;
        this.commit = (type) => {
            return commit.call(this, type);
        }
        this.dispatch = (type) => {
            return dispatch.call(this, type);
        }
        forEachValue(options.actions, (actionFn, actionName) => {
            registerAction(this, actionName, actionFn);
        });

        forEachValue(options.getters, (getterFn, getterName) => {
            registerGetter(this, getterName, getterFn);
        });

        forEachValue(options.mutations, (mutationFn, mutationName) => {
            registerMutation(this, mutationName, mutationFn)
        });

        this._vm = new Vue({
            data: {
                state: options.state
            }
        });
    }

    get state() {
        // return this.options.state; // 無法完成頁面中的雙向繫結，所以改用this._vm的形式
        return this._vm._data.state;
    }
    commit(type) {
        this.mutations[type]();
    }
    dispatch(type) {
        return this.actions[type]();
    }
}

function registerMutation(store, mutationName, mutationFn) {
    store.mutations[mutationName] = () => {
        mutationFn.call(store, store.state);
    }
}

function registerAction(store, actionName, actionFn) {
    store.actions[actionName] = () => {
        actionFn.call(store, store)
    }
}

function registerGetter(store, getterName, getterFn) {
    Object.defineProperty(store.getters, getterName, {
        get: () => {
            return getterFn(store.state)
        }
    })
}

// 將物件中的每一個值放入到傳入的函式中作為引數執行
function forEachValue(obj, fn) {
    Object.keys(obj).forEach(key => fn(obj[key], key));
}

function vuexInit() {
    const options = this.$options
    if (options.store) {
        // 元件內部設定了store,則優先使用元件內部的store
        this.$store = typeof options.store === 'function' ?
            options.store() :
            options.store
    } else if (options.parent && options.parent.$store) {
        // 元件內部沒有設定store,則從根App.vue下繼承$store方法
        this.$store = options.parent.$store
    }
}
複製程式碼

介面程式碼如下：

export default {
  name: 'app',
  created() {
    console.log('列印出this.$store.getters.getStatePlusOne的結果',this.$store.getters.getStatePlusOne);
  },
  mounted() {
    setTimeout(() => {
      this.$store.commit('incrementFive');
      console.log('store state自增5後的結果', this.$store.state.count);
    }, 2000);
    setTimeout(() => {
      this.$store.dispatch('countPlusSix');
      console.log('store dispatch自增6後的結果', this.$store.state.count);
    }, 3000);
  },
  computed: {
    count() {
      return this.$store.state.count;
    }
  }
}
複製程式碼

執行結果：成功在3秒之後dipatch自增6輸出11

五. module方法的實現

main.js程式碼如下：

const pageA = {
    state: {
        count: 100
    },
    mutations: {
        incrementA(state) {
            state.count++;
        }
    },
    actions: {
        incrementAAction(context) {
            context.commit('incrementA');
        }
    }
}

let store = new Vuex.Store({
    modules: {
        a: pageA
    },
    state: {
        count: 0
    },
    actions: {
        countPlusSix(context) {
            context.commit('plusSix');
        }
    },
    mutations: {
        incrementFive(state) {
            // console.log('初始state', JSON.stringify(state));
            state.count = state.count + 5;
        },
        plusSix(state) {
            state.count = state.count + 6;
        }
    },
    getters: {
        getStatePlusOne(state) {
            return state.count + 1
        }
    }

}, Vue);
複製程式碼

store.js的程式碼如下：

let _Vue;
export class Store {
    constructor(options = {}, Vue) {
        _Vue = Vue
        Vue.mixin({ beforeCreate: vuexInit })
        this.getters = {};
        this._mutations = {}; // 在私有屬性前加_
        this._wrappedGetters = {};
        this._actions = {};
        this._modules = new ModuleCollection(options)
        const { dispatch, commit } = this;
        this.commit = (type) => {
            return commit.call(this, type);
        }
        this.dispatch = (type) => {
            return dispatch.call(this, type);
        }
        const state = options.state;
        const path = []; // 初始路徑給根路徑為空
        installModule(this, state, path, this._modules.root);
        this._vm = new Vue({
            data: {
                state: state
            }
        });
    }

    get state() {
        // return this.options.state; // 無法完成頁面中的雙向繫結，所以改用this._vm的形式
        return this._vm._data.state;
    }
    commit(type) {
        this._mutations[type].forEach(handler => handler());
    }
    dispatch(type) {
        return this._actions[type][0]();
    }
}

class ModuleCollection {
    constructor(rawRootModule) {
        this.register([], rawRootModule)
    }
    register(path, rawModule) {
        const newModule = {
            _children: {},
            _rawModule: rawModule,
            state: rawModule.state
        }
        if (path.length === 0) {
            this.root = newModule;
        } else {
            const parent = path.slice(0, -1).reduce((module, key) => {
                return module._children(key);
            }, this.root);
            parent._children[path[path.length - 1]] = newModule;
        }
        if (rawModule.modules) {
            forEachValue(rawModule.modules, (rawChildModule, key) => {
                this.register(path.concat(key), rawChildModule);
            })
        }
    }
}

function installModule(store, rootState, path, module) {
    if (path.length > 0) {
        const parentState = rootState;
        const moduleName = path[path.length - 1];
        _Vue.set(parentState, moduleName, module.state)
    }
    const context = {
        dispatch: store.dispatch,
        commit: store.commit,
    }
    const local = Object.defineProperties(context, {
        getters: {
            get: () => store.getters
        },
        state: {
            get: () => {
                let state = store.state;
                return path.length ? path.reduce((state, key) => state[key], state) : state
            }
        }
    })
    if (module._rawModule.actions) {
        forEachValue(module._rawModule.actions, (actionFn, actionName) => {
            registerAction(store, actionName, actionFn, local);
        });
    }
    if (module._rawModule.getters) {
        forEachValue(module._rawModule.getters, (getterFn, getterName) => {
            registerGetter(store, getterName, getterFn, local);
        });
    }
    if (module._rawModule.mutations) {
        forEachValue(module._rawModule.mutations, (mutationFn, mutationName) => {
            registerMutation(store, mutationName, mutationFn, local)
        });
    }
    forEachValue(module._children, (child, key) => {
        installModule(store, rootState, path.concat(key), child)
    })

}

function registerMutation(store, mutationName, mutationFn, local) {
    const entry = store._mutations[mutationName] || (store._mutations[mutationName] = []);
    entry.push(() => {
        mutationFn.call(store, local.state);
    });
}

function registerAction(store, actionName, actionFn, local) {
    const entry = store._actions[actionName] || (store._actions[actionName] = [])
    entry.push(() => {
        return actionFn.call(store, {
            commit: local.commit,
            state: local.state,
        })
    });
}

function registerGetter(store, getterName, getterFn, local) {
    Object.defineProperty(store.getters, getterName, {
        get: () => {
            return getterFn(
                local.state,
                local.getters,
                store.state
            )
        }
    })
}

// 將物件中的每一個值放入到傳入的函式中作為引數執行
function forEachValue(obj, fn) {
    Object.keys(obj).forEach(key => fn(obj[key], key));
}

function vuexInit() {
    const options = this.$options
    if (options.store) {
        // 元件內部設定了store,則優先使用元件內部的store
        this.$store = typeof options.store === 'function' ?
            options.store() :
            options.store
    } else if (options.parent && options.parent.$store) {
        // 元件內部沒有設定store,則從根App.vue下繼承$store方法
        this.$store = options.parent.$store
    }
}
複製程式碼

主介面程式碼如下：

<template>
  <div id="app">
    ==============主頁================<br>
    主頁數量count為: {{count}}<br>
    pageA數量count為: {{countA}}<br>
    ==========以下為PageA內容==========<br>
    <page-a></page-a>
  </div>
</template>

<script>
import pageA from './pageA';

export default {
  name: 'app',
  components: {
    pageA
  },
  created() {
    console.log('列印出this.$store.getters.getStatePlusOne的結果',this.$store.getters.getStatePlusOne);
  },
  mounted() {
    setTimeout(() => {
      this.$store.commit('incrementFive');
      console.log('store state自增5後的結果', this.$store.state.count);
    }, 2000);
    setTimeout(() => {
      this.$store.dispatch('countPlusSix');
      console.log('store dispatch自增6後的結果', this.$store.state.count);
    }, 3000);
  },
  computed: {
    count() {
      return this.$store.state.count;
    },
    countA() {
      return this.$store.state.a.count;
    }
  }
}
</script>
複製程式碼

pageA頁面如下：

<template>
    <div>
        頁面A被載入
    </div>
</template>

<script>
export default {
  name: 'pageA',
  mounted() {
      setTimeout(() => {
          this.$store.dispatch('incrementAAction');
      }, 5000)
  },
}
</script>
複製程式碼

執行結果：在5秒後A頁面觸發incrementAAction，主介面中的countA變化為101，成功

自此：基本用了150行左右的程式碼實現了vuex 80%左右的功能了，其中還有namespace等不能夠使用，其他基本都和原始碼語法相同，如果你有興趣仔細再看看，可以移步github倉庫程式碼，程式碼是建立在閱讀了vuex原始碼之後寫的，所以看完了本文的程式碼，再去看vuex的程式碼，相信你一定會一目瞭然

六. 實現：Vue.use(Vuex)

最後為了和vuex原始碼做到最相似，同樣使用Vue.use(Vuex),使用如下的程式碼進行實現：

export function install(_Vue) {
    Vue = _Vue;
    Vue.mixin({
        beforeCreate: function vuexInit() {
            const options = this.$options;
            if (options.store) {
                this.$store = options.store;
            } else if (options.parent && options.parent.$store) {
                this.$store = options.parent.$store;
            }
        }
    })
}
複製程式碼

部門正在招新，為騰訊企業產品部，隸屬CSIG事業群。福利不少，薪水很高，就等你來。有興趣請猛戳下方兩個連結。 www.lagou.com/jobs/521039… www.zhipin.com/job_detail/…

參考資料：

Build a Vuex Module

How does a minimal Vuex implementation look like?

從0開始寫一個自己的Vuex

vuex 原始碼：如何實現一個簡單的 vuex

Vue 原始碼（三） —— Vuex

淺談Vue.use

Vuex官方文件

vuex Github倉庫

150行實現Promise 90%的功能
2018-03-02
Promise
150行JavaScript程式碼實現擴增實境
2019-01-13
JavaScript
vuex 原始碼：如何實現一個簡單的 vuex
2018-02-07
Vue原始碼
20 行 JS 程式碼實現貼上板功能
2018-02-12
JS
JavaScript 實現sprintf 功能程式碼
2014-10-14
JavaScript
實現【檢視更多】功能的程式碼
2019-05-11
用 150 行 Python 程式碼寫的量子計算模擬器
2019-02-16
Python
網路為本，博睿資料NPMD用20%的投入實現80%的功能
2021-06-22
NPM
用70行程式碼實現日誌分析程式
2013-01-11
行程
用Java程式碼實現一個簡單的聊天室功能
2020-09-24
Java
80行Python程式碼搞定全國區劃程式碼
2021-02-14
Python
20 行 JS 程式碼，實現複製到剪貼簿功能
2016-05-12
JS
用ASP實現線上壓縮與解壓縮功能程式碼
2019-11-02
Redis實現排名功能的示例程式碼CRKC
2022-03-01
Redis
js利用cookie實現的計數功能程式碼
2017-03-22
JSCookie
怎樣用一行 Python 程式碼實現並行
2020-07-10
Python並行
一行Python程式碼能實現什麼喪心病狂的功能？
2021-09-11
Python
day80:luffy:簡訊sdk接入&點選獲取驗證碼&註冊功能的實現&Celery實現簡訊傳送功能
2020-11-03
Vuex 實戰：如何在大規模 Vue 應用中組織 Vuex 程式碼 | 掘金技術徵文
2016-12-26
Vue
CSS實現的換行效果程式碼
2017-02-21
CSS
10行程式碼實現微信小程式支付功能，使用小程式雲開發實現小程式支付功能（
2021-09-09
行程微信小程式
直播網站程式原始碼，採用Redis實現購物車功能
2023-03-17
網站原始碼Redis
Vuex原始碼學習（一）功能梳理
2019-03-21
Vue原始碼
手撕Vuex-Vuex實現原理分析
2023-10-28
Vue
用java實現ftp的功能
2005-07-01
JavaFTP
jQuery實現的輸入文字計數功能程式碼
2017-03-14
jQuery
微信小程式功能之全屏滾動效果的實現程式碼
2018-12-04
微信小程式
FTP客戶端c程式碼功能實現
2023-02-23
FTP客戶端C程式
WPF DataGrid分頁功能實現程式碼
2012-11-25
實現強制換行的CSS程式碼
2017-02-24
CSS
setTimeout()實現的計數器功能程式碼例項
2017-04-07
如何實現LBS軌跡回放功能？含多平臺實現程式碼
2016-09-28
HOOK大法實現不修改程式程式碼給程式新增功能
2018-02-08
Hook
運用sklearn進行主成分分析(PCA)程式碼實現
2020-08-12
PCA
精讀《用160行js程式碼實現一個React》
2018-05-27
JSReact
無程式碼實現資料回收站功能
2020-09-29
js實現的點選表格行實現背景變色程式碼
2017-03-17
JS
用pandas實現SQL功能
2021-03-08
SQL

用150行程式碼實現Vuex 80%的功能

作者：殷榮檜@騰訊

目錄：

一. 完成最簡單的通過vuex定義全域性變數，在任何一個頁面可以通過this.$store.state.count可以直接使用

二. vuex中的getter方法的實現

三. mutation和commit方法的實現

四. actions和dispatch方法的實現

五. module方法的實現

六. 實現：Vue.use(Vuex)

相關文章

用150行程式碼實現Vuex 80%的功能

作者： 殷榮檜@騰訊

目錄：

一. 完成最簡單的通過vuex定義全域性變數，在任何一個頁面可以通過this.$store.state.count可以直接使用

二. vuex中的getter方法的實現

三. mutation和commit方法的實現

四. actions和dispatch方法的實現

五. module方法的實現

六. 實現：Vue.use(Vuex)

相關文章

作者：殷榮檜@騰訊