泛化的ScopeGuard模板类

 

#pragma once

#include <cassert>
#include <memory>
#include "non_copy_able.h"



class ScopeGuardImp : private NonCopyable
...{
public:
        ScopeGuardImp() ...{ };
        virtual ~ScopeGuardImp()...{ }
};



template<class Type, class Fore, class Aft>                    
class ScopeGuardMF : public ScopeGuardImp
...{ 
private:
        typedef typename void (Type::*MFun)(void);
private:
        Type    *m_type;
        Fore    m_fore;
        Aft        m_aft;
public:
        ScopeGuardMF(Type *ptype, Fore fore, Aft aft)
        ...{
                assert(ptype != NULL);
                m_type = ptype;
                m_fore = fore;
                m_aft = aft;
                (m_type->*m_fore)();
        }

        ~ScopeGuardMF()
        ...{
                (m_type->*m_aft)();
        }
};


template<class Fore, class Aft>
class ScopeGuardFun : public ScopeGuardImp
...{ 
private:
        Fore    m_fore;
        Aft        m_aft;
public:
        ScopeGuardFun(Fore fore, Aft aft)
        ...{
                m_fore = fore;
                m_aft = aft;
                m_fore();
        }
        
        ~ScopeGuardFun()
        ...{
                m_aft();
        }

};


class ScopeGuard  : private NonCopyable
...{
private:
        std::auto_ptr<ScopeGuardImp>   m_pimp;
public:
        template<typename Type, typename Fore, typename Aft>
        ScopeGuard(Type *ptype, Fore fore, Aft aft)
        ...{
                m_pimp.reset(new ScopeGuardMF<Type, Fore, Aft>(ptype, fore, aft));
        }
        
        template<typename Fore, typename Aft>
        ScopeGuard(Fore fore, Aft aft)
        ...{
                m_pimp.reset(new ScopeGuardFun<Fore, Aft>(fore, aft));
        }

        ~ScopeGuard() ...{ }

};

 

测试代码:

 

class Test
...{
public:
        int fore() ...{ printf("fore ");  return 8;}
        void after() ...{ printf("after "); }
};

void fore() ...{ printf("aaaaa "); }
double after() ...{ printf("bbbbb "); return 33.6;}

int main()
...{
        ...{
                Test test;
                ScopeGuard guard1(&test, &Test::fore, &Test::after);
        }
        ...{
                ScopeGuard guard2(fore, after);
        }
        
        
        cin.get();

        return 0;
};