arrayToMap.ts

import { returnItself } from '../function/returnItself'
import { ArrayKFn } from '../interface/ArrayKFn'
import { getKFn } from './getKFn'
import { ArrayDeconstruct } from '../interface/ArrayDeconstruct'

export function arrayToMap<T extends any[], K, V>(
  arr: T,
  k: ArrayKFn<ArrayDeconstruct<T>, K>,
): Map<K, ArrayDeconstruct<T>>
export function arrayToMap<T extends any[], K, V>(
  arr: T,
  k: ArrayKFn<ArrayDeconstruct<T>, K>,
  v?: ArrayKFn<ArrayDeconstruct<T>, V>,
): Map<K, V>
/**
 * 将数组映射为 Map
 * @param arr 数组
 * @param k 产生 Map 元素唯一标识的函数，或者对象元素中的一个属性名
 * @param v 产生 Map 值的函数，默认为返回数组的元素，或者对象元素中的一个属性名
 * @returns 映射产生的 map 集合
 */
export function arrayToMap<T extends any[], K, V>(
  arr: T,
  k: ArrayKFn<ArrayDeconstruct<T>, K>,
  v: ArrayKFn<ArrayDeconstruct<T>, V> = returnItself,
): Map<K, V> {
  const kFn = getKFn(k)
  const vFn = getKFn(v)
  return arr.reduce(
    (res, item, index, arr) =>
      res.set(kFn(item, index, arr), vFn(item, index, arr)),
    new Map<K, V>(),
  )
}