服务器之家:专注于VPS、云服务器配置技术及软件下载分享
分类导航

PHP教程|ASP.NET教程|Java教程|ASP教程|编程技术|正则表达式|C/C++|IOS|C#|Swift|Android|VB|R语言|JavaScript|易语言|vb.net|

服务器之家 - 编程语言 - 编程技术 - 如何声明 Currying 函数的类型?

如何声明 Currying 函数的类型?

2024-03-05 20:01全栈修仙之路 编程技术

Challenge 在本次挑战中,您需要为 Currying 函数声明相应的类型,以帮助 TypeScript 编译器推断出正确的类型。 declare function Currying(fn: any): anyconst curried1 = Currying((a: string, b: number, c: boolean) = true)const curried2 = Currying((a: string, b: number, c

Challenge

在本次挑战中,您需要为 Currying 函数声明相应的类型,以帮助 TypeScript 编译器推断出正确的类型。

declare function Currying(fn: any): any

const curried1 = Currying((a: string, b: number, c: boolean) => true)
const curried2 = Currying((a: string, b: number, c: boolean, d: boolean, e: boolean, f: string, g: boolean) => true)
const curried3 = Currying(() => true)

type cases = [
  Expect<Equal<
    typeof curried1,
     (a: string) => (b: number) => (c: boolean) => true>>,
  Expect<Equal<
    typeof curried2,
     (a: string) => (b: number) => (c: boolean) => (d: boolean) => (e: boolean) => (f: string) => (g: boolean) => true
  >>,
  Expect<Equal<typeof curried3, () => true>>,
]

在上面的代码中,我们使用了两个工具类型 Expect 和 Equal,它们的实现代码如下:

type Expect<T extends true> = T
type Equal<X, Y> =
  (<T>() => T extends X ? 1 : 2) extends
  (<T>() => T extends Y ? 1 : 2) ? true : false

Solution

首先,我们来分析一下第一个测试用例:

如何声明 Currying 函数的类型?图片

由上图可知,我们需要获取参数列表的类型和函数的返回值类型。参数列表类型要包含参数的名称和参数的类型。那么如何获取函数类型的参数列表类型和返回值类型呢?这时我们可以使用 TypeScript 内置的 Parameters 和 ReturnType 工具类型。

type T0 = Parameters<() => true> // []
type T1 = Parameters<(a: string, b: number, c: boolean) => true>
// [a: string, b: number, c: boolean]

type T2 = ReturnType<() => void> // void
type T3 = ReturnType<(a: string, b: number, c: boolean) => true> // true

如何声明 Currying 函数的类型?图片

在以上代码中,Parameters 工具类型用于获取函数类型的参数列表类型,它返回的是元组类型。而 ReturnType 工具类型则用于获取函数类型的返回值类型。它们的实现代码如下所示:

type Parameters<T extends (...args: any) => any> = 
  T extends (...args: infer P) => any ? P : never;
  
type ReturnType<T extends (...args: any) => any> = 
  T extends (...args: any) => infer R ? R : any;

在以上代码中,使用了 TypeScript 条件类型和 infer 类型推断。了解完以上代码,我们就知道如何获取函数类型的参数列表类型和返回值类型了。

接下来,我们要实现的功能就是使用函数的参数类型和返回值类型生成新的函数类型。下面我们来定义一个新的 ToCurrying 工具类型,它包含两个类型变量 Args 和 Return,分别表示参数的类型和返回值类型:

type ToCurrying<Args extends unknown[], Return> = unknown

然后,我们来继续分析第一个测试用例:

const curried1 = Currying((a: string, b: number, c: boolean) => true)

type cases = [
  Expect<Equal< typeof curried1,
    (a: string) => (b: number) => (c: boolean) => true>>
]

如何声明 Currying 函数的类型?图片

参考以上的图片,我们可以总结出 ToCurrying 工具类型的处理流程:

如何声明 Currying 函数的类型?图片

根据上述的处理流程,我们可以利用 TypeScript 条件类型、infer 类型推断和递归类型来实现对应的功能:

type ToCurrying<Args extends unknown[], Return> = 
  Args extends [...infer Head, infer Tail] 
   ? ToCurrying<Head, (arg: Tail) => Return> 
   : Return
   
type C0 = ToCurrying<[a: string, b: number, c: boolean], true>
// (arg: string) => (arg: number) => (arg: boolean) => true

type C1 = ToCurrying<[a: string, b: number, c: boolean, d: boolean, e: boolean, 
  f: string, g: boolean], true>
// (arg: string) => (arg: number) => (arg: boolean) => (arg: boolean) 
//   => (arg: boolean) => (arg: string) => (arg: boolean) => true

有了 ToCurrying 工具类型之后,我们来更新前面声明的 Currying 函数:

declare function Currying<T extends Function>(fn: T):
    T extends (...args: infer Args) => infer Return ?
    ToCurrying<Args, Return>
    : never
    
const curried1 = Currying((a: string, b: number, c: boolean) => true)
const curried2 = Currying((a: string, b: number, c: boolean, d: boolean, e: boolean, f: string, g: boolean) => true)
const curried3 = Currying(() => true)

type cases = [
  Expect<Equal<
    typeof curried1,
     (a: string) => (b: number) => (c: boolean) => true>>,
  Expect<Equal<
    typeof curried2,
     (a: string) => (b: number) => (c: boolean) => (d: boolean) => (e: boolean) 
       => (f: string) => (g: boolean) => true
  >>,
  Expect<Equal<typeof curried3, () => true>>,
]

更新后的 Currying 函数,已经可以满足前两个测试用例。但还不能满足最后一个测试用例:

如何声明 Currying 函数的类型?图片

这是因为获取 () => true 函数类型的参数列表类型时,返回的是空元组类型,针对这种情形,我们需要进行对应的处理:

declare function Currying<T extends Function>(fn: T):
    T extends (...args: infer Args) => infer Return ?
    Args extends [] 
    ? () => Return
    : ToCurrying<Args, Return>
    : never

在以上代码中,当发现 Args 类型变量对应的类型是空元组类型的话,我们直接返回 () => Return 函数类型。之后,我们就通过了所有的测试用例。最后,我们来看一下完整的代码:

declare function Currying<T extends Function>(fn: T):
    T extends (...args: infer Args) => infer Return ?
    Args extends []
    ? () => Return
    : ToCurrying<Args, Return>
    : never

type ToCurrying<Args extends unknown[], Return> =
    Args extends [...infer Head, infer Tail]
    ? ToCurrying<Head, (arg: Tail) => Return>
    : Return

原文地址:https://mp.weixin.qq.com/s/YAJNlp4koymBTFd9w9X1rw

延伸 · 阅读

精彩推荐