具有任意数量参数的模板函数

Template function with an arbitrary number of arguments

本文关键字：函数 参数 任意数      更新时间：2023-10-16

我想编写一个函数invoke_measure_time来测量作为参数给出的任何其他函数的执行时间。我部分解决了我的问题，我有以下一段代码：

int sum(int a, int b){
std::this_thread::sleep_for(std::chrono::milliseconds(5000));
return a + b;
}
template<typename T>
std::pair<std::chrono::milliseconds, T> invoke_measure_time(std::function<T()> f){
auto start = std::chrono::system_clock::now();
auto f_result = f();
auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now() - start);
return std::make_pair(duration, f_result);
}
int main(){
auto fun = std::bind(sum, 10, 20);
auto emma = invoke_measure_time<int>(fun);
std::cout << "time: " << emma.first.count() << ", result = " << emma.second << std::endl;
}

它工作正常，结果符合预期，但第一行main让我头疼。如何修改invoke_measure_time以接受任何函数，而不仅仅是没有参数的函数？我希望我的主要看起来像：

int main(){
auto emma = invoke_measure_time<int, int, int>(sum, 10, 20);
std::cout << "time: " << emma.first.count() << ", result = " << emma.second << std::endl;
}

或类似的东西。我怎样才能做到这一点？

如果您将 temaplte 类型更改为泛型类型，那么我们可以添加一个可变参数模板参数来获取函数的参数。

template<typename Function, typename... Args>
auto invoke_measure_time(Function f, Args&&... args) {
auto start = std::chrono::system_clock::now();
auto f_result = f(std::forward<Args>(args)...);
auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now() - start);
return std::make_pair(duration, f_result);
}

获取任何函数(如 type(，并将传递给它的参数转发给invoke_measure_time。 它还具有能够像

int main(){
auto emma = invoke_measure_time(sum, 10, 20);
std::cout << "time: " << emma.first.count() << ", result = " << emma.second << std::endl;
}

简单的解决方案是invoke_measure_time接受任何可以在没有参数的情况下调用的函数对象(为了简洁起见，我省略了计时内容(：

template <typename F>
auto invoke_and_measure(F f) -> decltype(f()) {
return f();
}

然后是任何功能，具有任意参数，例如您的sum：

int sum(int a,int b) { return a+b; }

可以包装到一个可以在没有参数的情况下调用的 lambda 中：

int main() {
auto x = invoke_and_measure([](){ return sum(1,2);});
std::cout << x;
}

现场示例

请注意，lambda 通常比bind更容易使用。有一些极端情况bind可以做，但不能做lambda，但实际上我从头顶上不知道;)

您可以执行以下操作：

template<typename F, typename ... Ts>
auto invoke_measure_time(F&& f, Ts&&... args)
{
auto start = std::chrono::system_clock::now();
auto f_result = std::invoke(std::forward<F>(f), std::forward<Ts>(args)...);
auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now() - start);
return std::make_pair(duration, f_result);
}

演示

顺便说一句，当f返回void是有问题的。

相关文章：