TypeScript能否实现类似Rust Extension Traits的类扩展功能？

在TypeScript中，你可以通过两种主要方式实现类似Rust扩展 trait 的功能，无需修改外部库的原始类代码：

一、原型扩展 + 类型声明（最接近Rust体验） #

直接给类的原型添加方法，同时扩展TypeScript的类型声明，让类型系统认可新增的方法，这样所有该类的实例（包括外部库返回的）都能直接调用扩展方法。

实现代码 #

// lib.ts（外部库，不可修改）
export class B {}

export class A {
  static getB(): B { return new B(); }
}

// index.ts（你的代码）
import { A, B } from "./lib";

// 1. 给B的原型添加实际方法
B.prototype.helloWorld = function () {
  console.log("Hello World");
};

// 2. 扩展B的类型声明，让TypeScript识别新方法
declare module "./lib" {
  interface B {
    helloWorld(): void;
  }
}

// 直接调用外部库返回的B实例的扩展方法
A.getB().helloWorld(); // 正常输出 "Hello World"

优缺点 #

• 优点：所有B实例自动拥有扩展方法，无需额外转换，和Rust扩展trait的使用体验一致
• 缺点：修改全局原型可能引发冲突；若B实例被冻结（Object.freeze），此方法失效；属于全局修改，隔离性较差

二、包装器模式（安全隔离） #

如果不想修改原型，可以创建一个包装类，将外部库的B实例包裹起来，在包装类中实现扩展方法。需要将外部库返回的B实例转换为包装类实例后使用。

实现代码 #

// index.ts（你的代码）
import { A, B } from "./lib";

// 创建包装类，封装B实例并添加扩展方法
class BWrapper {
  constructor(private readonly innerB: B) {}

  helloWorld() {
    console.log("Hello World");
  }

  // 可选：转发B原有的方法，保留原有功能
  getOriginalB(): B {
    return this.innerB;
  }
}

// 工具函数：快速将B实例转换为包装类
function wrapB(b: B): BWrapper {
  return new BWrapper(b);
}

// 使用时转换实例
wrapB(A.getB()).helloWorld(); // 正常输出 "Hello World"

// 可选：扩展外部类A，让它直接返回包装后的实例
class AExtended {
  static getB(): BWrapper {
    return wrapB(A.getB());
  }
}

// 直接调用扩展后的方法
AExtended.getB().helloWorld();

优缺点 #

• 优点：不修改原型，避免全局冲突；扩展逻辑独立，隔离性好；可创建多个不同的包装类实现不同扩展
• 缺点：需要手动转换实例，或重写返回B的方法；若大量外部方法返回B，需逐个处理（可通过工具函数批量转换优化）

总结 #

• 追求和Rust扩展trait一致的体验，优先选择原型扩展+类型声明
• 注重代码隔离性、避免全局修改，优先选择包装器模式

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