TypeScript：如何通过泛型键约束对象属性为数组并消除类型断言？

我需要实现一个无需类型断言的insert函数，要求仅允许操作值为数组的对象属性，不满足该约束的对象-键组合无法通过类型检查。

初始代码存在类型断言：

function insert<O extends object, K extends keyof O>(parent: O, key: K) {
  type Item = O[K] extends Array<infer E> ? E : never;
  return (entry: Item) => (key in parent)
    ? (parent[key] as Item[]).push(entry)
    : (parent[key] as Item[]) = [entry];
}

修改后的代码尝试添加约束，但仍需as O[K]断言，否则触发TypeScript错误：

'I[]' is assignable to the constraint of type 'O[K]', but 'O[K]' could be instantiated with a different subtype of constraint 'I[]'. ts(2322)

修改后的代码：

function insert<I, K extends keyof O, O extends { [key in K]?: I[]; }>(parent: O, key: K, item: I) {
  (key in parent)
    ? parent[key].push(item)
    : parent[key] = [item] as O[K];
}

无断言实现方案 #

通过精确的泛型约束，让TypeScript明确对象属性的类型为数组或undefined，从而无需类型断言即可安全赋值：

function insert<O extends object, K extends keyof O>(
  parent: O & { [P in K]: Array<any> | undefined },
  key: K,
  item: O[K] extends Array<infer E> ? E : never
) {
  if (parent[key]) {
    parent[key].push(item);
  } else {
    parent[key] = [item];
  }
}

代码说明

1. 泛型约束O & { [P in K]: Array<any> | undefined }确保parent[key]只能是数组或undefined，直接排除非数组类型的属性；
2. 参数item的类型通过条件类型自动推导为目标数组的元素类型，保证传入值与数组元素类型匹配；
3. 赋值[item]时，TypeScript能识别其类型与parent[key]的约束完全匹配，因此无需额外断言。

测试验证

// 合法场景：操作数组属性
const numObj = { nums: [1, 2] };
insert(numObj, 'nums', 3); // 正常执行，numObj.nums变为[1,2,3]

const strObj: { strs?: string[] } = {};
insert(strObj, 'strs', 'hello'); // 正常执行，strObj.strs被赋值为['hello']

// 非法场景：操作非数组属性（触发类型错误）
const invalidObj = { age: 25 };
insert(invalidObj, 'age', 30); // TypeScript报错：类型不匹配

内容的提问来源于stack exchange，提问作者kamov