Как затипизировать входные параметры функции, при проверке которые могут помеяться?

Question

[Kolabrod]

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

fn(x: any, y: any, z: any):any{
    if (!Array.isArray(x)) x = [x];
    if (!Array.isArray(y)) y = [y];
    if (!Array.isArray(z)) z = [z];
// .....
}

Comments

[Василий Банников]

А где xs объявлено?

Answer 1

[Василий Банников]

например:

function fn(x: number | Array<number>): void {
    if(!Array.isArray(x))
        x = [x];
    console.log(x);
}

В вашем коде компилятор, скорее всего, ругается на то что xs нигде не объявлен

Comments

[Kolabrod]

окей,
а потом мне нужно будет
x.forEach
Property 'forEach' does not exist on type 'number | number[]'.   Property 'forEach' does not exist on type 'number'.

[Василий Банников]

Kolabrod, напиши (x as Array<...>)

[Kolabrod]

Василий Банников, эм, прошу прощения где написать ?
всё равно вписит ошибка

fn(x:  number | Array<number>): any{
    if (!Array.isArray(x)) x = [x] as Array<number>;
   x.forEach....
// .....
}

[Alexandroppolus]

менять входящее значение - плохая практика, лучше так

function fn(x: number | Array<number>): void {
    const arrX = Array.isArray(x) ? x : [x];
    console.log(arrX);
}

[WbICHA]

Kolabrod, x = [x]
Ты серьёзно перезаписываешь аргумент?

[Kolabrod]

WbICHA, да

[Василий Банников]

Alexandroppolus, так действительно лучше выглядит.
Я думал, что TS уже научился типы на столько глубоко отслеживать, что видит изменения в ветках IF-ов

[Alexandroppolus]

Василий Банников, ну вообще нормально отслеживает

у автора видимо очень старая версия TS

[Kolabrod]

Alexandroppolus,
а так почему не отслеживает?

function fn(x: number | Array<number>, y: number | Array<number>): void {
    if (!Array.isArray(x)) x = [x];
    if (!Array.isArray(y)) y = [y];
    x.forEach((xs)=>{
        y.forEach((ys)=>{
            
        })
    })

}

fn(1,2)

[Василий Банников]

Kolabrod, ну видимо внутри лямбд не отслеживает.
Лямбда может быть вызвана когда угодно.

[Alexandroppolus]

Василий Банников, именно. Лямбда может быть вызвана асинхронно, к тому времени переменная "y" может снова стать числом.

function fn(x: number | Array<number>, y: number | Array<number>): void {
    const arrX = Array.isArray(x) ? x : [x];
    const arrY = Array.isArray(y) ? y : [y];

    arrX.forEach((xs)=>{
        arrY.forEach((ys)=>{
            
        })
    })
}

fn(1,2)

вот так всё норм, arrY массив навсегда

[Kolabrod]

Alexandroppolus, так-то оно так, но больше интересно было не инициализировать переменную

[Alexandroppolus]

Kolabrod, за каждой переменной навсегда закреплен тип. В твоем коде, х - это всегда "number | Array". После проверки Array.isArray(x) TS может считать, что да, прямо сейчас там только массив, но только на заведомо синхронном отрезке кода, где очевидно, что никто не вклинится и не поменяет х (с некоторыми исключениями, например, тот же eval не учитывается, хотя после него в переменных может быть что угодно).