c++引用计数实现

版权声明：本文为博主原创文章，未经博主允许不得转载。

[cpp] view plain copy print ?

1. #include <iostream>   
2. #include <stdio.h>   
3.   
4. using namespace std;  
5.   
6. class String {  
7. public:  
8.     String(const char *pdata);  
9.     String(const String &rhs);  
10.     String &operator = (const String &rhs);  
11.     ~String();  
12. private:  
13.     class StringValue {  
14.     public:  
15.         int refCount;  
16.         char *data;  
17.         StringValue(const char *pdata);  
18.         ~StringValue();  
19.     };  
20.   
21.     StringValue *value;//所有的引用对象均共享唯一一个value，value里面实际存储data和引用次数   
22. };  
23.   
24. String::StringValue::StringValue(const char *pdata):refCount(1) {  
25.     data = new char[strlen(pdata) + 1];  
26.     strcpy(data,pdata);  
27. }  
28.   
29. String::StringValue::~StringValue() {  
30.     delete [] data;  
31. }  
32.   
33. String::String(const char *pdata):value(new StringValue(pdata))  
34. {}  
35.   
36. String::String(const String &rhs) {//要对引用加1   
37.     value = rhs.value;  
38.     value->refCount++;//所有指向同一段data的对象的引用加1   
39. }  
40.   
41. String &String::operator =(const String &rhs) {  
42.     if (value == rhs.value)//注意，不是this == &rhs   
43.         return *this;  
44.     if(--value->refCount == 0)  
45.         delete value;  
46.   
47.     value = rhs.value;  
48.     ++value->refCount;  
49.   
50.     return *this;  
51. }  
52.   
53. String::~String() {  
54.     if (--value->refCount == 0)  
55.         delete value;  
56. }  
57.   
58. void test() {  
59.     String ss("ssss");  
60.     String s1 = ss;  
61.     String s2("dddd");  
62.     s2 = ss;  
63. }  
64.   
65. int main() {  
66.     test();  
67. }  

原文地址：http://blog.csdn.net/sunmenggmail/article/details/8172172