t-comm/es/list/flat.mjs

专业、稳定、纯粹的工具库

/**
 * 递归拉平数组
 * @param list 数组
 * @returns 数组
 *
 * @example
 *
 * flat([[[1, 2, 3], 4], 5])
 *
 * // [1, 2, 3, 4, 5]
 */
function flat(list) {
  return list.reduce(function (acc, item) {
    return acc.concat(Array.isArray(item) ? flat(item) : item);
  }, []);
}
/**
 * 拉平数组，不会递归处理
 * @param {Array<Object>} list - 对象数组
 * @param {string} key - 对象的key
 * @returns {object} 拉平后的对象
 *
 * @example
 *
 * const list = [{id: 1, name: 'a'}, {id: 2, name: 'b'}]
 *
 * flatten(list, 'id')
 *
 * // {1: {id: 1, name: 'a'}, 2: {id: 2, name: 'b'}}
 *
 */
function flatten(list, key) {
  return list.reduce(function (acc, item) {
    acc[item[key]] = item;
    return acc;
  }, {});
}

export { flat, flatten };