如何实现相似 lodash 的 get 与 merge 函数

lodash 基本上成为了写 javascript 工具库的标配，它普遍应用在各类服务端以及前端应用中，可是它的包体积略大了一些。对于服务端来讲，包的体积并非十分的重要，或者换句话说，不像前端那样对包的体积特别敏感，一分一毫都会影响页面打开的性能，从而影响用户体验。

正由于前端包体积对于用户体验的重要性，所以有各类各样减少包体积的方法。针对 lodash 来讲，你彻底没必要要引入 lodash 的全部工具函数，你只须要按需引入或者直接使用单函数包。关于按需引入你能够参考如下文章

Lessons on tree-shaking Lodash with Webpack and Babel

在针对个人我的站点中的 lodash 进行优化时，若是没记错的话，lodash 从之前 gzip 后的 80KB 变为了 20KB，相对来讲仍是比较大。而当我全局搜索了 lodash 的引用以后，发现 90% 的场景都是在使用 _.get。

另外，随着 ES6+ 的发展，以及浏览器与 Node 对它的支持，不少 lodash 的函数都很容易本身来实现或者说已被实现，如 _.assign，_.trim，_.startsWith 等等已被 ES6+ 实现，而 _.uniq 又很容易经过 new Set() 来解决。有人就在 github 上总结了 you-dont-need/You-Dont-Need-Lodash-Underscore，其中囊括了不少工具函数很简易的实现。

鉴于本站点就是我做为试验田用来实践各类技术，因而我决定本身来实现 lodash 的一些工具函数。get 与 merge 两个函数在我使用时比较多，且相对来讲比较复杂一些，这里贴一下个人实现代码。

本文地址: shanyue.tech/post/lodash…

get

在 js 中常常会出现嵌套调用这种状况，如 a.b.c.d.e，可是这么写很容易抛出异常。你须要这么写 a && a.b && a.b.c && a.b.c.d && a.b.c.d.e，可是显得有些啰嗦与冗长了。特别是在 graphql 中，这种嵌套调用更是难以免。

这时就须要一个 get 函数，使用 get(a, 'b.c.d.e') 简单清晰，而且容错性提升了不少。如下是须要经过的几个测试用例

get({ a: null }, 'a.b.c', 3)
// output: 3

get({ a: undefined }, 'a', 3)
// output: 3

get({ a: null }, 'a', 3)
// output: 3

get({ a: [{ b: 1 }]}, 'a[0].b', 3)
// output: 1
复制代码

path 中也多是数组的路径，所有转化成 . 运算符并组成数组

// a[3].b -> a.3.b
const paths = path.replace(/\[(\d+)\]/g, '.$1').split('.')
复制代码

而后层层迭代属性便可，另外注意 null 与 undefined 取属性会报错，因此使用 Object 包装一下。

function get (source, path, defaultValue = undefined) {
  // a[3].b -> a.3.b
  const paths = path.replace(/\[(\d+)\]/g, '.$1').split('.')
  let result = source
  for (const p of paths) {
    result = Object(result)[p]
    if (result === undefined) {
      return defaultValue
    }
  }
  return result
}
复制代码

merge

merge 用来递归合并对象，至关于深层的 Object.assign。在 graphql 中会普遍用到 merge，如会常用 merge 来合并全部的 resolver，特别是 Mutation 以下示例

const rootResolver = {
  Query: {
  
  },
  Mutation: {
    login () {}
  }
}

const userResolver = {
  User: {
    createUser() {}
  }
}

const resolver = merge(rootResolver, userResolver)
// output
// {
// Query: {},
// Mutation: {
// login () {},
// createUser () {}
// }
// }
复制代码

另外，在前端进行 graphql 的查询时也常常须要使用到 merge。如在进行页面的性能优化时，为了不一个 Query 耗时太久，页面渲染过于耗时，会拆成两个 Query，先渲染响应快的数据，在慢慢等待个别响应慢的数据。

如下是一个关于我的主页信息的 Query，可是其中有一个字段 dataNeedDelay3s 会在服务器耗时许久，会由于此字段加大了用户的等待时间，形成不友好的用户体验。此时会把此字段单独拆掉，优先渲染其它我的信息。

query PROFILE {
  me {
    id
    age
    name
    # 须要耗时3s的字段
    dataNeedDelay3s
  }
}

# 拆为如下两个茶轩
query PROFILE_ONE {
  me {
    id
    age
    name
  }
}

query PROFILE_TWO {
  me {
    dataNeedDelay3s
  }
}
复制代码

此时就有 merge 的需求，查询完成后把两次查询结果给拼到一块儿。

关于拆 graphql 的 Query 的需求无处不在，如在服务端渲染时，须要把权限资源与非权限资源分开。

这里讲述下如何实现 merge

function isObject (value) {
  const type = typeof value
  return value !== null && (type === 'object' || type === 'function')
}

// { a: [{ b: 2 }] } { a: [{ c: 2 }]} -> { a: [{b:2}, {c:2}]}
// merge({o: {a: 3}}, {o: {b:4}}) => {o: {a:3, b:4}}
function merge (source, other) {
  if (!isObject(source) || !isObject(other)) {
    return other === undefined ? source : other
  }
  // 合并两个对象的 key，另外要区分数组的初始值为 []
  return Object.keys({
    ...source,
    ...other
  }).reduce((acc, key) => {
    // 递归合并 value
    acc[key] = merge(source[key], other[key])
    return acc
  }, Array.isArray(source) ? [] : {})
}
复制代码