如何根据参数值为TypeScript函数指定返回类型

针对你的需求，推荐两种可靠的实现方式，既能让TypeScript正确推导返回类型，又能在函数实现阶段检查返回值的正确性：

方案一：映射类型（推荐，适合多类型扩展） #

先定义参数与返回值的类型映射关系，再用泛型约束参数为映射的key，返回值直接取映射的对应类型。这种方式扩展性极强，新增类型只需修改映射表和函数分支：

// 定义类型映射：参数值 → 返回类型
type TypeMap = {
  STRING: string | undefined;
  INTEGER: number | undefined;
  // 新增类型直接在这里添加，比如 BOOLEAN: boolean | undefined
};

function getValue<ValueType extends keyof TypeMap>(expectedType: ValueType): TypeMap[ValueType] {
  switch (expectedType) {
    case 'STRING':
      return "Hello" as TypeMap[ValueType];
    case 'INTEGER':
      return 25 as TypeMap[ValueType];
    // 新增类型的case直接在这里添加
    default:
      return undefined as TypeMap[ValueType];
  }
}

为什么之前的泛型写法报错？ #

你之前的泛型定义了R参数，允许调用者指定R为返回类型的子类型（比如getValue<'STRING', 'hello'>('STRING')），但函数返回的是固定值，无法匹配任意子类型，因此TypeScript会报错。去掉多余的R参数，直接用映射类型推导返回值，就能避免这个问题。

方案二：函数重载+分支类型约束 #

如果更倾向于重载签名的可读性，可以在实现体的每个分支里显式约束返回值类型，强制TypeScript检查分支返回值的正确性：

// 对外的重载签名
function getValueWithOverload(expectedType: 'STRING'): string | undefined;
function getValueWithOverload(expectedType: 'INTEGER'): number | undefined;
// 新增类型的重载签名直接在这里添加

// 函数实现体
function getValueWithOverload(
  expectedType: 'STRING' | 'INTEGER'
): string | number | undefined {
  switch (expectedType) {
    case 'STRING': {
      // 显式约束返回值类型，错误值会直接报错
      const result: string | undefined = "Hello";
      return result;
    }
    case 'INTEGER': {
      const result: number | undefined = 25;
      // 如果这里返回"some string"，TypeScript会立刻报错
      return result;
    }
    default:
      return undefined;
  }
}

为什么之前的重载写法不检查错误返回值？ #

之前的重载实现体返回类型是string | number | undefined的联合类型，TypeScript默认允许在任意分支返回联合中的任意值。通过在分支里显式定义对应类型的变量，就能强制检查返回值是否符合当前分支的预期类型。

使用效果 #

两种方案都能实现：

• 调用时自动推导返回类型：getValue('STRING')的返回值类型是string | undefined，getValue('INTEGER')是number | undefined
• 实现阶段检查错误返回值：如果在INTEGER分支返回字符串，TypeScript会直接报错

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