描述

数组转成 Map v3.0.9+

1.示例

import { arrayToMap } from 'sf-utils2'

const list = [
 { empId: 1, empName: '蔡徐坤'},
 { empId: 2, empName: '小明'}，
 { empId: 3, empName: '小红'}，
]

// 根据属性名转化
arrayToMap(list, 'empId')
// 输出
[[Entries]]
  0: {1 => Object}
  1: {2 => Object}
  2: {3 => Object}
size: 3


// 不传属性名
const list = [1, 2, 3, '4']
arrayToMap(list)
// 输出
[[Entries]]
  0: {1 => 1}
  1: {2 => 2}
  2: {3 => 3}
  3: {3 => '4'}
size: 4


// 返回属性值为数组
arrayToMap(list, 'empId', {valueType: 'array'})
// 输出
[[Entries]]
  0: {1 => [{empId: 1, empName: '蔡徐坤'}]}
  1: {2 => [{empId: 2, empName: '小明'}]}
  2: {3 => [{empId: 3, empName: '小红'}]}
size: 3


// 通过function，函数自定义
arrayToMap(list, (map, item, index) => {
    map.set(item.empId, item)
})

// 输出
[[Entries]]
  0: {1 => [{empId: 1, empName: '蔡徐坤'}]}
  1: {2 => [{empId: 2, empName: '小明'}]}
  2: {3 => [{empId: 3, empName: '小红'}]}
size: 3


// 多个属性名
const list = [
  { empId: 1, empName: '蔡徐坤' },
  { empId: 2, empName: '小明' },
  { empId: 3, empName: '蔡徐坤' },
]
console.log(arrayToMap(list, ['empId', 'empName']))
// =>
// [
//  ['1|蔡徐坤', {empId: 1, empName: '蔡徐坤'}],
//  ['2|小明', {empId: 2, empName: '小明'}],
//  ['3|蔡徐坤', {empId: 3, empName: '蔡徐坤'}]
// ]

2.入参说明

参数	说明	类型	是否必填
array	原数组	Array	--------
property	属性名或回调函数	Array、Function、string[]	--------
options	配置参数详情见下 👇	Object	--------

options 对象

参数	说明	类型	是否必填	默认值
valueType	每一项返回值类型是对象还是数组，前提建立当设置 proerty 是字符串有值时	object\|array	否	object
retainKeyWithNull	是否保留未定义的属性值	boolean v3.1.4+	否	false

3.源码

源码，点开查看 👈

import isArray from '@/base/isArray'
import isFunction from '@/base/isFunction'
import isString from "@/base/isString";
import isNullable from "@/base/isNullable";

/**
 * 根据数组 从对象中获取属性，并且以｜进行分割
 * @param {object} object
 * @param {string|string[]} properties
 * @returns {*}
 */
export const getPropValue = (object, properties) => {
  if (isString(properties)) return object?.[properties]
  return properties.map(prop => object?.[prop]).filter(Boolean).join(`|`)
}

/**
 * 将数组转成Map类型
 * @param {Object[]} array 数组
 * @param {string|Function|string[]} [property] 主键
 * @param {{valueType?: 'object' | 'array', retainKeyWithNull?: boolean}} [options] 是否保留未定义的属性值
 * @return {Map<any, any>|{}|*}
 * @example
 *
 * const list = [
 *  {empId: 1, empName: '蔡徐坤'},
 *  {empId: 2, empName: '小明'}
 * ]
 * arrayToMap(list, 'empId')
 * =>
   [[Entries]]
    0: {1 => Object}
    1: {2 => Object}
   size:2
 *
 * const list = [
 *  1, 2, 3, '4'
 * ]
 * arrayToMap(list)
   =>
   [[Entries]]
   0: {1 => 1}
   1: {2 => 2}
   2: {3 => 3}
   3: {'4' => '4'}
   size:4
 *
 */
function arrayToMap(array = [], property, options = { valueType: 'object', retainKeyWithNull: false }) {
  if (isArray(array)) {
    // 如果property是Function
    if (isFunction(property)) {
      const map = new Map()
      array.forEach((v, vi) => property(map, v, vi))
      return map
    }
    const valueType = String(options?.valueType).toLowerCase() || 'object'
    // 非Function，且property存在
    if (property) {
      return array.reduce((pre, cur) => {
        let value = getPropValue(cur, property)
        if (options.retainKeyWithNull && isNullable(value))  value = 'undefined'
        if (value) {
          if (valueType === 'array') {
            if (!isArray(pre.get(value))) pre.set(value, [])
            pre.get(value).push(cur)
          } else {
            pre.set(value, cur)
          }
        }
        return pre
      }, new Map())
    }
    // property不存在时
    return array.reduce((pre, cur) => {
      if (options.retainKeyWithNull && isNullable(value)) cur = 'undefined'
      if (cur) {
        if (valueType === 'array') {
          if (!isArray(pre.get(cur))) pre.set(cur, [])
          pre.get(cur).push(cur)
        } else {
          pre.set(cur, cur)
        }
      }
      return pre
    }, new Map())
  }
  return new Map()
}
export default arrayToMap