使用 TypeScript 实现类型安全的 Group By 和 Sum 操作
在处理数据集合时，按照指定字段分组并统计每组的总和是非常常见的需求。如果使用 JavaScript 实现这类逻辑，很容易因为字段名拼写错误、数据类型不匹配等问题导致运行时报错。而 TypeScript 的类型系统可以帮助我们在编译阶段就规避这些问题，实现类型安全的分组求和操作。
基础类型定义
首先我们需要定义通用的类型，确保分组字段和求和字段的类型在编译阶段就能得到校验。下面的代码定义了处理数据集合所需的基础类型：
// 定义任意对象的类型，键为字符串，值为任意类型
type AnyObject = Record<string, any>;
// 提取对象中值为数字类型的键
type NumericKeys<T extends AnyObject> = {
[K in keyof T]: T[K] extends number ? K : never;
}[keyof T];
// 提取对象中值为字符串类型的键，用于分组字段
type StringKeys<T extends AnyObject> = {
[K in keyof T]: T[K] extends string ? K : never;
}[keyof T];这里的 NumericKeys 类型会自动筛选出对象中值为 number 类型的键，避免我们把非数字字段作为求和字段；StringKeys 则筛选出值为 string 类型的键，适合作为分组依据。
实现 Group By 操作
分组操作的核心是遍历数据集合，按照指定的分组字段将数据归类到不同的组中。下面的实现会通过泛型约束，确保我们传入的分组字段确实是数据中存在的字符串类型字段：
/**
* 类型安全的 Group By 操作
* @param data 待分组的数据集合
* @param key 分组依据的字段名，必须是 data 元素中值为字符串类型的键
*/
function groupBy<T extends AnyObject>(
data: T[],
key: StringKeys<T>
): Record<string, T[]> {
const result: Record<string, T[]> = {};
for (const item of data) {
const groupKey = item[key] as string;
if (!result[groupKey]) {
result[groupKey] = [];
}
result[groupKey].push(item);
}
return result;
}当我们调用这个函数时，如果传入的分组字段不是数据中存在的字符串类型键，TypeScript 会直接报错，避免后续运行时出现分组字段不存在的问题。
实现 Sum 操作
求和操作需要针对分组后的结果，对每组中指定数字字段的值进行累加。同样通过泛型约束，确保求和字段是每组数据元素中存在的数字类型字段：
/**
* 类型安全的 Sum 操作，对分组后的每组数据计算指定字段的总和
* @param groupedData Group By 操作返回的分组结果
* @param sumKey 求和依据的字段名，必须是分组元素中值为数字类型的键
*/
function sumByGroup<T extends AnyObject>(
groupedData: Record<string, T[]>,
sumKey: NumericKeys<T>
): Record<string, number> {
const result: Record<string, number> = {};
for (const groupKey in groupedData) {
if (Object.prototype.hasOwnProperty.call(groupedData, groupKey)) {
const groupItems = groupedData[groupKey];
let total = 0;
for (const item of groupItems) {
total += item[sumKey] as number;
}
result[groupKey] = total;
}
}
return result;
}同样，如果传入的求和字段不是数字类型，或者不在数据元素的键中，TypeScript 会在编译阶段就提示错误，保证求和逻辑的合法性。
完整使用示例
下面我们用一个销售数据的例子来演示两个函数的配合使用，直观感受类型安全带来的好处：
// 定义销售数据的类型
interface SaleRecord {
category: string;
product: string;
amount: number;
date: string;
}
// 模拟销售数据集合
const salesData: SaleRecord[] = [
{ category: "电子产品", product: "手机", amount: 5000, date: "2024-01-01" },
{ category: "电子产品", product: "平板", amount: 3000, date: "2024-01-02" },
{ category: "家居用品", product: "台灯", amount: 200, date: "2024-01-01" },
{ category: "家居用品", product: "椅子", amount: 800, date: "2024-01-03" },
{ category: "电子产品", product: "耳机", amount: 1000, date: "2024-01-04" },
];
// 按照 category 字段分组，这里如果写错字段名比如 "cate"，TypeScript 会直接报错
const groupedByCategory = groupBy(salesData, "category");
console.log("分组结果:", groupedByCategory);
// 计算每个品类的销售总额，这里如果写 "product" 作为求和字段，因为 product 是字符串类型，TypeScript 会报错
const categorySalesSum = sumByGroup(groupedByCategory, "amount");
console.log("品类销售总额:", categorySalesSum);运行上面的代码，我们会得到正确的分组和求和结果：
- 分组后电子产品组有3条数据，家居用品组有2条数据
- 电子产品组总销售额为 5000 + 3000 + 1000 = 9000
- 家居用品组总销售额为 200 + 800 = 1000
如果尝试把分组字段改成不存在的 "cate"，或者把求和字段改成字符串类型的 "product"，TypeScript 都会立刻提示类型错误，不用等到代码运行才发现逻辑问题。
总结
通过 TypeScript 的泛型和类型推导，我们可以让 Group By 和 Sum 这类数据操作具备完整的类型安全能力。相比纯 JavaScript 实现，这种方式能在开发阶段就规避字段名错误、类型不匹配等常见问题，提升代码的健壮性和可维护性，尤其适合处理复杂的数据集合场景。
TypeScriptGroup_BySum类型安全数据处理 本作品最后修改时间：2026-05-22 16:23:50