设计模式简单代码之Flyweight模式

        在一雨田的Blog有一篇关于Flyweight模式的文章： http://blog.csdn.net/dylgsy/archive/2006/08/10/1045678.aspx

但在编译运行时，问题产生了：出版社和作者显示不了。

 

一雨田原来的代码：

1. // Flyweight.cpp
3. #pragma warning(disable: 4786)
5. #include <iostream>
6. #include <map>
7. #include <string>
8. #include <stdlib.h>
9. using namespace std;
12. class Book
13. {
14. public:
15.     string GetPublish() {return *m_publishCompany;}
16.     string GetWriter()  {return *m_writer;}
17.     int GetBookID()  {return m_bookID;}
18.     int GetPrice()  {return m_price;}
19.     string GetName()    {return m_name;}
21.     void SetPublish(string *s) {m_publishCompany = s;}
22.     void SetWriter(string *s) {m_writer = s;}
23.     void SetBookID(int id)  {m_bookID = id;}
24.     void SetPrice(int price) {m_price = price;}
25.     void SetName(string &s)  {m_name = s;}
27. private:
28.     string *m_publishCompany; // 出版社
29.     string *m_writer;   // 作者
30.     int m_bookID;    // 书籍编号
31.     int m_price;    // 价钱
32.     string m_name;    // 书名
33. };
35. class PublishFlyweight
36. {
37. public:
38.     PublishFlyweight(string s)
39.     {
40.         m_name = s;
41.     }
42.     string GetName()
43.     {
44.         return m_name;
45.     }
47. private:
48.     string m_name;
49. };
51. class PublishFlyweightFactory
52. {
53. public:
54.     PublishFlyweight* GetPublish(string key)
55.     {
56.         PublishFlyweight *p;
57.         map<string, PublishFlyweight*>::iterator it;
58.         it = mapPublish.find(key);
60.         // 存在这个出版社
61.         if(it != mapPublish.end() )
62.         {
63.             // 这里可能有点难懂，请查阅ＳＴＬ的帮助文档
64.             // 其实second就是指 map<string, PublishFlyweight*> 的 PublishFlyweight*
65.             p = (*it).second;
66.             cout << "已经有这个出版社: " << p->GetName() << " 你节省了" << strlen(p->GetName().c_str()) << "字节的空间" << endl;
67.         }
69.         else
70.         {// 插入这个PublishFlyweight
72.             p = new PublishFlyweight(key);
73.             mapPublish[key] = p;
74.         }
76.         return p;
77.     }
79. private:
80.     map<string, PublishFlyweight*> mapPublish;
81. };
83. class WriterFlyweight
84. {
85. public:
86.     WriterFlyweight(string s)
87.     {
88.         m_name = s;
89.     }
90.     string GetName()
91.     {
92.         return m_name;
93.     }
95. private:
96.     string m_name;
97. };
99. class WriterFlyweightFactory
100. {
101. public:
102.     WriterFlyweight* GetWriter(string key)
103.     {
104.         WriterFlyweight *p;
105.         map<string, WriterFlyweight*>::iterator it;
106.         it = mapWriter.find(key);
108.         // 存在这个Writer
109.         if(it != mapWriter.end() )
110.         {
111.             // 这里可能有点难懂，请查阅ＳＴＬ的帮助文档
112.             // 其实second就是指 map<string, WriterFlyweight*> 的 WriterFlyweight*
113.             p = (*it).second;
114.             cout << "已经有这个作者名字: " << p->GetName() << " 你节省了" << strlen(p->GetName().c_str()) << "字节的空间" << endl;
115.         }
117.         else
118.         {// 插入这个PublishFlyweight
120.             p = new WriterFlyweight(key);
121.             mapWriter[key] = p;
122.         }
124.         return p;
125.     }
127. private:
128.     map<string, WriterFlyweight*> mapWriter; 
129. };
132. void ShowBookInfo(Book book)
133. {
134.     cout << "书名：" << book.GetName() << endl;
135.     cout << "编号：" << book.GetBookID() << endl;
136.     cout << "价钱：" << book.GetPrice() << endl;
137.     cout << "出版：" << book.GetPublish() << endl;
138.     cout << "作者：" << book.GetWriter() << endl;
139.     cout << endl;
140. }
142. void main()
143. {
144.     PublishFlyweightFactory pff;
145.     WriterFlyweightFactory wff;
146.     Book book1, book2, book3;
148.     book1.SetPublish( &(pff.GetPublish("机械工业出版社")->GetName()) );
149.     book1.SetWriter( &(wff.GetWriter("候捷")->GetName()) );
150.     book1.SetBookID(0000);
151.     book1.SetPrice(20);
152.     book1.SetName(string("<<C++好野>>"));
154.     ShowBookInfo(book1);
156.     book2.SetPublish( &(pff.GetPublish("人民邮电出版社")->GetName()) );
157.     book2.SetWriter( &(wff.GetWriter("候捷")->GetName()) );
158.     book2.SetBookID(0001);
159.     book2.SetPrice(30);
160.     book2.SetName(string("<<C++是好劲>>"));
162.     ShowBookInfo(book2);
164.     book3.SetPublish( &(pff.GetPublish("机械工业出版社")->GetName()) );
165.     book3.SetWriter( &(wff.GetWriter("一雨田")->GetName()) );
166.     book3.SetBookID(0002);
167.     book3.SetPrice(50);
168.     book3.SetName(string("<<C++无得顶，我是铁头功...>>"));
170.     ShowBookInfo(book3);
172.     system("pause");
173. }

        是什么原因呢？仔细分析其源代码，发现“book1.SetPublish( &(pff.GetPublish("机械工业出版社")->GetName()) );”这一行使用了临时变量，在取“pff.GetPublish("机械工业出版社")->GetName() ”返回一个临时变量，而再取临时变量的地址&，运行到下一行时，这个临时地址将变成一个无效的地址，指向任何值都可以出现。因此，编程中是严禁这样使用的，编程时都应注意此类问题。修改好的代码如下：

