c++常用工具之 SCOPE_GUARD

原创已于 2023-03-21 15:41:48 修改 · 582 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#c++

收录于

于 2023-03-20 20:16:51 首次发布

C/C++ 专栏收录该内容

37 篇文章

订阅专栏

该代码实现了一个C++的作用域退出类`scope_exit`，用于在作用域结束时自动执行回调函数，类似于智能指针在析构时自动释放内存。`SCOPE_GUARD`宏用于创建`scope_exit`实例，确保在函数返回时执行清理操作，例如释放动态分配的内存。

// defer.hpp
#pragma once

#include <functional>

namespace toolkit {
class scope_exit {
  public:
    scope_exit() = default;

    scope_exit(const scope_exit&)            = delete;
    scope_exit& operator=(const scope_exit&) = delete;

    scope_exit(scope_exit&&)            = default;
    scope_exit& operator=(scope_exit&&) = default;

    template <typename F, typename... Args>
    scope_exit(F&& f, Args&&... args)
    {
        func_ = std::bind(std::forward<F>(f), std::forward<Args>(args)...);
    }

    ~scope_exit()
    {
        if (func_) {
            func_();
        }
    }

  private:
    using Callable = std::function<void()>;
    Callable func_;
};
}  // namespace toolkit

#define _CONCAT(__x, __y) __x##__y
#define _MAKE_SCOPE(__x)  toolkit::scope_exit _CONCAT(defer, __x) = [&]()
#define SCOPE_GUARD       _MAKE_SCOPE(__LINE__)

使用：

// main.cpp
#include <iostream>
#include <string>
#include "defer.hpp"


int main(int argc, char** argv)
{
    void* foo = (void*)malloc(10);
    SCOPE_GUARD
    {
        std::cout << "free" << std::endl;
        free(foo);
    };

    void* foo1 = (void*)malloc(10);
    SCOPE_GUARD
    {
        std::cout << "free" << std::endl;
        free(foo1);
    };
    return 0;
}