Typescript :: 条件链接函数

Question

我正在尝试实现一组链式函数，但不知何故我卡在这里。

interface ISimpleCalculator {
  plus(value: number): this;
  minus(value: number): this;
  divide(value: number): this;
  multiply(value: number): this;
  sum(): void
}

interface ISpecialCalculator extends ISimpleCalculator {
  specialPlus(value: number): ISimpleCalculator;
  specialMinus(value: number): ISimpleCalculator;
}

let testCalculator: ISpecialCalculator;
testCalculator  
  .plus(20)
  .multiply(2)
  .specialPlus(40)  
  .plus(20)
  .minus(5)
  .specialMinus(20)  //<-- Error! Property 'specialMinus' does not exist on type 'ISimpleCalculator'.
  .sum()

我想归档链中函数的类型检查。在上面的例子中，我希望ISpecialCalculator 中的函数specialPlus 和specialMinus 只能使用一次，ISimpleCalculator 可以多次使用。我对打字稿很陌生，我一直在尝试不同的方法（高级类型（Pick 和Omit）），但到目前为止没有成功。我想知道在这种情况下是否有其他方法可以提供帮助。

Answer 1

删除一些函数很简单，你可以使用Omit<this, 'specialPlus'> 如果我们测试它几乎可以工作，如果你调用specialPlus，如果你在再次调用specialPlus之后立即调用它会出错，你但是可以在调用specialMinus后调用它

interface ISpecialCalculator extends ISimpleCalculator {
  specialPlus(value: number): Omit<this, 'specialPlus'>;
  specialMinus(value: number): Omit<this, 'specialMinus'>;
}

declare let testCalculator: ISpecialCalculator;
testCalculator  
  .specialPlus(40)
   // .specialPlus(40) // error ?
  .specialMinus(20)
  .specialPlus(40) //ok ?
  .sum()

Playground Link

这是因为Omit 在声明testCalculator 时将作用于this 类型绑定，因此specialMinus 实际上将返回Omit<ISpecialCalculator, 'specialMinus'>，即使我们之前删除了它，它仍将包含specialPlus。我们想要的是Omit 处理上一个函数返回的this 类型。如果我们使用泛型类型参数为每个调用捕获this 的实际类型，并且从这个类型参数而不是来自多态this 的Omit 方法，我们可以做到这一点。

interface ISimpleCalculator {
  plus<TThis>(this: TThis,value: number): TThis;
  minus<TThis>(this: TThis,value: number): TThis;
  divide<TThis>(this: TThis,value: number): TThis;
  multiply<TThis>(this: TThis,value: number): TThis;
  sum(): void
}

interface ISpecialCalculator extends ISimpleCalculator {
  specialPlus<TThis>(this: TThis, value: number): Omit<TThis, 'specialPlus'>;
  specialMinus<TThis>(this: TThis, value: number): Omit<TThis, 'specialMinus'>;
}

declare let testCalculator: ISpecialCalculator;
testCalculator
  .specialPlus(40)
  // .specialPlus(40) // error ?
  .specialMinus(20)
  .plus(10)
  .specialPlus(40) // also error ?
  .plus(10)
  .sum()

Playground Link

Answer 2

specialPlus(value: number): ISimpleCalculator;

当您调用此函数时，您将得到一个不再具有特殊功能的简单计算器。特殊接口也应该返回this，它应该可以工作：

interface ISpecialCalculator extends ISimpleCalculator {  
   specialPlus(value: number): this;
   specialMinus(value: number): this;
}

Answer 3

尝试以下（根据问题测试的完整代码）：

interface ISimpleCalculator {
  plus(value: number): this
  minus(value: number): this
  divide(value: number): this
  multiply(value: number): this
  sum(): void
}

interface ISpecialCalculator extends ISimpleCalculator {
  specialPlus(value: number): this
  specialMinus(value: number): this
}

let testCalculator: ISpecialCalculator
testCalculator
  .plus(20)
  .multiply(2)
  .specialPlus(40)
  .plus(20)
  .minus(5)
  .specialMinus(20) 
  .sum()

如果您想限制 special[Plus|Minus] 的使用，那么您可以在实现 ISpecialCalculator 的具体类中完成此操作界面。

下面的代码可能会给你一些想法：

class Calculator implements ISpecialCalculator {
  specialPlusUsed = false
  specialMinusUsed = false

  specialPlus(value: number): this {
    if (this.specialPlusUsed) throw new Error("SpecialPlus can be used only once!")

    this.specialPlusUsed = true
    // Related calculations here...
    return this
  }

  specialMinus(value: number): this {
    if (this.specialMinusUsed) throw new Error("SpecialMinus can be used only once!")
    this.specialMinusUsed = true
    // Related calculations here...
    return this
  }

  plus(value: number): this {
    // Related calculations here...
    return this
  }

  minus(value: number): this {
    // Related calculations here...
    return this
  }

  divide(value: number): this {
    // Related calculations here...
    return this
  }

  multiply(value: number): this {
    // Related calculations here...
    return this
  }

  sum(): void {
    // Related calculations here...
  }
}