javascript - typeScript 安全的 groupBy

Question

鉴于所有已经存在的答案，我们尝试创建一个不会提示 TypeScript 错误的 groupBy 箭头函数。在 this answer 的指导下和 this one我们已经有了这个工作代码:

const groupBy = <TItem>(
  items: TItem[],
  key: string
): { [key: string]: TItem[] } =>
  items.reduce(
    (result, item) => ({
      ...result,
      [item[key]]: [...(result[item[key]] || []), item],
    }),
    {}
  )

然而，它仍然提示:

确实有可能使用的 key 是空的。所以TypeScript在这里并没有错。所以在那种情况下，它应该简单地使用占位符值，如 NA 或其他东西。

为此 groupBy 函数创建可重用的无错误 typescript 版本的正确方法是什么？

Answer 1

使用一些泛型来降低键类型并强制对象包含键:

type ObjectKey = string | number | symbol;

const groupBy = <K extends ObjectKey, TItem extends Record<K, ObjectKey>>(
  items: TItem[],
  key: K
): Record<ObjectKey, TItem[]> =>
  items.reduce(
    (result, item) => ({
      ...result,
      [item[key]]: [...(result[item[key]] || []), item],
    }),
    {} as Record<ObjectKey, TItem[]>
  );

Playground link