我可以根据参数更改 return 类型吗

Question

我不确定如何表述我的问题，如果您觉得问题标题不正确，请耐心等待。

function uselessFunction(value: number): string | number {
    if (value > 5) {
        return "false"
    }
    return 2
}

const uselessVar1 = uselessFunction(1);
// type is `string | number`
// expected would be `number`

const uselessVar2 = uselessFunction(10);
// type is `string | number`
// expected would be `string`.

我知道 typeguard 概念，以及如何使用它，但这不是我的 use-case。

function isString(value: string | number): value is string {
    if (typeof value === "string") {
        return true
    }
    return false
}

if (isString(uselessVar1) {
    // uselessVar1 type is `string`.
}

我想知道 typescript 是否有可能在 return 之后立即知道 uselessVar1 是 string 还是 number，以及如何知道我做到了。

此外，关于 how 对我来说很有趣，我也想知道这是否是一种不好的做法以及为什么。

如果您对原版更感兴趣 use-case...

/** 
 * Takes an input of type <T>[].
 * And returns bins of this list, of length binSize.
 * Example:
 *  
 * makeBinsFromArray([1, 2, 3, 4, 5], 3);
 * // [[1, 2, 3], [4, 5]]
 * 
 * makeBinsFromArray([1, 2, 3, 4, 5, 6, 7], 3);
 * // [[1, 2, 3], [4, 5, 6], [7]]
 * 
 * makeBinsFromArray([1, 2, 3, 4, 5, 6, 7], 3, true);
 * // [[1, 2, 3], [4, 5, 6], [7, undefined, undefined]]
*/
function makeBinsFromArray<T>(
  value: T[],
  binSize: number,
  fillLast?: boolean
): (T | (T | undefined))[][] {
  ...
}

const result = makeBinsFromArray([1, 2, 3], 2, true);
// expected type: (T | undefined)[][]

const result = makeBinsFromArray([1, 2, 3], 2);
// expected type: T[][];

使用这个函数，我想知道我在编译时是否有类型 T[][] 或 (T | undefined)[][]，这取决于我的 fillLast 参数。

Answer 1

function foo<Param extends boolean>(param?: Param): Param extends false | undefined ? T[][] : (T | undefined)[][]

可以使用泛型来“存储”参数是什么。然后在 return 类型注释中，我们使用条件来获取正确的 return 类型。

为简洁起见，我省略了正文和其他参数。

Answer 2

是的。

Function overloads 通常是您在这里想要的，您希望函数调用的每种方式都有不同的签名。

function makeBinsFromArray<T>(
  value: T[],
  binSize: number,
  fillLast: true
): (T | undefined)[][]

function makeBinsFromArray<T>(
  value: T[],
  binSize: number,
  fillLast?: false
): T[][]

function makeBinsFromArray<T>(
  value: T[],
  binSize: number,
  fillLast?: boolean
): (T | (T | undefined))[][] {
  /// implementation
}

const resultA = makeBinsFromArray([1, 2, 3], 2, true);
// (T | undefined)[][]

const resultB = makeBinsFromArray([1, 2, 3], 2);
// T[][];

Playground

但是，这是行不通的：

function uselessFunction(value: number): string | number {}
const uselessVar1 = uselessFunction(1); // `number`
const uselessVar2 = uselessFunction(10); // `string`.

您可以根据参数的类型更改 return，但这会根据值执行逻辑。并且数字类型不能在数学上与 Typescript 中的另一种数字类型进行比较，所以这行不通。 string | number 是您在这里得到的最好的。

好吧，not impossible，但这条路不适合虚张声势。也可能不是一个好主意。

我可以根据参数更改 return 类型吗

Can I change the return type based on a parameter

typescript

typescript-generics