智能指针 安全引用的实现

最近迷上了SC2. 好久没更新博客.生活一成不变,就是一个不停赚钱-花钱的过程.

 

 

偶然代码里面会出现这样的情况:

A实例引用了B实例, B实例引用了A实例. 虽然很讨厌这样的代码,但是有时候似乎不可避免. 之所以讨厌,是因为A死了要通知B (防止野指针), B死了也要通知A (也是防止野指针..).. 很麻烦.

 

所以期望这样的代码, B死了不用通知A, 或者A不通知B, 代码如下:

 

/**** 安全指针引用 2011-06-30 Add By AYA TSafeRef,SafePtr 要配合着用 ****/ #ifndef SAFEPTR_H #define SAFEPTR_H #include <list> #include "Base/BaseType.h" #include "Base/Assert.h" class SafeRef { public: SafeRef () { mIsRef = true; } bool mIsRef; }; class SafePtr { public: virtual ~SafePtr () { std::list <SafeRef*>::iterator itr; for (itr = mRefList.begin (); itr != mRefList.end (); ++itr) { (*itr)->mIsRef = false; } } _4B RefCount () { return mRefList.size (); } std::list <SafeRef*> mRefList; }; template<class T> class TSafeRef : public SafeRef { public: TSafeRef (T* ref) { SafeRef::mIsRef = true; mRef = ref; mRef->mRefList.push_back (this); } TSafeRef (const TSafeRef& ref) { if (ref.mIsRef) { SafeRef::mIsRef = true; mRef = ref.mRef; mRef->mRefList.push_back (this); } else { SafeRef::mIsRef = false; mRef = NULL; } } TSafeRef () { mRef = NULL; SafeRef::mIsRef = false; } ~TSafeRef () { if (SafeRef::mIsRef) { mRef->mRefList.remove (this); } } public: operator bool () { return SafeRef::mIsRef; } TSafeRef& operator = (const TSafeRef& ref) { /// 先卸载之前的 if (SafeRef::mIsRef) { SafeRef::mIsRef = false; mRef->mRefList.remove (this); } /// 新引用 if (ref.mIsRef) { SafeRef::mIsRef = true; mRef = ref.mRef; mRef->mRefList.push_back (this); } return *this; } T* operator -> () { _ASSERT (SafeRef::mIsRef, "非法指针引用!"); return mRef; } T* mRef; }; #endif

 

使用示例:

 

class OO : public SafePtr {

public:

   void Foo () {

    //Do Something..

   }

};

 

OO* pOO = new OO;

TSafeRef <OO> mOORef (pOO); //安全引用!

 

if (mOORef) {

    mOORef->Foo (); //调用引用指针的方法

}

delete pOO;

 

if (!mOORef) {

   //说明pOO已经死了!

}