lodash源码分析之baseOrderBy

本文为读 lodash 源码的第一百五十四篇,后续文章会更新到这个仓库中,欢迎 star:pocket-lodash

gitbook也会同步仓库的更新,gitbook地址:pocket-lodash

依赖

import baseEach from './baseEach.js'
import baseSortBy from './baseSortBy.js'
import baseGet from './baseGet.js'
import compareMultiple from './compareMultiple.js'
import isArrayLike from '../isArrayLike.js'

《lodash源码分析之baseEach》 《lodash源码分析之baseSortBy》 《lodash源码分析之baseGet》 《lodash源码分析之compareMultiple》 《lodash源码分析之isArrayLike》

源码分析

baseOrderBy 是用来实现 orderBy 的内部函数,会对参数进行一些合规化处理。

源码如下:

const identity = (value) => value

function baseOrderBy(collection, iteratees, orders) {
  if (iteratees.length) {
    iteratees = iteratees.map((iteratee) => {
      if (Array.isArray(iteratee)) {
        return (value) => baseGet(value, iteratee.length === 1 ? iteratee[0] : iteratee)
      }

      return iteratee
    })
  } else {
    iteratees = [identity]
  }

  let criteriaIndex = -1
  let eachIndex = -1

  const result = isArrayLike(collection) ? new Array(collection.length) : []

  baseEach(collection, (value) => {
    const criteria = iteratees.map((iteratee) => iteratee(value))

    result[++eachIndex] = {
      criteria,
      index: ++criteriaIndex,
      value
    }
  })

  return baseSortBy(result, (object, other) => compareMultiple(object, other, orders))
}

处理 iteratees

iteratees 可以是函数集、字符串集,如果是字符串集,则应该为属性路径,iteratee 的作用是根据传入的 value 值,返回一个新值,来作为排序时的比较值。

有多少个 iteratee,就有多少个维度的比较。

iteratees 的合规化处理源码如下:

if (iteratees.length) {
  iteratees = iteratees.map((iteratee) => {
    if (Array.isArray(iteratee)) {
      return (value) => baseGet(value, iteratee.length === 1 ? iteratee[0] : iteratee)
    }

    return iteratee
  })
} else {
  iteratees = [identity]
}

如果传入的是空数组,则 iteratees 默认为 [identity] ,即每个 iteratee 调用的时候,都是返回值本身。

如果 iteratees 传入的是 [['a', 'b', 'c']] 这样的路径数组,则表示要取 value 上的 a.b.c 路径上的值进行比较。

因为后面 iteratee 会使用函数调用的方式,因此这里使用一个函数来取值:

if (Array.isArray(iteratee)) {
  return (value) => baseGet(value, iteratee.length === 1 ? iteratee[0] : iteratee)
}

baseGet 除了支持数组路径外,也支持如 a.b.c 这样的字符串路径,这时传入的 iteratees 的值类似于 [['a.b.c']]

因此判断 iteratee.length 是否为 1 ,如果长度为 1,则将值取出,直接传入字符串给 baseGet

遍历 collection ,获取所有维度的比较值

如果 collection 为类数组,则先初始化一个长度与 collection 一致的结果集,用来保存排序后的结果。

接着,使用 baseEach 遍历 collection ,每个值 value 都传给所有的 iteratee 函数,计算出所有维度的比较值,结果存入 criteria 中,在排序的时候,会从 criteria 取出来比较。比较的过程在 compareMultiple 函数中已经分析。

最后就是调用 baseSortBy 使用 compareMultiple 作为比较函数来得到排序结果。

License

署名-非商业性使用-禁止演绎 4.0 国际 (CC BY-NC-ND 4.0)

最后,所有文章都会同步发送到微信公众号上,欢迎关注,欢迎提意见:

作者:对角另一面

results matching ""

    No results matching ""