typescript - 打字稿接受并返回受约束的泛型

Question

我想要一个接受字符串或字符串数​​组的函数。然后它将返回输入的相同数据类型。这是当前不起作用的代码示例。

export default <T extends string | string[]>(s: T): T =>
  Array.isArray(s)
    ? s.map(handleString)
    : handleString(s)

Answer 1

我会通过函数重载来做到这一点：

const handleString = (val: string) => val;

function example(s: string): string;
function example(s: string[]): string[];
function example(s: string | string[]): string | string[] {
    return Array.isArray(s)
      ? s.map(handleString)
      : handleString(s)
}

const a = example('a'); // a's type is string
const b = example(['b']); // b's type is string[]

游乐场链接

关于重载的文档