復雜的東西寫多了,如今寫點簡單的好了。由於功能上的需要,Vczh Library++3.0被我搞得很離譜。為了開發維護的遍歷、減少粗心犯下的錯誤以及增強單元測試、回歸測試和測試工具,因此記錄下一些開發上的小技巧,以便拋磚引玉,造福他人。歡迎高手來噴,菜鳥膜拜。
今天是關於內存的最後一篇了。上一篇文章講了為什麼不能對一個東西隨便memset。裡面的demo代碼出了點小bug,不過我不喜歡在發文章的時候裡面的demo代碼也拿去編譯和運行,所以大家有什麼發現的問題就評論吧。這樣也便於後來的人不會受到誤導。這次說的仍然是構造函數和析構函數的事情,不過我們將通過親手開發一個智能指針的方法,知道引用計數如何幫助管理資源,以及錯誤使用引用計數的情況。
首先先來看一下智能指針是如何幫助我們管理內存的。現在智能指針的實現非常多,我就假設這個類型叫Ptr<T>吧。這跟Vczh Library++ 3.0所使用的實現一樣。
1 class Base
2 {
3 public:
4 virtual ~Base(){}
5 };
6
7 class Derived1 : public Base
8 {
9 };
10
11 class Derived2 : public Base
12 {
13 };
14
15 //---------------------------------------
16
17 List<Ptr<Base>> objects;
18 objects.Add(new Derived1);
19 objects.Add(new Derived2);
20
21 List<Ptr<Base>> objects2;
22 objects2.Add(objects[0]);
當然這裡的List也是Vczh Library++3.0實現的,不過這玩意兒跟vector也好跟C#的List也好都是一個概念,因此也就不需要多加解釋了。我們可以看到智能指針的一個好處,只要沒有循環引用出現,你無論怎麼復制它,最終總是可以被析構掉的。另一個例子告訴我們智能指針如何處理類型轉換:
1 Ptr<Derived1> d1=new Derived1;
2 Ptr<Base> b=d1;
3 Ptr<Derived2> d2=b.Cast<Derived2>();
4 // d2是空,因為b指向的是Derived1而不是Derived2。
這就如同我們Derived1*可以隱式轉換到Base*,而當你使用dynamic_cast<Derived2*>(static_cast<Base*>(new Derived1))會得到0一樣。智能指針在幫助我們析構對象的同時,也要做好類型轉換的工作。
好了,現在先讓我們一步一步做出那個Ptr<T>。我們需要清楚這個智能指針所要實現的功能是什麼,然後我們一個一個來做。首先讓我們列出一張表:
1、沒有參數構造的時候,初始化為空
2、使用指針構造的時候,擁有那個指針,並且在沒有任何智能指針指向那個指針的時候刪除掉該指針。
3、智能指針進行復制的時候,兩個智能指針共同擁有該內部指針。
4、智能指針可以使用新的智能指針或裸指針重新賦值。
5、需要支持隱式指針類型轉換,static_cast不支持而dynamic_cast支持的轉換則使用Cast<T2>()成員函數來解決。
6、如果一個裸指針直接用來創建兩個智能指針的話,期望的情況是當兩個智能指針析構掉的時候,該指針會被delete兩次從而崩潰。
7、不處理循環引用。
最後兩點實際上是錯誤使用智能指針的最常見的兩種情況。我們從1到5一個一個實現。首先是1。智能指針可以隱式轉換成bool,可以通過operator->()拿到內部的T*。在沒有使用參數構造的時候,需要轉換成false,以及拿到0:
1 template<typename T>
2 class Ptr
3 {
4 private:
5 T* pointer;
6 int* counter;
7
8 void Increase()
9 {
10 if(counter)++*counter;
11 }
12
13 void Decrease()
14 {
15 if(counter && --*counter==0)
16 {
17 delete counter;
18 delete pointer;
19 counter=0;
20 pointer=0;
21 }
22 }
23
24 public:
25 Ptr():pointer(0),counter(0)
26 {
27 }
28
29 ~Ptr()
30 {
31 Decrease();
32 }
33
34 operator bool()const
35 {
36 return counter!=0;
37 }
38
39 T* operator->()const
40 {
41 return pointer;
42 }
43 };
在這裡我們實現了構造函數和析構函數。構造函數把內部指針和引用計數的指針都初始化為空,而析構函數則進行引用計數的減一操作。另外兩個操作符重載很容易理解。我們主要來看看Increase函數和Decrease函數都分別做了什麼。Increase函數在引用計數存在的情況下,把引用計數加一。而Decrease函數在引用計數存在的情況下,把引用計數減一,如果引用計數在減一過程中變成了0,則刪掉擁有的資源。
當然到了這個時候智能指針還不能用,我們必須替他加上復制構造函數,operator=操作符重載以及使用指針賦值的情況。首先讓我們來看使用指針賦值的話我們應該加上什麼:
1 Ptr(T* p):pointer(0),counter(0)
2 {
3 *this=p;
4 }
5
6 Ptr<T>& operator=(T* p)
7 {
8 Decrease();
9 if(p)
10 {
11 pointer=p;
12 counter=new int(1);
13 }
14 else
15 {
16 pointer=0;
17 counter=0;
18 }
19 return *this;
20 }
這裡還是偷工減料了的,構造函數接受了指針的話,還是轉給operator=去調用了。當一個智能指針被一個新指針賦值的時候,我們首先要減掉一個引用計數,因為原來的指針再也不被這個智能指針共享了。之後就進行判斷,如果來的是0,那麼就變成空。如果不是0,就擁有該指針,引用計數初始化成1。於是我們就可以這麼使用了:
1 Ptr<Base> b=new Derived1;
2 Ptr<Derived2> d2=new Derived2;
讓我們開始復制他們吧。復制的要領是,先把之前擁有的指針脫離掉,然後連接到一個新的智能指針上面去。我們知道非空智能指針有多少個,總的引用計數的和就是多少,只是分配到各個指針上面的數字不一樣而已:
1 Ptr(const Ptr<T>& p):pointer(p.pointer),counter(p.counter)
2 {
3 Increase();
4 }
5
6 Ptr<T>& operator=(const Ptr<T>& p)
7 {
8 if(this!=&p)
9 {
10 Decrease();
11 pointer=p.pointer;
12 counter=p.counter;
13 Increase();
14 }
15 return *this;
16 }
在上一篇文章有朋友指出重載operator=的時候需要考慮是不是自己賦值給自己,其實這是很正確的。我們寫每一類的時候,特別是當類擁有自己控制的資源的時候,需要非常注意這件事情。當然如果只是復制幾個對象而不會new啊delete還是close什麼handle,那檢查不檢查也無所謂了。在這裡我們非常清楚,當增加一個新的非空智能指針的時候,引用計數的總和會加一。當修改一個非空智能指針的結果也是非空的時候,引用計數的和保持不變。當然這是應該的,因為我們需要在所有非空智能指針都被毀掉的時候,釋放受保護的所有資源。
到了這裡一個智能指針基本上已經能用了,但是還不能處理父類子類的情況。這個是比較麻煩的,一個Ptr<Derived>事實上沒有權限訪問Ptr<Base>的內部對象。因此我們需要通過友元類來解決這個問題。現在讓我們來添加兩個新的函數吧,從一個任意的Ptr<C>復制到Ptr<T>,然後保證只有當C*可以隱式轉換成T*的時候編譯能夠通過:
1 template<X> friend class Ptr;
2
3 template<typename C>
4 Ptr(const Ptr<C>& p):pointer(p.pointer)