c ++：使用模板将任何lambda包装在另一个lambda中

Question

我想创建一个可以包装任何 lambda 以记录开始/结束调用的函数。

下面的代码可以工作，除了：

1. 任何具有捕获功能的 lambda

2. 任何返回 void 的 lambda（尽管这可以通过编写第二个函数轻松解决）

#include <iostream>
#include <functional>

template <class T, class... Inputs>
auto logLambda(T lambda) {
    return [&lambda](Inputs...inputs) {
        std::cout << "STARTING " << std::endl;
        auto result = lambda(inputs...);
        std::cout << "END " << std::endl;
        return result;
    };
}

int main() {
    int a = 1;
    int b = 2;

    // works
    auto simple = []() -> int {
        std::cout << "Hello" << std::endl; return 1;
    };
    logLambda(simple)();

    // works so long as explicit type is declared
    auto with_args = [](int a, int b) -> int {
        std::cout << "A: " << a << " B: " << b << std::endl;
        return 1;
    };
    logLambda<int(int, int), int, int>(with_args)(a, b);

    // Does not work
    // error: no matching function for call to ‘logLambda<int(int), int>(main()::<lambda(int)>&)’
    auto with_captures = [&a](int b) -> int {
        std::cout << "A: " << a << " B: " << b << std::endl;
        return 1;
    };
    logLambda<int(int), int>(with_captures)(b);

}

有没有办法做到这一点？宏也可以接受

Answer 1

使用 Raii 处理 void 和非 void 返回类型，
并按值捕获函子以避免悬空引用，
并使用通用 lambda 来避免必须为自己指定参数

结果如下：

template <class F>
auto logLambda(F f) {
    return [f](auto... args) -> decltype(f(args...)) {
        struct RAII {
            RAII()  { std::cout << "STARTING " << std::endl; }
            ~RAII() { std::cout << "END " << std::endl; }
        } raii;

        return f(args...);
    };
}

调用方式如下：

const char* hello = "Hello";
logLambda([=](const char* s){ std::cout << hello << " " << s << std::endl; })("world");

Demo

Answer 2

该代码具有未定义的行为。

auto logLambda(T lambda) {
    return [&lambda]

您正在通过引用捕获本地参数。