TypeScript：可选嵌套属性的KeyOf泛型类型约束问题

问题背景与代码 #

我有一个包含多个泛型参数的泛型函数：

function myFunction<
  L extends Q[K]["nodes"],
  Q extends Record<K, { nodes: L[] }>,
  K extends keyof Q>(
    document: Q,
    dataKey: K
  ): L[] {
  //do something with the Document to a result of type QueryResult (not relevant)
  let result: Q = {} as any;
  //now I want to get the nodes attribute type-safe
  let extracted = result[dataKey]?.["nodes"] ?? []; // should be of type ListItemType

  return extracted;
}

Q中需包含键K，Q[K]（命名为Data）至少具备如下属性：

type Data = {
  nodes?: L[]
}

但nodes和Q[K]均为可选属性。Q始终是某特定类型的子集，但含嵌套nodes字段的属性过多且类型各异，无法手动列举所有可能。需要为泛型添加哪些约束，才能确保Q[K]具备nodes属性，且L[]的类型被推断为Q[K]["nodes"]的类型？

示例与报错 #

interface Contained {
  key?: {
    nodes?: number[]
  }
}

let a : Contained = {} as any

myFunction(a, "key")

此示例中，希望L被推断为number，且无法修改Contained的定义。当前报错：

Argument of type 'Contained' is not assignable to parameter of type 'Record<"key", { nodes: unknown[]; }>'.
Types of property 'key' are incompatible.
Type '{ nodes: number[]; } | undefined' is not assignable to type '{ nodes: unknown[]; }'.
Type 'undefined' is not assignable to type '{ nodes: unknown[]; }'.

解决方案 #

问题出在原泛型约束过于严格，且泛型参数的顺序影响类型推断。我们需要调整泛型约束，适配可选属性的场景，同时让TypeScript能正确推断L的类型：

修改后的泛型函数 #

function myFunction<
  Q,
  K extends keyof Q,
  // 从Q[K]中提取nodes的元素类型，处理Q[K]为undefined的情况
  L = NonNullable<Q[K]> extends { nodes?: (infer T)[] } ? T : never
>(
  document: Q,
  dataKey: K
): L[] {
  let result: Q = {} as any;
  // 类型安全地提取nodes，自动推断为L[]
  let extracted = result[dataKey]?.nodes ?? [];

  return extracted as L[];
}

关键调整说明 #

• 泛型顺序优化：将Q和K放在泛型参数最前面，让TypeScript优先推断这两个基础类型，再基于它们推导L，提升类型推断的准确性。
• 条件类型动态提取：通过NonNullable<Q[K]> extends { nodes?: (infer T)[] } ? T : never，先排除Q[K]为undefined的情况，再用infer关键字提取nodes数组的元素类型作为L，完美适配可选属性场景。
• 放松不必要的约束：移除原有的Q extends Record<K, { nodes: L[] }>强制约束，改为动态适配Q的结构，兼容Q[K]和nodes为可选属性的场景。

验证效果 #

使用示例中的Contained接口测试：

interface Contained {
  key?: {
    nodes?: number[]
  }
}

let a : Contained = {} as any;

// L被正确推断为number，返回值类型为number[]
const nodes = myFunction(a, "key");

此时代码不再报错，类型推断完全符合预期。

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