在ACE_Message_Block中使用Allocator内存池

laocai

在ACE_Message_Block的构造函数中可以传入相应的Allocator作为其里面数据的分配器，其中allocator_strategy专用于分配内部ACE_Data_Block里面的buffer，data_block_allocator用于分配内部的ACE_Data_Block，message_block_allocator用于内部构造新的ACE_Message_Block时使用。不过具体使用还得注意一些细节，如果这些细节没有注意很可能会用错，因此本人对在ACE_Message_Block中使用Allocator做了很浅的封装，使用起来就不容易出错了。附件代码中的Allocator是我用ACE写的Slab内存池，用起来效率还很不错。当然把模板参数实例化一下，也可以用其它的符合ACE的Allocator。
因上传附件的大小受限制，在这里贴代码，有兴趣的朋友可以跟我联系取工程文件和相关测试代码：
Message_Block_Oilfield.h:

1. #ifndef __MESSAGE_BLOCK_OILFIELD_H__
2. #define __MESSAGE_BLOCK_OILFIELD_H__
4. # if !defined (ACE_LACKS_PRAGMA_ONCE)
5. #  pragma once
6. # endif
8. #include "Slab_Malloc.h"
10. #include <ace/Message_Block.h>
11. #include <ace/String_Base.h>
12. #include <ace/ace_wchar.h>
14. #include <string>
16. // to_str_ptr is idea of c_str_ptr that is from stlsoft
18. inline const char *to_str_ptr(const char *data, bool &result_want_del)
19. {
20.     result_want_del = false;
21.     return data;
22. }
24. inline const char *to_str_ptr(const wchar_t *data, bool &result_want_del)
25. {
26.     result_want_del = true;
27.     return (ACE_Wide_To_Ascii::convert(data));
28. }
30. template <typename T>
31. inline const char *to_str_ptr(std::basic_string<T> const &s, bool &result_want_del)
32. {
33.     return (to_str_ptr(s.c_str(), result_want_del));
34. }
36. template <typename T>
37. inline const char *to_str_ptr(ACE_String_Base<T> const &s, bool &result_want_del)
38. {
39.     return (to_str_ptr(s.c_str(), result_want_del));
40. }
42. /**
43. * 在这里打个比喻，ACE_Message_Block就像石油，而
44. * Message_Block_Oilfield就像是油田。我们只管从
45. * “油田”中获取“石油”使用，不用考虑怎么把
46. * “石油”归还（释放）给“油田”。ACE_Message_Block
47. * 使用完毕后还是要释放的，但它可以像正常new出来
48. * 的那样进行释放。
49. */
51. template <typename POOL_ALLOCATOR = Slab_Malloc_Allocator, typename POOL_ALLOCATOR_SINGLETON = Slab_Malloc_Allocator_Singleton>
52. class Message_Block_Oilfield
53. {
54. public:
55.     ACE_Message_Block *get_message_block(void);
56.     ACE_Message_Block *get_message_block(size_t len);
57.     ACE_Message_Block *get_message_block(const char *data, int len);
58.     ACE_Message_Block *get_message_block(const ACE_UINT8 *data, int len);
60.     // 这个模板多了两个无用的参数，是因为模板函数不能偏特化，不得已而为之。
61.     template <typename String_Type>
62.     ACE_Message_Block *get_message_block(String_Type data, int not_use1, long not_use2);
64. public:
65.     ACE_Data_Block *get_data_block(void);
66.     ACE_Data_Block *get_data_block(size_t size,
67.                                 ACE_Message_Block::ACE_Message_Type msg_type,
68.                                 const char *msg_data,
69.                                 ACE_Allocator *allocator_strategy = 0,
70.                                 ACE_Lock *locking_strategy = 0);
72. private:
73.     ACE_Message_Block *get_message_block(ACE_Data_Block *data_block,
74.                                         ACE_Message_Block::Message_Flags flags,
75.                                         ACE_Allocator *message_block_allocator);
77. private:
78.     static POOL_ALLOCATOR *allocator_;
79. };
81. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON> inline ACE_Data_Block *
82. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_data_block(void)
83. {
84.     ACE_Data_Block *db;
86.     ACE_NEW_MALLOC_RETURN(db,
87.                         static_cast<ACE_Data_Block *>(allocator_->malloc(sizeof(ACE_Data_Block))),
88.                         ACE_Data_Block(0,
89.                         ACE_Message_Block::MB_DATA,
90.                         0,
91.                         allocator_,
92.                         0,
93.                         0,
94.                         allocator_),
95.                         0);
97.     return db;
98. }
100. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON> inline ACE_Message_Block *
101. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_message_block(size_t len)
102. {
103.     ACE_Message_Block *mb;
105.     mb = this->get_message_block();
106.     mb->size(len);
108.     return mb;
109. }
111. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON> inline ACE_Message_Block *
112. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_message_block(const ACE_UINT8 *data, int len)
113. {
114.     return (get_message_block(reinterpret_cast<const char *>(data), len));
115. }
117. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON> inline ACE_Message_Block *
118. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_message_block(ACE_Data_Block *data_block,
119.                                      ACE_Message_Block::Message_Flags flags,
120.                                      ACE_Allocator *message_block_allocator)
121. {
122.     ACE_Message_Block *mb;
124.     ACE_NEW_MALLOC_RETURN(mb,
125.                         static_cast<ACE_Message_Block *>(allocator_->malloc(sizeof(ACE_Message_Block))),
126.                         ACE_Message_Block(data_block, flags, message_block_allocator),
127.                         0);
129.     return mb;
130. }
132. typedef ACE_Singleton<Message_Block_Oilfield<>, ACE_SYNCH_MUTEX> Message_Block_Oilfield_Singleton;
133. #define MESSAGE_BLOCK_OILFIELD Message_Block_Oilfield_Singleton::instance()
134. enum Parameter_Not_Use_Type
135. {
136.     PARAMETER_NOT_USE_VAL = 0
137. };
139. #if defined (ACE_TEMPLATES_REQUIRE_SOURCE)
140. # include "Message_Block_Oilfield.cpp"
141. #endif
143. #if defined (ACE_TEMPLATES_REQUIRE_PRAGMA)
144. # pragma implementation ("Message_Block_Oilfield.cpp")
145. #endif
147. #endif

