Блог

У меня есть функция, которая должна возвращать разные объекты в зависимости от параметра функции. Есть ли способ, как я могу добиться этого в 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]'».

Заранее благодарю

Это сделает:

 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 amp; 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();
}
 

Я думаю, что перегрузка может вам помочь:

 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
 

Это кажется хорошим вариантом использования для использования any в качестве типа возвращаемой переменной.

См. https://www.typescriptlang.org/docs/handbook/basic-types.html#any .