C++实现Golang defer功能

C++实现Golang defer功能

引用自 zltoolkit
头文件onceToken.h

#ifndef UTIL_ONCETOKEN_H_
#define UTIL_ONCETOKEN_H_

#include 
#include 
using namespace std;

namespace toolkit {

class onceToken {
public:
    typedef function task;

    template
    onceToken(const FUNC &onConstructed, function onDestructed = nullptr) {
        onConstructed();
        _onDestructed = std::move(onDestructed);
    }

    onceToken(nullptr_t, function onDestructed = nullptr) {
        _onDestructed = std::move(onDestructed);
    }

    ~onceToken() {
        if (_onDestructed) {
            _onDestructed();
        }
    }

private:
    onceToken() = delete;
    onceToken(const onceToken &) = delete;
    onceToken(onceToken &&) = delete;
    onceToken &operator=(const onceToken &) = delete;
    onceToken &operator=(onceToken &&) = delete;

private:
    task _onDestructed;
};

} 
#endif 

适用使用场景:在函数作用域结束后析构对应对象，或避免因抛异常时没有对象。

 socket->setOnErr([weak_self, weak_session, id](const SockException &err) {
        // 在本函数作用域结束时移除会话对象
        // 目的是确保移除会话前执行其 onError 函数
        // 同时避免其 onError 函数抛异常时没有移除会话对象
        onceToken token(nullptr, [&]() {
            // 移除掉会话
            auto strong_self = weak_self.lock();
            if (!strong_self) {
                return;
            }
            assert(strong_self->_poller->isCurrentThread());
            //从共享map中移除本session对象
            lock_guard lck(*strong_self->_session_mutex);
            strong_self->_session_map->erase(id);
        });

        // 获取会话强应用
        if (auto strong_session = weak_session.lock()) {
            // 触发 onError 事件回调
            strong_session->onError(err);
        }
    });