C++ 引用计数类模板

引用计数类模板

template<typename T>
class CRSIRefCnt
{
public:
	CRSIRefCnt() :
	  m_pstObject(NULL)
	  {
	  }

	  CRSIRefCnt(T* pstObject) :
	  m_pstObject(pstObject)
	  {
		  if(m_pstObject)
		  {
			  m_pstObject->AddRef();
		  }
	  }

	  CRSIRefCnt(const CRSIRefCnt<T>& stOther) :
	  m_pstObject(stOther.m_pstObject)
	  {
		  if (m_pstObject)
			  m_pstObject->AddRef();
	  }

	  ~CRSIRefCnt()
	  {
		  if (m_pstObject)
			  m_pstObject->Release();
	  }

	  CRSIRefCnt<T>& operator=(const CRSIRefCnt<T>& stOther)
	  {
		  if (this != &stOther && *this != stOther)
		  {
			  if (m_pstObject)
				  m_pstObject->Release();

			  m_pstObject = stOther.m_pstObject;

			  if (m_pstObject)
				  m_pstObject->AddRef();
		  }

		  return *this;
	  }

	  bool operator==(const CRSIRefCnt<T>& stOther) const
	  {
		  return m_pstObject == stOther.m_pstObject;
	  }

	  bool operator!=(const CRSIRefCnt<T>& stOther) const
	  {
		  return m_pstObject != stOther.m_pstObject;
	  }

	  const T& operator*() const
	  {
		  return *m_pstObject;
	  }

	  T& operator*()
	  {
		  return *m_pstObject;
	  }

	  const T* operator->() const
	  {
		  return m_pstObject;
	  }

	  T* operator->()
	  {
		  return m_pstObject;
	  }

	  bool IsValid() const
	  {
		  return m_pstObject != NULL;
	  }

private:
	T* m_pstObject;
};