1. /*
2.  来源：一雨田Blog
3.  问题：出版社和作者显示不了
4.  修改：qmroom
5.  */
8. // Flyweight.cpp
11. #include <iostream>
12. #include <map>
13. #include <string>
14. #include <stdlib.h>
15. using namespace std;
17. typedef char CHAR;
18. typedef CHAR *LPSTR;
19. typedef const CHAR *LPCSTR;
21. class Book
22. {
23. public:
24.     string GetPublish() {return *m_publishCompany;}
25.     string GetWriter()  {return *m_writer;}
26.     int GetBookID()  {return m_bookID;}
27.     int GetPrice()  {return m_price;}
28.     string GetName()    {return m_name;}
30.     void SetPublish(string *s) {m_publishCompany = s;}
31.     void SetWriter(string *s) {m_writer = s;}
32.     void SetBookID(int id)  {m_bookID = id;}
33.     void SetPrice(int price) {m_price = price;}
34.     void SetName(LPCSTR s)  {m_name = s;}
36. private:
37.     string *m_publishCompany; // 出版社
38.     string *m_writer;   // 作者
39.     int m_bookID;    // 书籍编号
40.     int m_price;    // 价钱
41.     string m_name;    // 书名
42. };
44. class PublishFlyweight
45. {
46. public:
47.     PublishFlyweight(string s)
48.     {
49.         m_name = new string(s);
50.     }
51.     virtual ~PublishFlyweight()
52.     {
53.         delete(m_name);
54.     }
55.     string *GetName()
56.     {
57.         return m_name;
58.     }
60. private:
61.     string *m_name;
62. };
64. class PublishFlyweightFactory
65. {
66. public:
67.     PublishFlyweight* GetPublish(string key)
68.     {
69.         PublishFlyweight *p;
70.         map<string, PublishFlyweight*>::iterator it;
71.         it = mapPublish.find(key);
73.         // 存在这个出版社
74.         if(it != mapPublish.end() )
75.         {
76.             // 这里可能有点难懂，请查阅ＳＴＬ的帮助文档
77.             // 其实second就是指 map<string, PublishFlyweight*> 的 PublishFlyweight*
78.             p = (*it).second;
79.             cout << "已经有这个出版社: " << *(p->GetName()) << " 你节省了" << strlen((*(p->GetName())).c_str()) << "字节的空间" << endl;
80.         }
81.         else
82.         {
83.             // 插入这个PublishFlyweight
84.             p = new PublishFlyweight(key);
85.             mapPublish[key] = p;
86.         }
88.         return p;
89.     }
91. private:
92.     map<string, PublishFlyweight*> mapPublish;
93. };
95. class WriterFlyweight
96. {
97. public:
98.     WriterFlyweight(string s)
99.     {
100.         m_name = new string(s);
101.     }
102.     virtual ~WriterFlyweight()
103.     {
104.         delete(m_name);
105.     }
106.     string *GetName()
107.     {
108.         return m_name;
109.     }
111. private:
112.     string *m_name;
113. };
115. class WriterFlyweightFactory
116. {
117. public:
118.     WriterFlyweight* GetWriter(string key)
119.     {
120.         WriterFlyweight *p;
121.         map<string, WriterFlyweight*>::iterator it;
122.         it = mapWriter.find(key);
124.         // 存在这个Writer
125.         if(it != mapWriter.end() )
126.         {
127.             // 这里可能有点难懂，请查阅ＳＴＬ的帮助文档
128.             // 其实second就是指 map<string, WriterFlyweight*> 的 WriterFlyweight*
129.             p = (*it).second;
130.             cout << "已经有这个作者名字: " << *(p->GetName()) << " 你节省了" << strlen((*(p->GetName())).c_str()) << "字节的空间" << endl;
131.         }
132.         else
133.         {
134.             // 插入这个PublishFlyweight
135.             p = new WriterFlyweight(key);
136.             mapWriter[key] = p;
137.         }
139.         return p;
140.     }
142. private:
143.     map<string, WriterFlyweight*> mapWriter; 
144. };
147. void ShowBookInfo(Book book)
148. {
149.     cout << "书名：" << book.GetName() << endl;
150.     cout << "编号：" << book.GetBookID() << endl;
151.     cout << "价钱：" << book.GetPrice() << endl;
152.     cout << "出版：" << book.GetPublish() << endl;
153.     cout << "作者：" << book.GetWriter() << endl;
154.     cout << endl;
155. }
157. void main()
158. {
159.     PublishFlyweightFactory pff;
160.     WriterFlyweightFactory wff;
161.     Book book1, book2, book3;
163.     book1.SetPublish( pff.GetPublish("机械工业出版社")->GetName() );
164.     book1.SetWriter( wff.GetWriter("qmroom")->GetName() );
165.     book1.SetBookID(0000);
166.     book1.SetPrice(20);
167.     book1.SetName(("<<C++好野>>"));
169.     ShowBookInfo(book1);
171.     book2.SetPublish( pff.GetPublish("人民邮电出版社")->GetName() );
172.     book2.SetWriter( wff.GetWriter("qmroom")->GetName() );
173.     book2.SetBookID(0001);
174.     book2.SetPrice(30);
175.     book2.SetName(("<<C++是好劲>>"));
177.     ShowBookInfo(book2);
179.     book3.SetPublish( pff.GetPublish("机械工业出版社")->GetName() );
180.     book3.SetWriter( wff.GetWriter("一雨田")->GetName() );
181.     book3.SetBookID(0002);
182.     book3.SetPrice(50);
183.     book3.SetName(("<<C++无得顶，我是铁头功...>>"));
185.     ShowBookInfo(book3);
187.     system("pause");
188. }

        其实，这段代码还有很多可以修改的地方，以便提高性能和便于维护和使用。请看设计模式简单代码之Flyweight模式（二）。

 

下一篇