文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

什么?函数类型重载还可以动态生成?

2024-12-02 03:43

关注

重载是指一个函数可以有不同的参数和返回值,也就是有不同的函数签名。

ts 支持函数重载,可以为同一个函数定义多个不同的类型:

重载的写法一共有三种(估计大多数人只会一种):

declare function func(name: string): string;
declare function func(name: number): number;

这种大家比较常用,声明两个同名函数,就能达到重载的目的:

函数可以用 interface 的方式声明,同样,也可以用 interface 的方式声明函数重载:

函数类型可以取交叉类型,也就是多种类型都可以,其实也是函数重载的意思:

重载虽然是很有用的特性,但有的时候重载多了写起来还是挺麻烦的。

比如 ts 提供的 lib.dom.ts 里就有这样的类型定义:

因为每种参数对应不同的返回值,所以它就重载了这么多。

这样写起来也太麻烦了吧,能不能用类型编程动态生成呢?

考虑下重载的三种写法,declare 和 interface 不行,但是 & 可以呀,能不能我传入一个联合类型,然后它给返回交叉类型呢?

比如这样:

都提示出来了肯定是可以的,我们看下砸实现的吧:

联合转交叉函数参数有逆变的性质,也就是类型缩小,比如参数能同时传 A、B、C 的话,参数类型怎么定义?

肯定是 A、B、C 的交集呀,也就你 A & B & C 的交叉类型,这样就可以接收 A、B、C 的参数了。

可以利用这个性质实现联合转交叉。

比如这样:

都提示出来了肯定是可以的,我们看下砸实现的吧:

联合转交叉

函数参数有逆变的性质,也就是类型缩小,比如参数能同时传 A、B、C 的话,参数类型怎么定义?

肯定是 A、B、C 的交集呀,也就你 A & B & C 的交叉类型,这样就可以接收 A、B、C 的参数了。

可以利用这个性质实现联合转交叉。

type UnionToIntersection<U> = 
(U extends U ? (x: U) => unknown : never) extends (x: infer R) => unknown
? R
: never

测试一下:

这里的类型参数 U 是传入的联合类型,加一个 U extends U 是为了触发分布式条件类型的特性。

什么是分布式条件类型呢?

当类型参数为联合类型,并且在条件类型左边直接引用该类型参数的时候,TypeScript 会把每一个元素单独传入来做类型运算,最后再合并成联合类型,这种语法叫做分布式条件类型。

比如这样一个联合类型:

type Union = 'a' | 'b' | 'c';

我们想把其中的 a 大写,就可以这样写:

type UppercaseA<Item extends string> = 
Item extends 'a' ? Uppercase<Item> : Item;

回到联合转交叉的这个高级类型:

加一个 U extends U 或者 U extends any 就能触发分布式条件类型的特性,让联合类型分成每一个类型单独传入做计算,最后把结果合并成联合类型。

然后再把它放到函数参数的位置,构造一个函数类型,通过模式匹配的方式提取参数的类型到 infer 声明的局部变量 R 里返回。

这样的结果就是交叉类型。

原因上面说过了,函数参数有逆变的性质,传入联合类型会返回交叉类型。

实现了联合转交叉之后,函数重载也就可以写出来了:

比如三个重载的返回值分别是 Aaa、Bbb、Ccc:

我们想基于这个生成重载的类型定义,传入联合类型返回重载的函数:

就可以这样写:

type UnionToOverloadFunction<T extends keyof ReturnValueMap> = 
UnionToIntersection<
T extends any ? (type: T) => ReturnValueMap[T] : never
>;

类型参数 T 是 ReturnValueMap 里的 key,约束为 keyof ReturnValueMap。

通过 T extends any 触发联合类型在分布式条件类型中的分发特性,让 'aaa' 'bbb' 'ccc' 分别传入做计算,返回构造出的函数类型的联合。

我们先单独测试下这部分:

可以看到返回的是构造出的函数类型的联合类型。

然后就用上面的 UnionToIntersection 转交叉就可以了:

这样就实现了重载函数的动态生成:

对比下最开始那种写法:

是不是清爽多了!而且还可以写一些动态逻辑。

总结

ts 函数重载一共有三种写法:declare function、interface、交叉类型 &。

当重载比较多的时候,直接列出来还是比较麻烦的,这时候可以用类型编程来动态生成函数重载。

我们实现了联合转交叉,利用了函数参数的逆变性质,也就是当参数可能是多个类型时,会返回它们的交叉类型。

然后又利用分布式条件类型的性质,当传入联合类型时,会把类型单独传入做计算,最后把结果合并成联合类型。

利用这个实现了传入联合类型返回构造出的函数的联合类型,然后再结合联合转交叉就实现了函数重载的动态生成。

当你写重载写的太多的时候,不妨试一下用类型编程的方式动态生成吧!

来源:神光的编程秘籍内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