如何基于枚举参数为TypeScript工厂函数指定返回类型？

你遇到的问题核心是TypeScript无法自动关联factoryMap[type]()的返回值与泛型条件类型FactoryReturnType<T>的精准匹配：

• 你定义的FactoryMap类型是Record<MType, () => FactoryReturnType<MType>>，这意味着所有工厂函数的返回类型都是Q | V（因为FactoryReturnType<MType>本质是联合类型）。
• 当调用factoryMap[type]()时，TypeScript只能推断出返回值是Q | V，但你的函数返回类型是条件类型FactoryReturnType<T>——TS无法证明Q | V中的每个分支都能严格对应T对应的具体返回类型，因此抛出类型不兼容错误。

举个例子：当T是MType.Q时，FactoryReturnType<T>是Q，但TS从FactoryMap的类型定义里，无法确定factoryMap[type]()一定返回Q（它可能返回V），所以就会报错。

方法1：用映射类型实现精准类型推断（推荐） #

通过定义枚举与返回类型的映射关系，让TypeScript能精准推断每个枚举参数对应的返回类型：

enum MType {
    Q = 'q',
    V = 'v',
}

interface Q {}
interface V {}

// 定义枚举到返回类型的映射表
type TypeMap = {
    [MType.Q]: Q;
    [MType.V]: V;
};

function factoryQ(): Q {
    return {};
}
function factoryV(): V {
    return {};
}

function factory<T extends MType>(type: T): TypeMap[T] {
    // 让工厂映射的类型与TypeMap严格对应
    const factoryMap: { [K in MType]: () => TypeMap[K] } = {
        [MType.Q]: factoryQ,
        [MType.V]: factoryV,
    };

    return factoryMap[type]();
}

const q = factory(MType.Q); // 类型为 Q
const v = factory(MType.V); // 类型为 V

这种方式的优势是类型严谨，后续扩展枚举时，只需要更新TypeMap和factoryMap，TS会自动校验类型一致性。

方法2：使用函数重载 #

直接通过重载签名明确每个枚举参数对应的返回类型：

enum MType {
    Q = 'q',
    V = 'v',
}

interface Q {}
interface V {}

function factoryQ(): Q {
    return {};
}
function factoryV(): V {
    return {};
}

// 重载签名：明确参数与返回值的对应关系
function factory(type: MType.Q): Q;
function factory(type: MType.V): V;
// 实现签名：处理联合类型逻辑
function factory(type: MType): Q | V {
    const factoryMap = {
        [MType.Q]: factoryQ,
        [MType.V]: factoryV,
    };
    return factoryMap[type]();
}

const q = factory(MType.Q); // 类型为 Q
const v = factory(MType.V); // 类型为 V

重载方式更直观，适合逻辑简单的场景，外部调用时会严格遵循重载签名的类型推断。

方法3：类型断言（快速解决但不严谨） #

如果只是临时绕过错误，可以用类型断言告诉TS返回值符合要求，但这种方式会失去TS的类型校验能力，后续修改代码容易出错：

function factory<T extends MType>(type: T): FactoryReturnType<T> {
    type FactoryMap = Record<MType, () => FactoryReturnType<MType>>

    const factoryMap: FactoryMap = {
        [MType.Q]: factoryQ,
        [MType.V]: factoryV,
    }

    return factoryMap[type]() as FactoryReturnType<T>;
}

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