Как написать функцию TypeScript?

Question

[Ne7Le4Der]

let a: string[] | CustomClass[];

    test(a);

   function test(data: string[]) {
        //Some code
   }

Argument of type 'string[] | 'CustomClass[]' is not assignable to parameter of type 'string[]'.

Мне нужно чтобы при string[] выполнялась функция, а при CustomClass[] просто возвращался переданный аргумент. Пробовал сделать перегрузку, но не смог сделать правильно.

Comments

[WbICHA]

function test(data: string[]): void;
function test(data: CustomClass[]): CustomClass[];
function test(data: string[] | CustomClass[]): void | CustomClass[] {
    //Some code
}

[Ne7Le4Der]

WbICHA, Спасибо за ответ, но, видимо я не очень верно задал вопрос.

Как внутри перегруженной функции отделить логику string[] от CustomClass[]?

[Rerurk]

WbICHA, скажите пожалуйста, в ts есть полиморфизм? Если это так то очень прекрасно

[WbICHA]

Ne7Le4Der, можно, только нужно самому делать проверку на тип. Правда, если это пустой массив, то не определить никак.

[Aetae]

WbICHA, если это пустой массив, то это и не важно.) TS же сам знает тип.

[WbICHA]

Aetae, это важно, если от типа меняется возвращаемый тип.
К примеру:

function test(data: string[]): string;
function test(data: CustomClass[]): CustomClass[];
function test(data: string[] | CustomClass[]): string | CustomClass[] {
    //Some code
}

В одном случае вернётся строка, в другом массив. При вызове тс будет отображать всё верно, однако в самой функции определить тип пустого массива мы не сможем никак и что в таком случае возвращать? Строку или массив?
Перегрузки в тсе опасны, в такой функции можно очень сильно накосячить и тс это спокойно съест, поэтому к перегрузкам надо подходить очень аккуратно.

[Aetae]

WbICHA, ты прав.

[Дмитрий Беляев]

скажите пожалуйста, в ts есть полиморфизм? Если это так то очень прекрасно

Rerurk, а почему его там должно не быть?

[Дмитрий Беляев]

Aetae, WbICHA, по хорошему в таких случаях надо одну из перегрузок типизировать как не пустой массив, например просто строку можно получить из массива строк, только если он не пустой, а значит если напишем так:

function test(data: [string, ...string[]]): string;
function test(data: CustomClass[]): CustomClass[];
function test(data: string[] | CustomClass[]): string | CustomClass[] {
    //Some code
}

то пустой массив можно будет передать только как CustomClass[]

[Дмитрий Беляев]

WbICHA, а вообще, в перегрузку нельзя передать юнион из аргументов разных сигнатур

[WbICHA]

Дмитрий Беляев, не совсем понял, речь о том, что функция не сможет принять тип string[] | CustomClass[]?
Если да, то это конечно, тут как раз недавно вопрос на эту тему был.)
https://qna.habr.com/q/1058108

[Aetae]

Дмитрий Беляев, ну это ж не решение проблемы, если я хочу пулять туда пустой массив независимо от типа.)

Answer 1

[Дмитрий Беляев]

Что-то вроде такого можно сделать:

type TestResult<T extends string[] | CustomClass[]> = T extends CustomClass[] ? T : void;
function test<T extends string[] | CustomClass[]>(data: T): TestResult<T> {
    if (data[0] instanceof CustomClass) {
      return data as TestResult<T>;
    }

    return undefined as TestResult<T>;
}

https://www.typescriptlang.org/play?#code/CYUwxgNg...

Но без приведения типов внутри не обойтись. Ну и как писал WbICHA в комментах, такие типы не безопасны для пустого массива, так как в рантайме типов нет, а значит и нет возможности проверить, какого типа пришел пустой массив.

Comments

[WbICHA]

Хм, использовать дженерик вместо оверлоада, как-то даже не задумывался раньше, интересный вариант.)