如何用TypeScript编写支持数字/字符串的类型推导柯里化函数?
实现支持数字/字符串的柯里化add函数
可以通过泛型实现这个需求,利用泛型捕获第一个参数的具体类型,从而限制后续参数和返回值的类型与之一致。具体代码如下:
const add = <T extends number | string>(a: T) => (b: T): T => { return (a + b) as T; };
代码说明:
<T extends number | string>:定义泛型T,限制其只能是number或string类型,避免传入其他不支持+操作的类型。- 第一个参数
a: T:用泛型T标记第一个参数的类型,TypeScript会自动推断T的具体类型(比如传入数字时T就是number,传入字符串时T就是string)。 - 第二个函数的参数
b: T和返回值: T:强制第二个参数的类型与第一个参数一致,返回值类型也保持相同。 (a + b) as T:由于TypeScript对number + string的联合类型运算会默认推断为string,这里用类型断言确保返回值符合泛型T的类型(我们已经限制了T只能是number或string,且两个参数类型一致,所以这个断言是安全的)。
效果验证:
- 数字场景:
add(1)(2)返回3(类型为number),若尝试传入add(1)("2"),TypeScript会直接报错,提示参数类型不匹配。 - 字符串场景:
add("foo")("bar")返回"foobar"(类型为string),若传入add("foo")(3)也会触发类型错误。
内容的提问来源于stack exchange,提问作者Galadre
相关产品推荐
相关产品推荐