复制代码

laocai

Message_Block_Oilfield.cpp:

1. #ifndef __MESSAGE_BLOCK_OILFIELD_CPP__
2. #define __MESSAGE_BLOCK_OILFIELD_CPP__
4. #include "Message_Block_Oilfield.h"
6. #if !defined (ACE_LACKS_PRAGMA_ONCE)
7. # pragma once
8. #endif /* ACE_LACKS_PRAGMA_ONCE */
10. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON>
11. POOL_ALLOCATOR *Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::allocator_ = POOL_ALLOCATOR_SINGLETON::instance();
13. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON> ACE_Message_Block *
14. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_message_block(void)
15. {
16.     ACE_Message_Block *mb;
17.     ACE_Data_Block *db;
19.     db = this->get_data_block();
20.     mb = this->get_message_block(db, 0, allocator_);
22.     return mb;
23. }
25. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON> ACE_Message_Block *
26. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_message_block(const char *data, int len)
27. {
28.     ACE_Message_Block *mb;
29.     ACE_Data_Block *db;
31.     db = this->get_data_block(len, ACE_Message_Block::MB_DATA, data);
32.     mb = this->get_message_block(db, 0, allocator_);
33.     mb->wr_ptr(len);
35.     return mb;
36. }
38. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON>
39. template<typename String_Type>
40. ACE_Message_Block *
41. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_message_block(String_Type data, int not_use1, long not_use2)
42. {
43.     ACE_Message_Block *mb;
44.     bool result_want_del = false;
45.     const char *result_data = to_str_ptr(data, result_want_del);
47.     mb = get_message_block(result_data, ACE_OS::strlen(result_data));
48.     if (result_want_del)
49.         delete []result_data;
51.     return mb;
52. }
54. template<typename POOL_ALLOCATOR, typename POOL_ALLOCATOR_SINGLETON> ACE_Data_Block *
55. Message_Block_Oilfield<POOL_ALLOCATOR, POOL_ALLOCATOR_SINGLETON>::get_data_block(size_t size,
56.                                    ACE_Message_Block::ACE_Message_Type msg_type,
57.                                    const char *msg_data,
58.                                    ACE_Allocator *allocator_strategy,
59.                                    ACE_Lock *locking_strategy)
60. {
61.     ACE_Data_Block *db;
63.     ACE_NEW_MALLOC_RETURN(db,
64.         static_cast<ACE_Data_Block *>(allocator_->malloc(sizeof(ACE_Data_Block))),
65.         ACE_Data_Block(size,
66.         msg_type,
67.         0,
68.         (allocator_strategy != 0) ? allocator_strategy : allocator_,
69.         locking_strategy,
70.         0,
71.         allocator_),
72.         0);
74.     ACE_OS::memcpy(db->base(), msg_data, db->size());
76.     return db;
77. }
79. #endif

复制代码