typescript - 我可以根据参数更改返回类型吗

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 概念，以及我如何使用它，但这不是我的用例。

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

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

我想知道 typescript 是否有可能在返回后立即知道 uselessVar1 是 string 还是 number，我怎样才能做到这一点。

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

如果您对原始用例更感兴趣...

/** 
 * 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 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`.

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

嗯，not impossible ，但这条路不适合胆小的人。也可能不是一个好主意。