如何让返回多类型的TypeScript函数返回特定类型？

这两个问题本质上都是要让TypeScript根据输入参数的具体值，精准推断出函数的返回类型，而不是返回一个宽泛的联合类型。我来一步步帮你搞定：

问题一：让hello函数根据输入字符串返回对应类型 #

你原来的代码里，arg没有指定具体的字面量类型，所以TypeScript没办法把输入和输出类型关联起来。这里有两种常用的解决方案：

方案1：函数重载 #

直接给函数定义多个重载签名，明确告诉TS不同输入对应的返回类型：

interface Alpha {}
interface Beta {}
interface Gamma {}

// 重载签名：定义每种输入对应的返回类型
function hello(arg: 'Alpha'): Alpha;
function hello(arg: 'Beta'): Beta;
function hello(arg: 'Gamma'): Gamma;
// 实现签名：要兼容所有重载的输入输出
function hello(arg: 'Alpha' | 'Beta' | 'Gamma'): Alpha | Beta | Gamma {
  if (arg === 'Alpha') return {} as Alpha;
  if (arg === 'Beta') return {} as Beta;
  if (arg === 'Gamma') return {} as Gamma;
  throw new Error('meow');
}

const x = hello('Beta'); // 现在x的类型是Beta，完美识别！

方案2：泛型+类型映射 #

如果后续要扩展更多类型，这种方式更灵活，不用每次加新重载：

interface Alpha {}
interface Beta {}
interface Gamma {}

// 定义输入参数的允许值
type ArgOptions = 'Alpha' | 'Beta' | 'Gamma';
// 建立输入到返回类型的映射
type ReturnTypeMap = {
  Alpha: Alpha;
  Beta: Beta;
  Gamma: Gamma;
};

function hello<T extends ArgOptions>(arg: T): ReturnTypeMap[T] {
  if (arg === 'Alpha') return {} as ReturnTypeMap[T];
  if (arg === 'Beta') return {} as ReturnTypeMap[T];
  if (arg === 'Gamma') return {} as ReturnTypeMap[T];
  throw new Error('meow');
}

const x = hello('Beta'); // 同样能精准推断为Beta类型

问题二：根据查询参数返回非空/可空date的Recipe #

你原来的重载写法没生效，主要是因为RecipeQuery的定义太宽泛，TS无法区分date.null的具体值。咱们调整一下类型定义，再用重载或泛型实现：

第一步：明确类型定义 #

先把查询参数拆成两个字面量类型的联合，让TS能识别不同的输入情况：

interface Recipe { date: Date | null }
// 用联合类型明确两种查询参数的可能
type RecipeQuery = 
  | { date: { null: true } }
  | { date: { null: false } };
// 定义返回类型的映射逻辑
type QueryResult<Q extends RecipeQuery> = 
  Q['date']['null'] extends true 
    ? Recipe 
    : Omit<Recipe, 'date'> & { date: Date };

方案1：函数重载实现 #

// 重载签名对应两种输入的返回类型
function query(q: { date: { null: true } }): Recipe;
function query(q: { date: { null: false } }): Omit<Recipe, 'date'> & { date: Date };
// 实际实现逻辑
function query(q: RecipeQuery) {
  if (q.date.null === true) return { date: null } as Recipe;
  return { date: new Date() } as Omit<Recipe, 'date'> & { date: Date };
}

// 测试验证
const res1 = query({ date: { null: true } }); // res1.date 类型是 Date | null
const res2 = query({ date: { null: false } }); // res2.date 类型是 Date，符合预期！

方案2：泛型+条件类型实现 #

这种方式更简洁，不用写多个重载签名：

function query<Q extends RecipeQuery>(q: Q): QueryResult<Q> {
  if (q.date.null === true) return { date: null } as QueryResult<Q>;
  return { date: new Date() } as QueryResult<Q>;
}

const res1 = query({ date: { null: true } }); // date: Date | null
const res2 = query({ date: { null: false } }); // date: Date

这里用Omit<Recipe, 'date'> & { date: Date }替换了你原来的Merge，因为TS原生的Omit能明确移除原来的date字段，再替换成非空的Date类型，比自定义的Merge更直观可靠。

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