根据 TypeScript 中的参数定义字典中函数的返回类型

Question

我有一个函数，它会根据函数参数返回不同的对象。 有没有办法在 Typescript 中实现这一点？

这是我已经走了多远：

type returnMap = {
    'a': number,
    'b': string[]
}

function func<T extends returnMap, K extends keyof T>(val: K): T[K] {
    if (val == 'a') {return 5}
    if (val == 'b') {return ['a', 'b']}
    return 42
}
console.log(func('a'))  // 5

但是，它在返回值处显示“TS2322：类型 '5' 不可分配给类型 'T[K]'”。

提前谢谢你

Answer 1

这样就可以了：

interface ReturnMap {
  'a': number;
  'b': string[];
}

function func<K extends keyof ReturnMap>(val: K): ReturnMap[K];
function func(val: keyof ReturnMap): ReturnMap[keyof ReturnMap] {
  if (val === 'a') return 5;
  if (val === 'b') return ['a', 'b'];
  throw new Error();
}

const r1 = func('a'); // const r1: number
const r2 = func('b'); // const r2: string[]

// Argument of type '"c"' is not assignable to parameter of type '"a" | "b"'.ts(2345)
const r3 = func('c'); // const r3: number | string[]

注意函数重载语法：

function func<K extends keyof ReturnMap>(val: K): ReturnMap[K];
function func(val: keyof ReturnMap): ReturnMap[keyof ReturnMap] { // ...

如果您只使用第二个签名，则返回的类型将始终为 number | string[]。

如果您只使用第一个签名，则 ReturnMap[K] 被评估（在函数体内）为逆变的 number & string[]，因此您会得到您在上面的问题中描述的错误。

或者，如果您不想使用重载签名 - 您可以将结果投射到函数体中：

function func<K extends keyof ReturnMap>(val: K): ReturnMap[K] {
  if (val === 'a') return 5 as ReturnMap[K];
  if (val === 'b') return ['a', 'b'] as ReturnMap[K];
  throw new Error();
}

Answer 2

我认为重载可能会对你有所帮助：

type returnMap = {
    'a': number,
    'b': string[],
    'x': 42,
    'z': 42,
}

function func<T extends returnMap, K extends Exclude<keyof returnMap,'a'|'b'>>(val: K): 42;
function func<T extends returnMap, K extends 'b'>(val: K): string[];
function func<T extends returnMap, K extends 'a'>(val: K): number;
function func<T extends returnMap, K extends keyof T>(val: K): number | string[] {
    if (val == 'a') {
        return 5
    }
    if (val == 'b') {
        return ['a', 'b']
    }
    return 42
}

const result = func('a') // number
const result1 = func('b') // string[]
const result2 = func('x') // 42
const result3 = func('z') // 42

Answer 3

这似乎是使用 any 作为返回变量类型的一个很好的用例。

见https://www.typescriptlang.org/docs/handbook/basic-types.html#any。