引用计数

// CMyString.h: interface for the CMyString class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_CMYSTRING_H__FECC10BF_AB85_41AB_B3F3_7A16FA65D4AF__INCLUDED_)
#define AFX_CMYSTRING_H__FECC10BF_AB85_41AB_B3F3_7A16FA65D4AF__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include "CRefCount.h"

class CMyString 
{
public:
CMyString();
CMyString(const char* pSzData);
CMyString(const CMyString& obj);
void SetData(const char* pSzData);
void Print() const;
~CMyString();
private:
CRefCount* m_cRef;
};

#endif // !defined(AFX_CMYSTRING_H__FECC10BF_AB85_41AB_B3F3_7A16FA65D4AF__INCLUDED_)

// CMyString.cpp: implementation of the CMyString class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "CMyString.h"

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CMyString::CMyString()
{
  m_cRef = NULL;
}

CMyString::~CMyString()
{
  if (m_cRef != NULL)
  {
    m_cRef->RefDec();
    m_cRef = NULL;
  }
}

CMyString::CMyString(const char *pSzData)
{
  m_cRef = CRefCount::CreatClass(pSzData);
}

CMyString::CMyString(const CMyString& obj)
{
  m_cRef = obj.m_cRef;

  if (m_cRef != NULL)
  {
    m_cRef->RefAdd();
  }
}

void CMyString::SetData(const char* pSzData)
{
  if (m_cRef == NULL)
  {
    m_cRef = CRefCount::CreatClass(pSzData);
  }
  else
  {
    m_cRef->SetData(pSzData);
  }
}

void CMyString::Print() const
{
  if (!m_cRef)
  {
    return;
  }
  m_cRef->Print();
}

// CRefCount.h: interface for the CRefCount class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_CREFCOUNT_H__DEED2C87_0696_4C7B_95BE_21CB60D47ABC__INCLUDED_)
#define AFX_CREFCOUNT_H__DEED2C87_0696_4C7B_95BE_21CB60D47ABC__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

class CMyString;

class CRefCount  
{
  friend class CMyString;
private:
  static CRefCount* CreatClass(const char* pSzData);
    CRefCount();
  CRefCount(const char* pSzData);
    ~CRefCount();
  void Init();
  int RefAdd();
  int RefDec();
  void SetData(const char* pSzData);
  void Print() const;
private:
  char* m_pSzData;
  int   m_iStrLen;
  int   m_iSpaceLen;
  int   m_iRefCount;
};

#endif // !defined(AFX_CREFCOUNT_H__DEED2C87_0696_4C7B_95BE_21CB60D47ABC__INCLUDED_)

// CRefCount.cpp: implementation of the CRefCount class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "CRefCount.h"
#include <string.h>
#include <stdlib.h>
#include <iostream.h>
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CRefCount::CRefCount()
{
  Init();
}

CRefCount::CRefCount(const char* pSzData)
{
  Init();
  SetData(pSzData);
}

CRefCount::~CRefCount()
{
  if (m_pSzData != NULL)
  {
    delete[] m_pSzData;
    m_pSzData = NULL;
  }
  m_iRefCount = 0;
  m_iSpaceLen = 0;
  m_iStrLen   = 0;
}

CRefCount* CRefCount::CreatClass(const char* pSzData)
{
  return new CRefCount(pSzData);
}

void CRefCount::Init()
{
  m_iSpaceLen = 0;
  m_iStrLen = 0;
  m_iRefCount = 1;
  m_pSzData = NULL;
}

int CRefCount::RefAdd()
{
  return ++m_iRefCount;
}

int CRefCount::RefDec()
{
  if (0 == --m_iRefCount)
  {
    delete this;
    // 这句有必要加上，因为对象释放了，也就不存在m_iRefCount
    return 0;
  }
  return m_iRefCount;
}


void CRefCount::SetData(const char* pSzData)
{
  if (m_pSzData != NULL)
  {
    delete[] m_pSzData;
    m_pSzData = NULL;
  }
  
  // 申请空间，失败则程序退出 
  int iStrLen = strlen(pSzData) + sizeof(char);
  m_pSzData = new char[iStrLen];
  
  if (!m_pSzData)
    exit(-1);
  
  memcpy(m_pSzData, pSzData, iStrLen);
  m_iStrLen = iStrLen - 1;
  m_iSpaceLen = iStrLen;
}

void CRefCount::Print() const
{
  if (!m_pSzData)
  {
    return;
  }
  cout << m_pSzData << endl;
}

还有一种写时引用计数，比如有3个对象共享数据。

其中一个修改不会影响其他两个的数据。