libhv/base/hscope.h

#ifndef HW_SCOPE_H_
#define HW_SCOPE_H_

#include <functional>

#include "hdef.h"

class ScopeCleanup {
 public:
    typedef std::function<void()> FT;

    template<typename Fn, typename... Args>
    ScopeCleanup(Fn&& fn, Args&&... args) {
        cleanup_ = std::bind(std::forward<Fn>(fn), std::forward<Args>(args)...);
    }

    ~ScopeCleanup() {
        cleanup_();
    }

 private:
    FT cleanup_;
};

template<typename T>
class ScopeFree {
 public:
    ScopeFree(T* p) : _p(p) {}
    ~ScopeFree()    {SAFE_FREE(_p);}
 private:
    T*  _p;
};

template<typename T>
class ScopeDelete {
 public:
    ScopeDelete(T* p) : _p(p) {}
    ~ScopeDelete()    {SAFE_DELETE(_p);}
 private:
    T*  _p;
};

template<typename T>
class ScopeDeleteArray {
 public:
    ScopeDeleteArray(T* p) : _p(p) {}
    ~ScopeDeleteArray()    {SAFE_DELETE_ARRAY(_p);}
 private:
    T*  _p;
};

template<typename T>
class ScopeRelease {
 public:
    ScopeRelease(T* p) : _p(p) {}
    ~ScopeRelease()    {SAFE_RELEASE(_p);}
 private:
    T*  _p;
};

template<typename T>
class ScopeLock {
 public:
    ScopeLock(T& mutex) : _mutex(mutex) {_mutex.lock();}
    ~ScopeLock()    {_mutex.unlock();}
 private:
    T& _mutex;
};

#endif  // HW_SCOPE_H_