TypeScript中如何正确返回NonNullable类型？解决类型报错问题

这个问题我太熟悉了——TypeScript在泛型场景下的控制流分析确实存在局限：你明明通过if (value == null)抛出错误确保了后续value不可能为空，但因为K是泛型类型参数，编译器没办法自动把Stuff[K]细化为NonNullable<Stuff[K]>，导致返回值类型不匹配。

不用非空断言!也能完美解决，核心是用**TypeScript断言函数（Assertion Functions）**给编译器明确的类型承诺，下面是具体方案：

第一步：定义通用的非空断言工具类 #

先封装一个可复用的工具模块，包含类型别名和断言函数，让代码更整洁：

// util/null.ts
type Null = null | undefined;
export default Null;

// 类型守卫：判断值是否非空
export function isNotNull<T>(value: T): value is NonNullable<T> {
  return value != null;
}

// 断言函数：确保值非空，否则抛出错误
export function assertNotNull<T>(value: T, throwError: (value: T) => never): asserts value is NonNullable<T> {
  if (value == null) throwError(value);
}

第二步：修改getStuff函数使用断言函数 #

把原来的手动判断抛出逻辑替换成assertNotNull，这样TypeScript就能准确识别value的非空类型：

interface Stuff { foo: number | null; bar: string | null; }
let stuff: Stuff;

function getStuff(): Stuff;
function getStuff<K extends keyof Stuff>(key: K): NonNullable<Stuff[K]>;
function getStuff<K extends keyof Stuff>(key?: K): Stuff | NonNullable<Stuff[K]> {
  if (key == null) return stuff;
  const value = stuff[key];
  
  // 用断言函数告诉编译器：执行到这里value一定是非空的
  assertNotNull(value, () => new Error(`${key} is null`));
  
  // 现在TypeScript能正确推断value的类型是NonNullable<Stuff[K]>
  return value;
}

const myStuff = getStuff('foo'); // 类型自动推断为number，无需额外空值检查

为什么这个方案有效？ #

asserts value is NonNullable<T>是TypeScript的断言函数语法，它向编译器做出承诺：当这个函数执行完成后，传入的value必然符合NonNullable<T>类型。结合泛型K，编译器能正确关联Stuff[K]和NonNullable<Stuff[K]>的类型关系，彻底消除类型错误，同时比非空断言!更严谨——它是通过运行时检查来保障类型安全的。

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