首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

如何根据参数类型获取函数的返回类型?

根据参数类型获取函数的返回类型可以通过使用 TypeScript 中的条件类型来实现。条件类型是一种在类型系统中进行条件判断的方式,可以根据不同的条件返回不同的类型。

下面是一个示例代码,展示了如何根据参数类型获取函数的返回类型:

代码语言:txt
复制
type ReturnTypeIfPromise<T> = T extends Promise<infer R> ? R : T;
type ReturnTypeIfArray<T> = T extends Array<infer R> ? R : T;

function getReturnType<T>(func: (...args: any[]) => T): ReturnTypeIfPromise<ReturnTypeIfArray<T>> {
  return {} as any;
}

// 示例函数
function add(a: number, b: number): number {
  return a + b;
}

// 获取函数的返回类型
type ResultType = ReturnTypeIfPromise<ReturnTypeIfArray<typeof add>>;
console.log(ResultType); // 输出:number

在上述示例中,我们定义了两个条件类型 ReturnTypeIfPromiseReturnTypeIfArrayReturnTypeIfPromise 用于判断函数返回类型是否为 Promise,如果是,则返回 Promise 的泛型参数类型;否则返回原始类型。ReturnTypeIfArray 用于判断函数返回类型是否为数组,如果是,则返回数组元素的类型;否则返回原始类型。

然后,我们定义了 getReturnType 函数,它接受一个函数作为参数,并使用条件类型来获取函数的返回类型。在示例中,我们传入了 add 函数,并使用 getReturnType 函数获取了 add 函数的返回类型。

需要注意的是,上述示例中的 getReturnType 函数仅返回了一个空对象,实际应用中需要根据具体情况进行相应的处理。

此外,还需要注意的是,以上示例中的代码是使用 TypeScript 编写的,如果要在 JavaScript 中实现类似的功能,可以考虑使用其他方式,如使用 typeof 运算符获取函数的返回类型。

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

领券