vuex 源码：深入 vuex 之辅助函数 mapState
前言
当一个组件要获取多个 state 的时候，声明计算属性就会变得重复和冗余了。我们可以使用到辅助函数 mapState 来更快更简洁地生成计算属性。
所以我们得清楚，mapState 的作用就是帮我们把一个对象或数组里的值转化成计算属性的写法。同理，其它的辅助函数也是大同小异，只要知道了 mapState 的实现，其它的也基本都明白了。
注：本次阅读的是 vuex 的 2.0.0 版本，源码请戳 这里。
准备
解读前，我们需要熟悉一些方法的使用：
解读
先来 mapState 的使用方式：
import { mapState } from 'vuex'
export default {
computed: mapState([
// 映射 this.count 为 store.state.count
'count'
])
}
mapState 返回一个对象，我们知道以上的代码最后会变成这样：
import { mapState } from 'vuex'
export default {
computed: {
count () {
return this.$store.state.count
}
}
}
那么我们就来开始看看 mapState 做了什么处理。
还是从 vuex 的 APi 看起，打开 src/index.js 文件，最下面的代码中可以看到 vuex 暴露出的 mapState：
export default {
Store,
install,
mapState,
mapMutations,
mapGetters,
mapActions
}
定位后可以找到最前面的引入：
import { mapState, mapMutations, mapGetters, mapActions } from './helpers'
打开 src/helpers.js 文件，里面便有 mapState 的实现。
normalizeMap
想知道 mapStat 这个方法的实现，还得知道里面的 normalizeMap 这个方法的实现。定位找到 normalizeMap 方法：
function normalizeMap (map) {
return Array.isArray(map)
? map.map(key => ({ key, val: key }))
: Object.keys(map).map(key => ({ key, val: map[key] }))
}
这个方法主要是格式化 mapState 传进来的 states 参数。我们会知道 states 参数会是两种形式，一种是以数组的方式传入，另一种则是以对象的方法传入。
例如以下代码：
// 以数组的方式传入
mapState([
'count',
'add'
])
// 以对象的方法传入
mapState({
count: state => state.count,
countAlias: 'count'
})
经过 normalizeMap 方法处理后会变成这样：
// 以数组的方式传入
[
{
key: 'count',
val: 'count'
},
{
key: 'add',
val: 'add'
}
]
// 以对象的方法传入
[
{
key: count,
val: state => state.count
},
{
key: countAlias,
val: 'count'
}
]
mapState
知道了 normalizeMap 方法的实现，再回头看 mapState 方法的实现：
export function mapState (states) {
const res = {}
normalizeMap(states).forEach(({ key, val }) => {
res[key] = function mappedState () {
return typeof val === 'function'
? val.call(this, this.$store.state, this.$store.getters)
: this.$store.state[val]
}
})
return res
}
对刚刚 normalizeMap 格式化后返回的数组进行遍历，拼接一个符合 computed 的对象（需有返回值）。
对 normalizeMap 返回数组的对象里的 val 有两个判断。如果不是函数，直接查找 this.$store.state[val] 返回 state。如果是函数，则需要使用 call 将 val 这个函数的 this 指向 vue 实例，然后将 state 和 getters 传入，最后执行 val 函数。
val 函数
如果 val 是函数，可能有点难理解，举个例子，传入的参数可能是这样子的：
computed: mapState({
countPlusLocalState (state) {
return state.count + this.localCount
}
})
经过 normalizeMap 方法后返回的对象为：
[
{
key: 'countPlusLocalState',
val: function (state) {
return state.count + this.localCount
}
}
]
再经过 mapState 最后返回的 res 是。这里会将 val 函数执行一遍，将函数的返回值 return 出来。
{
countPlusLocalState: function mappedState () {
return this.$store.state.count + this.localCount
}
}
至此，mapState 的解读已经结束了。另外，mapState 还经常使用到 es6 的扩展运算符，这个不是 vuex 的实现，而是 es6 的一个新特性：
computed: {
localComputed () { /* ... */ },
// 使用对象展开运算符将此对象混入到外部对象中
...mapState({
// ...
})
}
例外需要注意，如果要使用 es6 的扩展运算符，还需要引入一个 babel 包：babel-plugin-transform-object-rest-spread。
总结
mapState 的代码不多，主要的功能就是将传入的数组或对象转成 computed 计算属性能够识别的代码。比较难理解的就是对象带有函数的返回值有点绕，多看几遍理解理解。
mapState 里面的代码实现非常的简洁精湛，主要用到了 js 的一些内置函数做处理，如果是我，估计会一直写 for 循环实现吧哈哈，从中收获到不少知识的。至此，vuex 的解读算告一段落。
