TypeScript：如何让方法返回类型与输入类型一致并消除冗余

嘿，这两个问题其实挺典型的，咱们一步步来搞定：

1. 修复返回类型无法精准推导的问题 #

当前你的方法返回Point | Vector联合类型，是因为TypeScript没办法自动把输入参数的具体类型和返回类型做绑定。最直观的解决方案是用函数重载，明确告诉TS：传入Point就返回Point，传入Vector就返回Vector。

示例代码如下：

class Point {
  constructor(public x: number, public y: number, public z: number) {}
}

class Vector {
  constructor(public x: number, public y: number, public z: number) {}
}

// 假设sum是Point类的方法
class Point {
  // 函数重载签名：定义不同输入对应的输出类型
  sum(o: Point): Point;
  sum(o: Vector): Vector;
  // 实际实现的逻辑签名
  sum(o: Point | Vector): Point | Vector {
    // 这里写具体逻辑
  }
}

这样当你调用point.sum(new Point(1,2,3))时，TypeScript会自动推断返回类型是Point，而不是模糊的联合类型。

2. 消除代码冗余 #

因为Point和Vector的结构完全一致（都有x/y/z属性），创建实例的逻辑只是类名不同，咱们可以把重复的计算逻辑抽离出来，只写一次：

优化后的完整代码：

// 定义公共接口，让Point和Vector实现它（可选，但能让结构更清晰）
interface XYZ {
  x: number;
  y: number;
  z: number;
}

class Point implements XYZ {
  constructor(public x: number, public y: number, public z: number) {}

  // 重载签名
  sum(o: Point): Point;
  sum(o: Vector): Vector;
  // 实现逻辑
  sum(o: Point | Vector): Point | Vector {
    // 抽离重复的计算逻辑，只写一次
    const newX = this.x + o.x;
    const newY = this.y + o.y;
    const newZ = this.z + o.z;

    // 根据参数类型返回对应实例
    if (o instanceof Point) {
      return new Point(newX, newY, newZ);
    } else {
      return new Vector(newX, newY, newZ);
    }
  }
}

class Vector implements XYZ {
  constructor(public x: number, public y: number, public z: number) {}
}

这样就把重复的x/y/z计算逻辑合并了，代码更简洁也更易维护。如果后续需要修改计算规则，只改一处就行。

额外优化思路 #

如果Point和Vector的逻辑高度相似，还可以考虑让它们继承同一个基类，比如：

abstract class XYZBase {
  constructor(public x: number, public y: number, public z: number) {}
}

class Point extends XYZBase {}
class Vector extends XYZBase {}

这样后续扩展其他类似类时，sum方法的逻辑复用性会更强。

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