TypeScript遍历对象时如何根据键推断值类型？

你的问题核心在于：普通遍历或简单生成器返回的键值对会被TypeScript拆解为独立的联合类型，失去键与值的类型关联，导致无法通过键的判断缩小值的类型。下面提供两种可行的解决方案：

方法1：用泛型回调函数封装遍历逻辑 #

通过将遍历逻辑封装到泛型函数中，让回调参数保持键与值的类型关联，TypeScript就能正确推断对应类型：

interface MyInterface {
  name: string;
  age: number;
  hobbies?: string[];
}

const obj: MyInterface = {
  name: 'John',
  age: 23,
};

function iterateWithType<T>(obj: T, handler: <K extends keyof T>(key: K, value: T[K]) => void) {
  for (const key in obj) {
    handler(key as keyof T, obj[key as keyof T]);
  }
}

// 调用时TypeScript能正确识别每个key对应的value类型
iterateWithType(obj, (key, value) => {
  if (key === 'age') {
    value; // 类型为number
    value.toFixed(); // 可正常调用number方法
  } else if (key === 'name') {
    value; // 类型为string
    value.toUpperCase(); // 可正常调用string方法
  } else if (key === 'hobbies') {
    value; // 类型为string[] | undefined
    value?.forEach(hobby => console.log(hobby));
  }
});

方法2：结合Object.entries与类型守卫 #

如果偏好使用for...of循环，可以对Object.entries的结果做类型断言，再通过自定义类型守卫关联键值类型：

interface MyInterface {
  name: string;
  age: number;
  hobbies?: string[];
}

const obj: MyInterface = {
  name: 'John',
  age: 23,
};

// 自定义类型守卫，确认键值对的具体类型
function isKeyValPair<K extends keyof MyInterface>(
  pair: [string, unknown],
  key: K
): pair is [K, MyInterface[K]] {
  return pair[0] === key;
}

for (const [key, value] of Object.entries(obj) as [keyof MyInterface, MyInterface[keyof MyInterface]][]) {
  if (isKeyValPair([key, value], 'age')) {
    value; // 类型为number
  } else if (isKeyValPair([key, value], 'name')) {
    value; // 类型为string
  } else if (isKeyValPair([key, value], 'hobbies')) {
    value; // 类型为string[] | undefined
  }
}

原生成器方法失效的原因 #

你之前的生成器返回Generator<[K, T[K]]>，其中K是keyof T，实际生成的是元组联合类型（如["name", string] | ["age", number]）。但在for...of循环中，TypeScript会将键和值拆解为独立的联合类型（键为"name" | "age" | "hobbies"，值为string | number | string[] | undefined），失去了两者的类型关联，因此无法通过键的判断自动缩小值的类型。

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