如何使TypeScript方法返回类型与参数类型关联（非空对应）

问题背景 #

原方法定义：

public myMethod(param: MyClass | null): MyClass | null

需求：仅当参数为null时返回null，传入MyClass实例时，返回值一定是MyClass实例（不会为null）。

已尝试代码及报错 #

尝试使用泛型条件类型实现，但编译报错：

public myMethod<T extends MyClass | null>(param: T): T extends null ? null : MyClass {
  if (param === null) {
    return null;
  }

  return new MyClass(param.cnt + 1); // 示例，构造参数由入参派生
}

报错信息：MyClass 无法赋值给 T extends null ? null : MyClass

解决方案 #

方案1：添加类型断言 #

TypeScript在函数体内无法自动推导泛型条件类型的返回值，需要手动用类型断言告知编译器返回值符合预期：

public myMethod<T extends MyClass | null>(param: T): T extends null ? null : MyClass {
  if (param === null) {
    return null as T extends null ? null : MyClass;
  }

  return new MyClass(param.cnt + 1) as T extends null ? null : MyClass;
}

方案2：使用函数重载（推荐） #

通过重载明确不同参数对应的返回类型，代码更直观，编译器也能准确推断类型：

// 重载签名：定义参数与返回值的对应关系
public myMethod(param: null): null;
public myMethod(param: MyClass): MyClass;
// 实现签名：包含所有可能的参数和返回类型
public myMethod(param: MyClass | null): MyClass | null {
  if (param === null) {
    return null;
  }
  return new MyClass(param.cnt + 1);
}

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