获取函数的参数数量和返回值类型demo

获取函数的参数数量和返回值类型

有的时候，我们由于某一些特殊的需求，希望获取一个函数的返回值类型，或者是参数数量，可以借助模板来实现。假如我们有一个函数

int f(double, float) {
    return 1;
}

返回值类型

当我们需要获取返回值类型的时候，我们首先声明一个模板函数。

template <class R, class... Args>
R getRetValue(R(*)(Args...));

上面R(*)(Args...)定义了一个函数指针。

或者

template <class R, class... Args>
R getRetValue(R(Args...));

这个函数不需要被实现，因为我们只是想得到它的返回值类型，并不是真的调用这个函数，此时我们可以通过以下一句话获取函数类型

using ret_t = decltype(getRetValue(f));

decltype是根据进数返回值进行类型推断的，只需要拿到函数声明即可，并不在乎这个函数是不是真的被定义了。当函数中传入了参数f之后，模板会自动地进行类型推断，将返回值传递给R，将参数类型double, float打包传递给Args。此时ret_t 就是函数f的返回值类型int。

完整测试代码

#include <iostream>
#include <utility>
using namespace std;
int f(double, float)
{
    return 1;
}
double g(double, float)
{
    return 1.0;
}
template <class R, class... Args>
R getRetValue(R(Args...));

//利用函数重载判断类型
void test(int v)
{
    cout << "this is int" << endl;
}
void test(double v)
{
    cout << "this is double" << endl;
}

int main()
{
    using ret_t_f = decltype(getRetValue(f));
    ret_t_f testf;
    test(testf);
    using ret_t_g = decltype(getRetValue(g));
    ret_t_g testg;
    test(testg);
    return 0;
}

输出

this is int
this is double

参数个数

如果我们需要获取参数的个数，那么可以用如下代码来实现：

template <class R, class... Args>
constexpr size_t getNumArgs(R(*)(Args...)) {
	return sizeof...(Args);
}

跟上面的做法类似，使用sizeof…可以获取参数的个数。

当函数是类成员函数的时候，上面的两个函数需要对应修改为：

template <class R, class C, class... Args>
R getRetValue(R(C::*)(Args...));

template <class R, class C, class... Args>
constexpr size_t getNumArgs(R(C::*)(Args...)) {
	return sizeof...(Args);
}

其中(C::*)表示这是类C的一个成员函数指针，当我们传入一个类成员函数指针时候，R、C、Args都会被对应的推断出来。

完整测试例子

#include <iostream>
#include <utility>
using namespace std;
int f(double, float)
{
    return 1;
}
double g(double)
{
    return 1.0;
}
template <class R, class... Args>
constexpr size_t getNumArgs(R (*)(Args...))
{
    return sizeof...(Args);
}

template <class R, class C, class... Args>
constexpr size_t getNumArgs(R (C::*)(Args...))
{
    return sizeof...(Args);
}

class A {
public:
    static void f(int, int, int) {};
};

int main()
{
    cout << getNumArgs(f) << endl;
    cout << getNumArgs(g) << endl;
    cout << getNumArgs(A::f) << endl;
}