JSkeyBy方法实现数组转对象技巧

本文详细介绍了使用 JavaScript 将对象数组转换为键值对象的 `keyBy` 方法。核心在于利用 `reduce` 函数，通过指定属性作为键，原始对象作为值，构建键值映射。文章深入探讨了处理 key 值冲突的多种策略，包括默认覆盖、合并为数组等。此外，还针对缺失 key 的情况提供了解决方案，可以跳过或指定默认键进行存储。对于嵌套属性作为 key 的情况，文章也给出了通过 `split` 和循环访问的处理方法。通过本文，开发者可以灵活运用 `keyBy` 方法，高效地将 JavaScript 对象数组转换为所需的键值对象，提升数据处理效率。

核心答案是使用 reduce 方法将数组转换为键值对象，1. 默认后一个对象覆盖前一个处理 key 冲突；2. 可通过 merge 参数将冲突值存储为数组；3. 对于缺失 key 的情况可指定默认键存储；4. 支持通过 split 和循环访问嵌套属性作为 key。转换结果以指定属性值为键，原对象为值，完整形成键值映射对象。

将 JavaScript 对象数组转换为键值对象，核心在于利用数组的 reduce 方法，根据指定的 key 提取每个对象的值作为新对象的键，对象本身作为值。 简单来说，就是把数组里的每个东西，按照你指定的属性，变成一个对象的 key。

function keyBy(array, key) {
  return array.reduce((obj, item) => {
    obj[item[key]] = item;
    return obj;
  }, {});
}

// 示例
const users = [
  { id: '1', name: 'Alice' },
  { id: '2', name: 'Bob' },
  { id: '3', name: 'Charlie' }
];

const usersById = keyBy(users, 'id');

console.log(usersById);
// 输出:
// {
//   '1': { id: '1', name: 'Alice' },
//   '2': { id: '2', name: 'Bob' },
//   '3': { id: '3', name: 'Charlie' }
// }

如何处理 key 值冲突的情况？

默认情况下，后面的对象会覆盖前面的对象。 如果你需要处理冲突，可以在 reduce 函数中加入判断逻辑。 比如，你可以选择保留第一个对象，或者将冲突的对象合并成一个数组。

function keyBy(array, key, merge = false) {
  return array.reduce((obj, item) => {
    const keyValue = item[key];
    if (obj[keyValue]) {
      if (merge) {
        if (Array.isArray(obj[keyValue])) {
          obj[keyValue].push(item);
        } else {
          obj[keyValue] = [obj[keyValue], item];
        }
      }
      // 如果不merge，就直接覆盖，这是默认行为
    } else {
      obj[keyValue] = item;
    }
    return obj;
  }, {});
}

// 示例，模拟id冲突
const users = [
  { id: '1', name: 'Alice' },
  { id: '2', name: 'Bob' },
  { id: '1', name: 'Charlie' }
];

const usersById = keyBy(users, 'id', true); // 开启merge

console.log(usersById);
// 输出:
// {
//   '1': [ { id: '1', name: 'Alice' }, { id: '1', name: 'Charlie' } ],
//   '2': { id: '2', name: 'Bob' }
// }

如果 key 的值不存在怎么办？

如果指定的 key 在某些对象中不存在，那么对应的值将会是 undefined。 你可以在 reduce 函数中添加一个检查，跳过这些对象，或者赋予一个默认值。

function keyBy(array, key, defaultValue = null) {
  return array.reduce((obj, item) => {
    const keyValue = item[key];
    if (keyValue === undefined || keyValue === null) {
      // 可以选择跳过，或者赋予默认值
      // return obj; // 跳过
      obj[defaultValue] = item; // 赋予默认值，所有缺失key的对象都会被归到defaultValue下
    } else {
      obj[keyValue] = item;
    }
    return obj;
  }, {});
}

// 示例，模拟key缺失
const users = [
  { id: '1', name: 'Alice' },
  { name: 'Bob' }, // 缺少 id
  { id: '3', name: 'Charlie' }
];

const usersById = keyBy(users, 'id', 'unknown');

console.log(usersById);
// 输出:
// {
//   '1': { id: '1', name: 'Alice' },
//   'unknown': { name: 'Bob' },
//   '3': { id: '3', name: 'Charlie' }
// }

如何处理 key 是嵌套属性的情况？

如果 key 是嵌套的，比如 address.city，你可以使用 lodash 的 _.get 方法，或者自己实现一个简单的函数来获取嵌套属性的值。

function keyBy(array, key) {
  return array.reduce((obj, item) => {
    // 简单实现嵌套属性的获取
    const keys = key.split('.');
    let value = item;
    for (const k of keys) {
      if (value && typeof value === 'object' && k in value) {
        value = value[k];
      } else {
        value = undefined;
        break;
      }
    }

    if (value !== undefined) {
      obj[value] = item;
    }
    return obj;
  }, {});
}

// 示例，模拟嵌套key
const users = [
  { id: '1', name: 'Alice', address: { city: 'New York' } },
  { id: '2', name: 'Bob', address: { city: 'Los Angeles' } },
  { id: '3', name: 'Charlie', address: { city: 'New York' } }
];

const usersByCity = keyBy(users, 'address.city');

console.log(usersByCity);
// 输出:
// {
//   'New York': { id: '3', name: 'Charlie', address: { city: 'New York' } },
//   'Los Angeles': { id: '2', name: 'Bob', address: { city: 'Los Angeles' } }
// }

理论要掌握，实操不能落！以上关于《JSkeyBy方法实现数组转对象技巧》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！