如何使用 Typescript 从枚举输入中推断方法返回类型？

Question

我有一个 AudioProcessor 枚举：

export enum AudioProcessors {
  BiquadFilter = 'BiquadFilter',
  StereoPanner = 'StereoPanner',
  DynamicsCompressor = 'Dynamics​Compressor​',
  None = 'None'
}

还有一个 AudioProcessorNode 类型：

export type AudioProcessorNode = BiquadFilterNode | StereoPannerNode | DynamicsCompressorNode;

我想创建一个类方法，它接受枚举并返回对应的节点（来自某些类属性），但我不确定如何让 Typescript 推断类型以便：

public getProcessor(audioProcessor: AudioProcessors): AudioProcessorNode {        
   throw new Error(`Method not implemented. ${audioProcessor}`);
}

audioClass.getProcessor(AudioProcessors.BiquadFilter) // returns BiquadFilterNode 

看起来我需要像ReturnType 这样的东西，但我不知道如何将它放在我的方法签名中。

Answer 1

您可以使用重载使方法返回适当的类型：

public getProcessor(audioProcessor: AudioProcessors.BiquadFilter): BiquadFilterNode 
public getProcessor(audioProcessor: AudioProcessors.DynamicsCompressor): DynamicsCompressorNode
public getProcessor(audioProcessor: AudioProcessors.StereoPanner): StereoPannerNode 
public getProcessor(audioProcessor: AudioProcessors): AudioProcessorNode {        
    throw new Error(`Method not implemented. ${audioProcessor}`);
}

或者您可以使用从枚举成员到适当返回类型的映射接口：

interface AudioProcessorNodeMap {
    [AudioProcessors.BiquadFilter] : BiquadFilterNode;
    [AudioProcessors.DynamicsCompressor] : DynamicsCompressorNode;
    [AudioProcessors.StereoPanner] : StereoPannerNode;
    [AudioProcessors.None] : void;
}

class x{

    public getProcessor<T extends AudioProcessors>(audioProcessor: T): AudioProcessorNodeMap[T]
    public getProcessor(audioProcessor: AudioProcessors): AudioProcessorNode {        
        throw new Error(`Method not implemented. ${audioProcessor}`);
    }
}

此版本的优点是如果您忘记在接口中添加成员，您将收到方法错误。