在 TypeScript 中,函数重载是一种强大特性,它允许为同一函数提供多个不同的调用接口。此特性通过定义多个函数签名来实现,每个签名指定不同的参数类型或数量,尽管它们共享相同的函数名。
步骤一:定义多个函数签名
函数重载的第一步是定义多个函数签名。这些签名包括函数名称、参数类型和返回类型,但没有函数体。
function combine(input1: string, input2: string): string;
function combine(input1: number, input2: number): number;
在这个例子中,我们定义了两个重载签名,一个接受两个字符串参数并返回一个字符串,另一个接受两个数字参数并返回一个数字。
步骤二:实现函数
函数的实际实现必须包含所有重载签名的逻辑。TypeScript 会根据传入的参数类型匹配相应的重载签名。
function combine(
input1: string | number,
input2: string | number
): string | number {
if (typeof input1 === "string" && typeof input2 === "string") {
return input1 + input2;
} else if (typeof input1 === "number" && typeof input2 === "number") {
return input1 + input2;
}
throw new Error("Invalid arguments");
}
在这个实现中,我们首先检查参数类型,然后根据类型执行相应的操作。如果参数类型不匹配重载签名,则抛出错误。
使用场景
-
处理不同类型的输入:在需要处理多种输入类型(如字符串和数字)的情况下,函数重载可以使代码更加简洁和可读。例如,上面的例子中,
combine
函数可以同时处理字符串拼接和数字相加。 -
简化 API 设计:在设计 API 时,函数重载可以让同一个函数名称支持不同的参数组合,从而简化 API 的使用。例如,可以设计一个
getData
函数,根据不同的参数类型返回不同的数据格式。 -
提高代码可维护性:通过函数重载,可以将处理不同类型的逻辑集中在一个函数中,从而提高代码的可维护性和扩展性。
函数重载里的函数参数数量不一样怎么办?
在 TypeScript 中,函数重载允许定义多个同名函数,这些函数可以有不同的参数数量和参数类型。这意味着函数重载签名中的参数数量是可以不一样的。
比如说,
// 定义多个重载签名
function greet(name: string): string;
function greet(firstName: string, lastName: string): string;
// 实现函数
function greet(nameOrFirstName: string, lastName?: string): string {
if (lastName) {
return `Hello, ${nameOrFirstName} ${lastName}!`;
} else {
return `Hello, ${nameOrFirstName}!`;
}
}
// 调用函数
console.log(greet("Alice")); // 输出:Hello, Alice!
console.log(greet("Alice", "Smith")); // 输出:Hello, Alice Smith!
注意事项
- 函数的实现必须包含所有重载签名的逻辑。
- 实现函数时要特别注意参数类型检查,以确保类型安全。
- 重载签名应尽量避免过于复杂,以保持代码的可读性和简洁性。