jmem_hashmap_compact.h 69 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421
  1. /**
  2. * Tencent is pleased to support the open source community by making Tars available.
  3. *
  4. * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
  5. *
  6. * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
  7. * in compliance with the License. You may obtain a copy of the License at
  8. *
  9. * https://opensource.org/licenses/BSD-3-Clause
  10. *
  11. * Unless required by applicable law or agreed to in writing, software distributed
  12. * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
  13. * CONDITIONS OF ANY KIND, either express or implied. See the License for the
  14. * specific language governing permissions and limitations under the License.
  15. */
  16. #ifndef _JMEM_HASHMAP_COMPACT_H
  17. #define _JMEM_HASHMAP_COMPACT_H
  18. #include "util/tc_hashmap_compact.h"
  19. #include "util/tc_autoptr.h"
  20. #include "jmem/jmem_policy.h"
  21. #include "tup/Tars.h"
  22. namespace tars
  23. {
  24. /************************************************************************
  25. 重要说明:
  26. 1 使用方式与TarsHashmap完全一致;
  27. 2 与JHashmap相比, 在64位os下面,每个chunk能够节约大概30个字节;
  28. 3 效率比JHashmap稍慢一点,可以忽略不计;
  29. 4 每个chunk的大小最大只能是64k(与实际数据块大小无关,实际数据块可以多个chunk拼接在一起)
  30. ************************************************************************/
  31. /************************************************************************
  32. 基本说明如下:
  33. 基于Tars协议的内存hashmap
  34. 编解码出错则抛出TarsDecodeException和TarsEncodeException
  35. 可以对锁策略和存储策略进行组合, 例如:
  36. 基于信号量锁, 文件存储的hashmap:
  37. TarsHashMap<Test::QueueElement, SemLockPolicy, FileStorePolicy>
  38. 基于信号量锁, 共享内存存储的hashmap
  39. TarsHashMap<Test::QueueElement, SemLockPolicy, ShmStorePolicy>
  40. 基于线程锁, 内存存储的hashmap
  41. TarsHashMap<Test::QueueElement, ThreadLockPolicy, MemStorePolicy>
  42. 使用上, 不同的组合, 初始化函数不完全一样
  43. 初始化函数有:
  44. SemLockPolicy::initLock(key_t)
  45. ShmStorePolicy::initStore(key_t, size_t)
  46. FileStorePolicy::initStore(const char *file, size_t)
  47. 等, 具体参见jmem_policy.h
  48. ***********************************************************************
  49. 基本特性说明:
  50. > 内存数据的map, 根据最后Get时间的顺序淘汰数据;
  51. > 支持缓写/dump到文件/在线备份;
  52. > 支持不同大小内存块的配置, 提供内存的使用率;
  53. > 支持回收到指定空闲比率的空间;
  54. > 支持仅设置Key的操作, 即数据无value, 只有Key, 类似与stl::set;
  55. > 支持自定义hash算法;
  56. > hash数可以根据内存块比率设置, 并优化有素数, 提高hash的散列性;
  57. > 支持几种方式的遍历, 通常遍历时需要对map加锁;
  58. > 对于hash方式的遍历, 遍历时可以不需要对map加锁, 推荐使用;
  59. > 支持自定义操作对象设置, 可以非常快速实现相关的接口;
  60. > 支持自动编解码, Key和Value的结构都通过tars2cpp生成;
  61. > tars协议支持自动扩展字段, 因此该hashmap支持自动扩展字段(Key和Value都必须是通过tars编码的);
  62. > map支持只读模式, 只读模式下set/erase/del等修改数据的操作不能使用, get/回写/在线备份正常使用
  63. > 支持自动淘汰, set时, 内存满则自动淘汰, 在非自动淘汰时, 内存满直接返回RT_READONLY
  64. > 对于mmap文件, 支持自动扩展文件, 即内存不够了, 可以自动扩展文件大小(注意hash的数量不变, 因此开始就需要考虑hash的数量), 而且不能跨JHashmap对象(即两个hashmap对象访问同一块文件,通知一个hashmap扩展以后,另外一个对象并不知道扩展了)
  65. ***********************************************************************
  66. hashmap链说明:
  67. hashmap链一共包括了如下几种链表:
  68. > Set时间链: 任何Set操作都会修改该链表, Set后数据被设置为脏数据, 且移动到Set链表头部;
  69. > Get时间链: 任何Get操作都会修改该链表, 除非链表只读, 注意Set链同时也会修改Get链
  70. > Dirty时间链: dirty链是Set链的一部分, 用于回写数据用
  71. > Backup链:备份链是Get链的一部分, 当备份数据时, 顺着Get链从尾部到头部开始备份;
  72. ***********************************************************************
  73. 相关操作说明:
  74. > 可以设置map只读, 则所有写操作返回RT_READONLY, 此时Get操作不修改链表
  75. > 可以设置知否自动淘汰, 默认是自动淘汰的.如果不自动淘汰,则set时,无内存空间返回:RT_NO_MEMORY
  76. > 可以更改hash的算法, 调用setHashFunctor即可
  77. > 可以将某条数据设置为干净, 此时移出到Dirty链表指Dirty尾部的下一个元素;
  78. > 可以将某条数据设置为脏, 此时会移动到Set链表头部;
  79. > 每个数据都有一个上次回写时间(SyncTime), 如果启动回写操作, 则在一定时间内会回写;
  80. > 可以dump到文件或者从文件中load, 这个时候会对map加锁
  81. > 可以调用erase批量淘汰数据直到内存空闲率到一定比例
  82. > 可以调用sync进行数据回写, 保证一定时间内没有回写的数据会回写, map回写时间通过setSyncTime设置, 默认10分钟
  83. > 可以setToDoFunctor设置操作类, 以下是操作触发的情况:
  84. ***********************************************************************
  85. ToDoFunctor的函数说明:
  86. > 通常继承ToDoFunctor, 实现相关函数就可以了, 可以实现以下功能:Get数据, 淘汰数据, 删除数据, 回写数据, 备份数据
  87. > ToDoFunctor::erase, 当调用map.erase时, 该函数会被调用
  88. > ToDoFunctor::del, 当调用map.del时, 该函数会被调用, 注意删除时数据可能都不在cache中;
  89. > ToDoFunctor::sync, 当调用map.sync时, 会触发每条需要回写的数据该函数都被调用一次, 在该函数中处理回写请求;
  90. > ToDoFunctor::backup, 当调用map.backup时, 会触发需要备份的数据该函数会被调用一次, 在该函数中处理备份请求;
  91. > ToDoFunctor::get, 当调用map.get时, 如果map中无数据, 则该函数被调用, 该函数从db中获取数据, 并返回RT_OK, 如果db无数据则返回RT_NO_DATA;
  92. > ToDoFunctor所有接口被调用时, 都不会对map加锁, 因此可以操作map
  93. ***********************************************************************
  94. map的重要函数说明:
  95. > set, 设置数据到map中, 会更新set链表
  96. 如果满了, 且可以自动淘汰, 则根据Get链淘汰数据, 此时ToDoFunctor的sync会被调用
  97. 如果满了, 且可以不能自动淘汰, 则返回RT_NO_MEMORY
  98. > get, 从map获取数据, 如果有数据, 则直接从map获取数据并返回RT_OK;
  99. 如果没有数据, 则调用ToDoFunctor::get函数, 此时get函数需要返回RT_OK, 同时会设置到map中, 并返回数据;
  100. 如果没有数据, 则ToDoFunctor::get函数也无数据, 则需要返回RT_NO_DATA, 此时只会把Key设置到map中, 并返回RT_ONLY_KEY;
  101. 在上面情况下, 如果再有get请求, 则不再调用ToDoFunctor::get, 直接返回RT_ONLY_KEY;
  102. > del, 删除数据, 无论cache是否有数据, ToDoFunctor::del都会被调用;
  103. 如果只有Key, 则该数据也会被删除;
  104. > erase, 淘汰数据, 只有cache存在数据, ToDoFunctor::erase才会被调用
  105. 如果只有Key, 则该数据也会被淘汰, 但是ToDoFunctor::erase不会被调用;
  106. > erase(int radio), 批量淘汰数据, 直到空闲块比率到达radio;
  107. ToDoFunctor::erase会被调用;
  108. 只有Key的记录也会被淘汰, 但是ToDoFunctor::erase不会被调用;
  109. > sync: 缓写数据, 超时没有回写且是脏数据需要回写, 回写完毕后, 数据会自动设置为干净数据;
  110. 可以多个线程或进程同时缓写;
  111. ToDoFunctor::sync会被调用;
  112. 只有Key的记录, ToDoFunctor::sync不会被调用;
  113. > backup: 备份数据, 顺着顺着Get链从尾部到头部开始备份;
  114. ToDoFunctor::backup会被调用;
  115. 只有Key的记录, ToDoFunctor::backup不会被调用;
  116. 由于备份游标只有一个, 因此多个进程同时备份的时候数据可能会每个进程有一部分
  117. 如果备份程序备份到一半down了, 则下次启动备份时会接着上次的备份进行, 除非将backup(true)调用备份
  118. ***********************************************************************
  119. 返回值说明:
  120. > 注意函数所有int的返回值, 如无特别说明, 请参见TC_HashMapCompact::RT_
  121. ***********************************************************************
  122. 遍历说明:
  123. > 可以用lock_iterator对map进行以下几种遍历, 在遍历过程中其实对map加锁处理了
  124. > end(): 迭代器尾部
  125. > begin(): 按照block区块遍历
  126. > rbegin():按照block区块逆序遍历
  127. > beginSetTime(): 按照Set时间顺序遍历
  128. > rbeginSetTime(): 按照Set时间顺序遍历
  129. > beginGetTime(): 按照Get时间顺序遍历
  130. > rbeginGetTime(): 按照Get时间逆序遍历
  131. > beginDirty(): 按时间逆序遍历脏数据链(如果setClean, 则也可能在脏链表上)
  132. > 其实回写数据链是脏数据量的子集
  133. > 注意:lock_iterator一旦获取, 就对map加锁了, 直到lock_iterator析够为止
  134. >
  135. > 可以用hash_iterator对map进行遍历, 遍历过程中对map没有加锁, 推荐使用
  136. > hashBegin(): 获取hash遍历迭代器
  137. > hashEnd(): hash遍历尾部迭代器
  138. > 注意:hash_iterator对应的其实是一个hash桶链, 每次获取数据其实会获取桶链上面的所有数据
  139. */
  140. template<typename K,
  141. typename V,
  142. typename LockPolicy,
  143. template<class, class> class StorePolicy>
  144. class TarsHashMapCompact : public StorePolicy<TC_HashMapCompact, LockPolicy>
  145. {
  146. public:
  147. /**
  148. * 定义数据操作基类
  149. * 获取,遍历,删除,淘汰时都可以使用该操作类
  150. */
  151. class ToDoFunctor
  152. {
  153. public:
  154. /**
  155. * 数据记录
  156. */
  157. struct DataRecord
  158. {
  159. K _key;
  160. V _value;
  161. bool _dirty;
  162. uint32_t _iSyncTime;
  163. uint32_t _expiret;
  164. uint8_t _ver;
  165. DataRecord() : _dirty(true), _iSyncTime(0)
  166. {
  167. }
  168. };
  169. /**
  170. * 析够
  171. */
  172. virtual ~ToDoFunctor(){};
  173. /**
  174. * 淘汰数据
  175. * @param stDataRecord: 被淘汰的数据
  176. */
  177. virtual void erase(const DataRecord &stDataRecord){};
  178. /**
  179. * 删除数据
  180. * @param bExists: 是否存在数据
  181. * @param stDataRecord: 数据, bExists==true时有效, 否则只有key有效
  182. */
  183. virtual void del(bool bExists, const DataRecord &stDataRecord){};
  184. /**
  185. * 回写数据
  186. * @param stDataRecord: 数据
  187. */
  188. virtual void sync(const DataRecord &stDataRecord){};
  189. /**
  190. * 备份数据
  191. * @param stDataRecord: 数据
  192. */
  193. virtual void backup(const DataRecord &stDataRecord){};
  194. /**
  195. * 获取数据, 默认返回RT_NO_GET
  196. * stDataRecord中_key有效, 其他数据需要返回
  197. * @param stDataRecord: 需要获取的数据
  198. *
  199. * @return int, 获取到数据, 返回:TC_HashMapCompact::RT_OK
  200. * 没有数据,返回:TC_HashMapCompact::RT_NO_DATA,
  201. * 系统默认GET,返回:TC_HashMapCompact::RT_NO_GET
  202. * 其他,则返回:TC_HashMapCompact::RT_LOAD_DATA_ERR
  203. */
  204. virtual int get(DataRecord &stDataRecord)
  205. {
  206. return TC_HashMapCompact::RT_NO_GET;
  207. }
  208. };
  209. typedef typename ToDoFunctor::DataRecord DataRecord;
  210. ///////////////////////////////////////////////////////////////////
  211. /**
  212. * 自动锁, 用于迭代器
  213. */
  214. class JhmAutoLock : public TC_HandleBase
  215. {
  216. public:
  217. /**
  218. * 构造
  219. * @param mutex
  220. */
  221. JhmAutoLock(typename LockPolicy::Mutex &mutex) : _lock(mutex)
  222. {
  223. }
  224. protected:
  225. //不实现
  226. JhmAutoLock(const JhmAutoLock &al);
  227. JhmAutoLock &operator=(const JhmAutoLock &al);
  228. protected:
  229. /**
  230. * 锁
  231. */
  232. TC_LockT<typename LockPolicy::Mutex> _lock;
  233. };
  234. typedef TC_AutoPtr<JhmAutoLock> JhmAutoLockPtr;
  235. ///////////////////////////////////////////////////////////////////
  236. /**
  237. * 数据项
  238. */
  239. class JhmLockItem
  240. {
  241. public:
  242. /**
  243. * 构造函数
  244. * @param item
  245. */
  246. JhmLockItem(const TC_HashMapCompact::HashMapLockItem &item)
  247. : _item(item)
  248. {
  249. }
  250. /**
  251. * 拷贝构造
  252. * @param it
  253. */
  254. JhmLockItem(const JhmLockItem &item)
  255. : _item(item._item)
  256. {
  257. }
  258. /**
  259. * 复制
  260. * @param it
  261. *
  262. * @return JhmLockItem&
  263. */
  264. JhmLockItem& operator=(const JhmLockItem &item)
  265. {
  266. if(this != &item)
  267. {
  268. _item = item._item;
  269. }
  270. return (*this);
  271. }
  272. /**
  273. *
  274. * @param item
  275. *
  276. * @return bool
  277. */
  278. bool operator==(const JhmLockItem& item)
  279. {
  280. return (_item == item._item);
  281. }
  282. /**
  283. *
  284. * @param item
  285. *
  286. * @return bool
  287. */
  288. bool operator!=(const JhmLockItem& item)
  289. {
  290. return !((*this) == item);
  291. }
  292. /**
  293. * 是否是脏数据
  294. *
  295. * @return bool
  296. */
  297. bool isDirty() { return _item.isDirty(); }
  298. /**
  299. * 是否只有Key
  300. *
  301. * @return bool
  302. */
  303. bool isOnlyKey() { return _item.isOnlyKey(); }
  304. /**
  305. * 最后回写时间
  306. *
  307. * @return uint32_t
  308. */
  309. uint32_t getSyncTime() { return _item.getSyncTime(); }
  310. /**
  311. * 获取值
  312. * @return int
  313. * TC_HashMapCompact::RT_OK:数据获取OK
  314. * 其他值, 异常
  315. */
  316. int get(K& k)
  317. {
  318. string sk;
  319. int ret = _item.get(sk);
  320. if(ret != TC_HashMapCompact::RT_OK)
  321. {
  322. return ret;
  323. }
  324. tars::TarsInputStream<BufferReader> is;
  325. is.setBuffer(sk.c_str(), sk.length());
  326. k.readFrom(is);
  327. return ret;
  328. }
  329. /**
  330. * 获取值
  331. * @return int
  332. * TC_HashMapCompact::RT_OK:数据获取OK
  333. * TC_HashMapCompact::RT_ONLY_KEY: key有效, v无效为空
  334. * 其他值, 异常
  335. */
  336. int get(K& k, V& v)
  337. {
  338. string sk;
  339. string sv;
  340. int ret = _item.get(sk, sv);
  341. if(ret != TC_HashMapCompact::RT_OK && ret != TC_HashMapCompact::RT_ONLY_KEY)
  342. {
  343. return ret;
  344. }
  345. tars::TarsInputStream<BufferReader> is;
  346. is.setBuffer(sk.c_str(), sk.length());
  347. k.readFrom(is);
  348. if(ret != TC_HashMapCompact::RT_ONLY_KEY)
  349. {
  350. is.setBuffer(sv.c_str(), sv.length());
  351. v.readFrom(is);
  352. }
  353. return ret;
  354. }
  355. protected:
  356. TC_HashMapCompact::HashMapLockItem _item;
  357. };
  358. ///////////////////////////////////////////////////////////////////
  359. /**
  360. * 迭代器
  361. */
  362. struct JhmLockIterator
  363. {
  364. public:
  365. /**
  366. * 构造
  367. * @param it
  368. * @param lock
  369. */
  370. JhmLockIterator(const TC_HashMapCompact::lock_iterator it, const JhmAutoLockPtr &lock)
  371. : _it(it), _item(it._iItem), _lock(lock)
  372. {
  373. }
  374. /**
  375. * 拷贝构造
  376. * @param it
  377. */
  378. JhmLockIterator(const JhmLockIterator &it)
  379. : _it(it._it), _item(it._item), _lock(it._lock)
  380. {
  381. }
  382. /**
  383. * 复制
  384. * @param it
  385. *
  386. * @return JhmLockIterator&
  387. */
  388. JhmLockIterator& operator=(const JhmLockIterator &it)
  389. {
  390. if(this != &it)
  391. {
  392. _it = it._it;
  393. _item = it._item;
  394. _lock = it._lock;
  395. }
  396. return (*this);
  397. }
  398. /**
  399. *
  400. * @param it
  401. *
  402. * @return bool
  403. */
  404. bool operator==(const JhmLockIterator& it)
  405. {
  406. return (_it == it._it && _item == it._item);
  407. }
  408. /**
  409. *
  410. * @param mv
  411. *
  412. * @return bool
  413. */
  414. bool operator!=(const JhmLockIterator& it)
  415. {
  416. return !((*this) == it);
  417. }
  418. /**
  419. * 前置++
  420. *
  421. * @return JhmLockIterator&
  422. */
  423. JhmLockIterator& operator++()
  424. {
  425. ++_it;
  426. _item = JhmLockItem(_it._iItem);
  427. return (*this);
  428. }
  429. /**
  430. * 后置++
  431. *
  432. * @return JhmLockIterator&
  433. */
  434. JhmLockIterator operator++(int)
  435. {
  436. JhmLockIterator jit(_it, _lock);
  437. ++_it;
  438. _item = JhmLockItem(_it._iItem);
  439. return jit;
  440. }
  441. /**
  442. * 获取数据项
  443. *
  444. * @return JhmLockItem&
  445. */
  446. JhmLockItem& operator*() { return _item; }
  447. /**
  448. * 获取数据项
  449. *
  450. * @return JhmLockItem*
  451. */
  452. JhmLockItem* operator->() { return &_item; }
  453. protected:
  454. /**
  455. * 迭代器
  456. */
  457. TC_HashMapCompact::lock_iterator _it;
  458. /**
  459. * 数据项
  460. */
  461. JhmLockItem _item;
  462. /**
  463. * 锁
  464. */
  465. JhmAutoLockPtr _lock;
  466. };
  467. typedef JhmLockIterator lock_iterator ;
  468. ///////////////////////////////////////////////////////////////////
  469. /**
  470. * 锁, 用于非锁迭代器
  471. *
  472. */
  473. class JhmLock : public TC_HandleBase
  474. {
  475. public:
  476. /**
  477. * 构造
  478. * @param mutex
  479. */
  480. JhmLock(typename LockPolicy::Mutex &mutex) : _mutex(mutex)
  481. {
  482. }
  483. /**
  484. * 获取锁
  485. *
  486. * @return typename LockPolicy::Mutex
  487. */
  488. typename LockPolicy::Mutex& mutex()
  489. {
  490. return _mutex;
  491. }
  492. protected:
  493. //不实现
  494. JhmLock(const JhmLock &al);
  495. JhmLock &operator=(const JhmLock &al);
  496. protected:
  497. /**
  498. * 锁
  499. */
  500. typename LockPolicy::Mutex &_mutex;
  501. };
  502. typedef TC_AutoPtr<JhmLock> JhmLockPtr;
  503. ///////////////////////////////////////////////////////////////////
  504. /**
  505. * 数据项
  506. */
  507. class JhmItem
  508. {
  509. public:
  510. /**
  511. * 构造函数
  512. * @param item
  513. */
  514. JhmItem(const TC_HashMapCompact::HashMapItem &item, const JhmLockPtr &lock)
  515. : _item(item), _lock(lock)
  516. {
  517. }
  518. /**
  519. * 拷贝构造
  520. * @param it
  521. */
  522. JhmItem(const JhmItem &item)
  523. : _item(item._item), _lock(item._lock)
  524. {
  525. }
  526. /**
  527. * 复制
  528. * @param it
  529. *
  530. * @return JhmItem&
  531. */
  532. JhmItem& operator=(const JhmItem &item)
  533. {
  534. if(this != &item)
  535. {
  536. _item = item._item;
  537. _lock = item._lock;
  538. }
  539. return (*this);
  540. }
  541. /**
  542. *
  543. * @param item
  544. *
  545. * @return bool
  546. */
  547. bool operator==(const JhmItem& item)
  548. {
  549. return (_item == item._item);
  550. }
  551. /**
  552. *
  553. * @param item
  554. *
  555. * @return bool
  556. */
  557. bool operator!=(const JhmItem& item)
  558. {
  559. return !((*this) == item);
  560. }
  561. /**
  562. * 获取当前hash桶的所有数量, 注意只获取有key/value的数据
  563. * 对于只有key的数据, 不获取
  564. * 如果协议解码有问题也不获取
  565. * @param
  566. */
  567. void get(vector<pair<K, V> > &v)
  568. {
  569. vector<TC_HashMapCompact::BlockData> vtData;
  570. {
  571. TC_LockT<typename LockPolicy::Mutex> lock(_lock->mutex());
  572. _item.get(vtData);
  573. }
  574. for(size_t i = 0; i < vtData.size(); i++)
  575. {
  576. pair<K, V> pk;
  577. try
  578. {
  579. tars::TarsInputStream<BufferReader> is;
  580. is.setBuffer(vtData[i]._key.c_str(), vtData[i]._key.length());
  581. pk.first.readFrom(is);
  582. is.setBuffer(vtData[i]._value.c_str(), vtData[i]._value.length());
  583. pk.second.readFrom(is);
  584. v.push_back(pk);
  585. }
  586. catch(exception &ex)
  587. {
  588. }
  589. }
  590. }
  591. /**
  592. * 获取当前hash桶的所有过期数据, 注意只获取有key/value的数据
  593. * 对于只有key的数据, 不获取
  594. * 如果协议解码有问题也不获取
  595. * @param
  596. */
  597. void getExpire(uint32_t t, vector<pair<K, V> > &v)
  598. {
  599. vector<TC_HashMapCompact::BlockData> vtData;
  600. {
  601. TC_LockT<typename LockPolicy::Mutex> lock(_lock->mutex());
  602. _item.getExpire(t, vtData);
  603. }
  604. for(size_t i = 0; i < vtData.size(); i++)
  605. {
  606. pair<K, V> pk;
  607. try
  608. {
  609. tars::TarsInputStream<BufferReader> is;
  610. is.setBuffer(vtData[i]._key.c_str(), vtData[i]._key.length());
  611. pk.first.readFrom(is);
  612. is.setBuffer(vtData[i]._value.c_str(), vtData[i]._value.length());
  613. pk.second.readFrom(is);
  614. v.push_back(pk);
  615. }
  616. catch(exception &ex)
  617. {
  618. }
  619. }
  620. }
  621. /**
  622. * 设置当前桶下的所有数据为脏数据
  623. *
  624. * @param
  625. *
  626. * @return int
  627. */
  628. int setDirty()
  629. {
  630. {
  631. TC_LockT<typename LockPolicy::Mutex> lock(_lock->mutex());
  632. return _item.setDirty();
  633. }
  634. }
  635. protected:
  636. TC_HashMapCompact::HashMapItem _item;
  637. JhmLockPtr _lock;
  638. };
  639. ///////////////////////////////////////////////////////////////////
  640. /**
  641. * 迭代器
  642. */
  643. struct JhmIterator
  644. {
  645. public:
  646. /**
  647. * 构造
  648. * @param it
  649. * @param lock
  650. */
  651. JhmIterator(const TC_HashMapCompact::hash_iterator &it, const JhmLockPtr &lock)
  652. : _it(it), _item(it._iItem, lock), _lock(lock)
  653. {
  654. }
  655. /**
  656. * 拷贝构造
  657. * @param it
  658. */
  659. JhmIterator(const JhmIterator &it)
  660. : _it(it._it), _item(it._item), _lock(it._lock)
  661. {
  662. }
  663. /**
  664. * 复制
  665. * @param it
  666. *
  667. * @return JhmIterator&
  668. */
  669. JhmIterator& operator=(const JhmIterator &it)
  670. {
  671. if(this != &it)
  672. {
  673. _it = it._it;
  674. _item = it._item;
  675. }
  676. return (*this);
  677. }
  678. /**
  679. *
  680. * @param it
  681. *
  682. * @return bool
  683. */
  684. bool operator==(const JhmIterator& it)
  685. {
  686. return (_it == it._it && _item == it._item);
  687. }
  688. /**
  689. *
  690. * @param mv
  691. *
  692. * @return bool
  693. */
  694. bool operator!=(const JhmIterator& it)
  695. {
  696. return !((*this) == it);
  697. }
  698. /**
  699. * 前置++
  700. *
  701. * @return JhmIterator&
  702. */
  703. JhmIterator& operator++()
  704. {
  705. TC_LockT<typename LockPolicy::Mutex> lock(_lock->mutex());
  706. ++_it;
  707. _item = JhmItem(_it._iItem, _lock);
  708. return (*this);
  709. }
  710. /**
  711. * 后置++
  712. *
  713. * @return JhmIterator&
  714. */
  715. JhmIterator operator++(int)
  716. {
  717. TC_LockT<typename LockPolicy::Mutex> lock(_lock->mutex());
  718. JhmIterator jit(_it, _lock);
  719. ++_it;
  720. _item = JhmItem(_it._iItem, _lock);
  721. return jit;
  722. }
  723. /**
  724. * 获取数据项
  725. *
  726. * @return JhmItem&
  727. */
  728. JhmItem& operator*() { return _item; }
  729. /**
  730. * 获取数据项
  731. *
  732. * @return JhmItem*
  733. */
  734. JhmItem* operator->() { return &_item; }
  735. protected:
  736. /**
  737. * 迭代器
  738. */
  739. TC_HashMapCompact::hash_iterator _it;
  740. /**
  741. * 数据项
  742. */
  743. JhmItem _item;
  744. /**
  745. * 锁
  746. */
  747. JhmLockPtr _lock;
  748. };
  749. typedef JhmIterator hash_iterator ;
  750. ////////////////////////////////////////////////////////////////////////////
  751. //
  752. /**
  753. * 构造函数
  754. */
  755. TarsHashMapCompact()
  756. {
  757. _todo_of = NULL;
  758. }
  759. /**
  760. * 初始化数据块平均大小
  761. * 表示内存分配的时候,会分配n个最小块, n个(最小快*增长因子), n个(最小快*增长因子*增长因子)..., 直到n个最大块
  762. * n是hashmap自己计算出来的
  763. * 这种分配策略通常是你数据快记录变长比较多的使用, 便于节约内存,如果数据记录基本不是变长的, 那最小块=最大快,增长因子=1就可以了
  764. * @param iMinDataSize: 最小数据块大小
  765. * @param iMaxDataSize: 最大数据块大小
  766. * @param fFactor: 增长因子 >= 1.0
  767. */
  768. void initDataBlockSize(size_t iMinDataSize, size_t iMaxDataSize, float fFactor)
  769. {
  770. this->_t.initDataBlockSize(iMinDataSize, iMaxDataSize, fFactor);
  771. }
  772. /**
  773. * 设置hash比率(设置chunk数据块/hash项比值, 默认是2)
  774. * 有需要更改必须在create之前调用
  775. *
  776. * @param fRadio
  777. */
  778. void initHashRadio(float fRadio) { this->_t.initHashRadio(fRadio);}
  779. /**
  780. * 设置hash方式
  781. * @param hash_of
  782. */
  783. void setHashFunctor(TC_HashMapCompact::hash_functor hashf)
  784. {
  785. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  786. this->_t.setHashFunctor(hashf);
  787. }
  788. /**
  789. * 获取hash方式
  790. *
  791. * @return TC_HashMapCompact::hash_functor&
  792. */
  793. TC_HashMapCompact::hash_functor &getHashFunctor() { return this->_t.getHashFunctor(); }
  794. /**
  795. * 设置淘汰操作类
  796. * @param erase_of
  797. */
  798. void setToDoFunctor(ToDoFunctor *todo_of) { this->_todo_of = todo_of; }
  799. /**
  800. * 获取每种大小内存块的头部信息
  801. *
  802. * @return vector<TC_MemChunk::tagChunkHead>: 不同大小内存块头部信息
  803. */
  804. vector<TC_MemChunk::tagChunkHead> getBlockDetail()
  805. {
  806. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  807. return this->_t.getBlockDetail();
  808. }
  809. /**
  810. * 所有block中chunk的个数
  811. *
  812. * @return size_t
  813. */
  814. size_t allBlockChunkCount()
  815. {
  816. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  817. return this->_t.allBlockChunkCount();
  818. }
  819. /**
  820. * 每种block中chunk的个数(不同大小内存块的个数相同)
  821. *
  822. * @return size_t
  823. */
  824. vector<size_t> singleBlockChunkCount()
  825. {
  826. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  827. return this->_t.singleBlockChunkCount();
  828. }
  829. /**
  830. * 获取hash桶的个数
  831. *
  832. * @return size_t
  833. */
  834. size_t getHashCount()
  835. {
  836. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  837. return this->_t.getHashCount();
  838. }
  839. /**
  840. * 元素的个数
  841. *
  842. * @return size_t
  843. */
  844. size_t size()
  845. {
  846. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  847. return this->_t.size();
  848. }
  849. /**
  850. * 脏数据元素个数
  851. *
  852. * @return size_t
  853. */
  854. size_t dirtyCount()
  855. {
  856. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  857. return this->_t.dirtyCount();
  858. }
  859. /**
  860. * Only数据元素个数
  861. *
  862. * @return size_t
  863. */
  864. size_t onlyKeyCount()
  865. {
  866. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  867. return this->_t.onlyKeyCount();
  868. }
  869. /**
  870. * 设置每次淘汰数量
  871. * @param n
  872. */
  873. void setEraseCount(size_t n)
  874. {
  875. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  876. this->_t.setEraseCount(n);
  877. }
  878. /**
  879. * 获取每次淘汰数量
  880. *
  881. * @return size_t
  882. */
  883. size_t getEraseCount()
  884. {
  885. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  886. return this->_t.getEraseCount();
  887. }
  888. /**
  889. * 设置只读
  890. * @param bReadOnly
  891. */
  892. void setReadOnly(bool bReadOnly)
  893. {
  894. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  895. this->_t.setReadOnly(bReadOnly);
  896. }
  897. /**
  898. * 是否只读
  899. *
  900. * @return bool
  901. */
  902. bool isReadOnly()
  903. {
  904. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  905. return this->_t.isReadOnly();
  906. }
  907. /**
  908. * 设置是否可以自动淘汰
  909. * @param bAutoErase
  910. */
  911. void setAutoErase(bool bAutoErase)
  912. {
  913. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  914. this->_t.setAutoErase(bAutoErase);
  915. }
  916. /**
  917. * 是否可以自动淘汰
  918. *
  919. * @return bool
  920. */
  921. bool isAutoErase()
  922. {
  923. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  924. return this->_t.isAutoErase();
  925. }
  926. /**
  927. * 设置淘汰方式
  928. * TC_HashMapCompact::ERASEBYGET
  929. * TC_HashMapCompact::ERASEBYSET
  930. * @param cEraseMode
  931. */
  932. void setEraseMode(char cEraseMode)
  933. {
  934. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  935. this->_t.setEraseMode(cEraseMode);
  936. }
  937. /**
  938. * 获取淘汰方式
  939. *
  940. * @return bool
  941. */
  942. char getEraseMode()
  943. {
  944. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  945. return this->_t.getEraseMode();
  946. }
  947. /**
  948. * 头部信息
  949. *
  950. * @return TC_HashMapCompact::tagMapHead
  951. */
  952. TC_HashMapCompact::tagMapHead& getMapHead()
  953. {
  954. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  955. return this->_t.getMapHead();
  956. }
  957. /**
  958. * 设置回写时间(秒)
  959. * @param iSyncTime
  960. */
  961. void setSyncTime(uint32_t iSyncTime)
  962. {
  963. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  964. this->_t.setSyncTime(iSyncTime);
  965. }
  966. /**
  967. * 获取回写时间
  968. *
  969. * @return uint32_t
  970. */
  971. uint32_t getSyncTime()
  972. {
  973. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  974. return this->_t.getSyncTime();
  975. }
  976. /**
  977. * dump到文件
  978. * @param sFile
  979. * @param bDoClear: 是否清空
  980. * @return int
  981. * TC_HashMapCompact::RT_DUMP_FILE_ERR: dump到文件出错
  982. * TC_HashMapCompact::RT_OK: dump到文件成功
  983. */
  984. int dump2file(const string &sFile, bool bDoClear = false)
  985. {
  986. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  987. int ret = this->_t.dump2file(sFile);
  988. if(ret != TC_HashMapCompact::RT_OK)
  989. {
  990. return ret;
  991. }
  992. if(bDoClear)
  993. this->_t.clear();
  994. return ret;
  995. }
  996. /**
  997. * 从文件load
  998. * @param sFile
  999. *
  1000. * @return int
  1001. * TC_HashMapCompact::RT_LOAL_FILE_ERR: load出错
  1002. * TC_HashMapCompact::RT_VERSION_MISMATCH_ERR: 版本不一致
  1003. * TC_HashMapCompact::RT_OK: load成功
  1004. */
  1005. int load5file(const string &sFile)
  1006. {
  1007. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1008. return this->_t.load5file(sFile);
  1009. }
  1010. /**
  1011. * 清空hash map
  1012. * 所有map中的数据都被清空
  1013. */
  1014. void clear()
  1015. {
  1016. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1017. return this->_t.clear();
  1018. }
  1019. /**
  1020. * 检查数据状态
  1021. * @param k
  1022. *
  1023. * @return int
  1024. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1025. * TC_HashMapCompact::RT_ONLY_KEY:只有Key
  1026. * TC_HashMapCompact::RT_DIRTY_DATA: 是脏数据
  1027. * TC_HashMapCompact::RT_OK: 是干净数据
  1028. * 其他返回值: 错误
  1029. */
  1030. int checkDirty(const K &k)
  1031. {
  1032. tars::TarsOutputStream<BufferWriter> osk;
  1033. k.writeTo(osk);
  1034. string sk(osk.getBuffer(), osk.getLength());
  1035. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1036. return this->_t.checkDirty(sk);
  1037. }
  1038. /**
  1039. * 设置为干净数据i, 修改SET/GET时间链, 会导致数据不回写
  1040. * @param k
  1041. *
  1042. * @return int
  1043. * TC_HashMapCompact::RT_READONLY: 只读
  1044. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1045. * TC_HashMapCompact::RT_ONLY_KEY:只有Key
  1046. * TC_HashMapCompact::RT_OK: 设置成功
  1047. * 其他返回值: 错误
  1048. */
  1049. int setClean(const K& k)
  1050. {
  1051. tars::TarsOutputStream<BufferWriter> osk;
  1052. k.writeTo(osk);
  1053. string sk(osk.getBuffer(), osk.getLength());
  1054. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1055. return this->_t.setClean(sk);
  1056. }
  1057. /**
  1058. * 设置为脏数据, 修改SET/GET时间链, 会导致数据回写
  1059. * @param k
  1060. * @return int
  1061. * TC_HashMapCompact::RT_READONLY: 只读
  1062. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1063. * TC_HashMapCompact::RT_ONLY_KEY:只有Key
  1064. * TC_HashMapCompact::RT_OK: 设置脏数据成功
  1065. * 其他返回值: 错误
  1066. */
  1067. int setDirty(const K& k)
  1068. {
  1069. tars::TarsOutputStream<BufferWriter> osk;
  1070. k.writeTo(osk);
  1071. string sk(osk.getBuffer(), osk.getLength());
  1072. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1073. return this->_t.setDirty(sk);
  1074. }
  1075. /**
  1076. * 获取数据, 修改GET时间链
  1077. * (如果没设置自定义Get函数,没有数据时返回:RT_NO_DATA)
  1078. * @param k
  1079. * @param v
  1080. * @param iSyncTime:数据上次回写的时间, 没有缓写则为0
  1081. *
  1082. * @return int:
  1083. * TC_HashMapCompact::RT_NO_DATA: 没有数据
  1084. * TC_HashMapCompact::RT_READONLY: 只读模式
  1085. * TC_HashMapCompact::RT_ONLY_KEY:只有Key
  1086. * TC_HashMapCompact::RT_OK:获取数据成功
  1087. * TC_HashMapCompact::RT_LOAD_DATA_ERR: load数据失败
  1088. * 其他返回值: 错误
  1089. */
  1090. int get(const K& k, V &v, uint32_t &iSyncTime, uint32_t& iExpireTime, uint8_t& iVersion)
  1091. {
  1092. iSyncTime = 0;
  1093. iExpireTime = 0;
  1094. iVersion = 1;
  1095. int ret = TC_HashMapCompact::RT_OK;
  1096. tars::TarsOutputStream<BufferWriter> osk;
  1097. k.writeTo(osk);
  1098. string sk(osk.getBuffer(), osk.getLength());
  1099. string sv;
  1100. {
  1101. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1102. ret = this->_t.get(sk, sv, iSyncTime, iExpireTime, iVersion);
  1103. }
  1104. //读取到数据了, 解包
  1105. if(ret == TC_HashMapCompact::RT_OK)
  1106. {
  1107. tars::TarsInputStream<BufferReader> is;
  1108. is.setBuffer(sv.c_str(), sv.length());
  1109. v.readFrom(is);
  1110. return ret;
  1111. }
  1112. if(ret != TC_HashMapCompact::RT_NO_DATA || _todo_of == NULL)
  1113. {
  1114. return ret;
  1115. }
  1116. //只读模式
  1117. if(isReadOnly())
  1118. {
  1119. return TC_HashMapCompact::RT_READONLY;
  1120. }
  1121. //获取函数
  1122. typename ToDoFunctor::DataRecord stDataRecord;
  1123. stDataRecord._key = k;
  1124. ret = _todo_of->get(stDataRecord);
  1125. if(ret == TC_HashMapCompact::RT_OK)
  1126. {
  1127. v = stDataRecord._value;
  1128. iExpireTime = stDataRecord._expiret;
  1129. return this->set(stDataRecord._key, stDataRecord._value, stDataRecord._expiret, stDataRecord._ver, stDataRecord._dirty);
  1130. }
  1131. else if(ret == TC_HashMapCompact::RT_NO_GET)
  1132. {
  1133. return TC_HashMapCompact::RT_NO_DATA;
  1134. }
  1135. else if(ret == TC_HashMapCompact::RT_NO_DATA)
  1136. {
  1137. iExpireTime = stDataRecord._expiret;
  1138. ret = this->set(stDataRecord._key);
  1139. if(ret == TC_HashMapCompact::RT_OK)
  1140. {
  1141. return TC_HashMapCompact::RT_ONLY_KEY;
  1142. }
  1143. return ret;
  1144. }
  1145. return TC_HashMapCompact::RT_LOAD_DATA_ERR;
  1146. }
  1147. /**
  1148. * 获取数据, 修改GET时间链
  1149. * @param k
  1150. * @param v
  1151. *
  1152. * @return int:
  1153. * TC_HashMapCompact::RT_NO_DATA: 没有数据
  1154. * TC_HashMapCompact::RT_ONLY_KEY:只有Key
  1155. * TC_HashMapCompact::RT_OK:获取数据成功
  1156. * TC_HashMapCompact::RT_LOAD_DATA_ERR: load数据失败
  1157. * 其他返回值: 错误
  1158. */
  1159. int get(const K& k, V &v)
  1160. {
  1161. uint32_t iSyncTime;
  1162. uint32_t iExpireTime;
  1163. uint8_t iVer;
  1164. return get(k, v, iSyncTime, iExpireTime, iVer);
  1165. }
  1166. /**
  1167. * 根据key, 获取相同hash值的所有数据
  1168. * 注意:c匹配对象操作中, map是加锁的, 需要注意
  1169. * @param h
  1170. * @param vv
  1171. * @param c, 匹配仿函数: bool operator()(K v);
  1172. *
  1173. * @return int, RT_OK
  1174. */
  1175. template<typename C>
  1176. int getHash(size_t h, vector<pair<K, V> > &vv, C c)
  1177. {
  1178. int ret = TC_HashMapCompact::RT_OK;
  1179. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1180. TC_HashMapCompact::FailureRecover check(&this->_t);
  1181. size_t index = h % this->_t.getHashCount();
  1182. size_t iAddr = this->_t.item(index)->_iBlockAddr;
  1183. TC_HashMapCompact::Block block(&this->_t, iAddr);
  1184. while(block.getHead() != 0)
  1185. {
  1186. TC_HashMapCompact::BlockData data;
  1187. ret = block.getBlockData(data);
  1188. if(ret == TC_HashMapCompact::RT_OK)
  1189. {
  1190. K tk;
  1191. V tv;
  1192. try
  1193. {
  1194. tars::TarsInputStream<BufferReader> is;
  1195. is.setBuffer(data._key.c_str(), data._key.length());
  1196. tk.readFrom(is);
  1197. if(c(tk))
  1198. {
  1199. is.setBuffer(data._value.c_str(), data._value.length());
  1200. tv.readFrom(is);
  1201. vv.push_back(make_pair(tk, tv));
  1202. }
  1203. }
  1204. catch(exception &ex)
  1205. {
  1206. }
  1207. }
  1208. if(!block.nextBlock())
  1209. {
  1210. break;
  1211. }
  1212. }
  1213. return TC_HashMapCompact::RT_OK;
  1214. }
  1215. /**
  1216. * 根据key, 获取相同hash值的所有数据
  1217. * 注意:c匹配对象操作中, map是加锁的, 需要注意
  1218. * @param h
  1219. * @param vv
  1220. * @param c, 匹配仿函数: bool operator()(K v);
  1221. *
  1222. * @return int, RT_OK
  1223. */
  1224. template<typename C>
  1225. int getHash(size_t h, vector<DataRecord> &vv, C c)
  1226. {
  1227. int ret = TC_HashMapCompact::RT_OK;
  1228. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1229. TC_HashMapCompact::FailureRecover check(&this->_t);
  1230. size_t index = h % this->_t.getHashCount();
  1231. size_t iAddr = this->_t.item(index)->_iBlockAddr;
  1232. TC_HashMapCompact::Block block(&this->_t, iAddr);
  1233. while(block.getHead() != 0)
  1234. {
  1235. TC_HashMapCompact::BlockData data;
  1236. ret = block.getBlockData(data);
  1237. if(ret == TC_HashMapCompact::RT_OK)
  1238. {
  1239. K tk;
  1240. V tv;
  1241. try
  1242. {
  1243. tars::TarsInputStream<BufferReader> is;
  1244. is.setBuffer(data._key.c_str(), data._key.length());
  1245. tk.readFrom(is);
  1246. if(c(tk))
  1247. {
  1248. is.setBuffer(data._value.c_str(), data._value.length());
  1249. tv.readFrom(is);
  1250. DataRecord stDataRecord;
  1251. stDataRecord._key = tk;
  1252. stDataRecord._value = tv;
  1253. stDataRecord._ver = data._ver;
  1254. stDataRecord._dirty = data._dirty;
  1255. stDataRecord._expiret = data._expiret;
  1256. stDataRecord._iSyncTime = data._synct;
  1257. vv.push_back(stDataRecord);
  1258. }
  1259. }
  1260. catch(exception &ex)
  1261. {
  1262. }
  1263. }
  1264. if(!block.nextBlock())
  1265. {
  1266. break;
  1267. }
  1268. }
  1269. return TC_HashMapCompact::RT_OK;
  1270. }
  1271. /**
  1272. * 恢复数据
  1273. * 对于block记录无法读取的数据自动删除
  1274. * @param bRepair: 是否修复
  1275. * @return 返回删除的记录数
  1276. */
  1277. size_t recover(bool bRepair)
  1278. {
  1279. size_t c = this->_t.getHashCount();
  1280. size_t e = 0;
  1281. for(size_t i = 0; i < c; i++)
  1282. {
  1283. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1284. e += this->_t.recover(i, bRepair);
  1285. }
  1286. return e;
  1287. }
  1288. /**
  1289. * 设置数据, 修改时间链, 内存不够时会自动淘汰老的数据
  1290. * @param k: 关键字
  1291. * @param v: 值
  1292. * @param bDirty: 是否是脏数据
  1293. * @return int:
  1294. * TC_HashMapCompact::RT_READONLY: map只读
  1295. * TC_HashMapCompact::RT_NO_MEMORY: 没有空间(不淘汰数据情况下会出现)
  1296. * TC_HashMapCompact::RT_OK: 设置成功
  1297. * 其他返回值: 错误
  1298. */
  1299. int set(const K& k, const V& v, bool bDirty = true, uint32_t iExpireTime = 0, uint8_t iVersion = 0)
  1300. {
  1301. tars::TarsOutputStream<BufferWriter> osk;
  1302. k.writeTo(osk);
  1303. string sk(osk.getBuffer(), osk.getLength());
  1304. tars::TarsOutputStream<BufferWriter> osv;
  1305. v.writeTo(osv);
  1306. string sv(osv.getBuffer(), osv.getLength());
  1307. int ret = TC_HashMapCompact::RT_OK;
  1308. vector<TC_HashMapCompact::BlockData> vtData;
  1309. {
  1310. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1311. ret = this->_t.set(sk, sv, iExpireTime, iVersion, bDirty, vtData);
  1312. }
  1313. //操作淘汰数据
  1314. if(_todo_of)
  1315. {
  1316. for(size_t i = 0; i < vtData.size(); i++)
  1317. {
  1318. K tk;
  1319. V tv;
  1320. try
  1321. {
  1322. tars::TarsInputStream<BufferReader> is;
  1323. is.setBuffer(vtData[i]._key.c_str(), vtData[i]._key.length());
  1324. tk.readFrom(is);
  1325. is.setBuffer(vtData[i]._value.c_str(), vtData[i]._value.length());
  1326. tv.readFrom(is);
  1327. typename ToDoFunctor::DataRecord stDataRecord;
  1328. stDataRecord._key = tk;
  1329. stDataRecord._value = tv;
  1330. stDataRecord._dirty = vtData[i]._dirty;
  1331. stDataRecord._iSyncTime = vtData[i]._synct;
  1332. stDataRecord._expiret = vtData[i]._expiret;
  1333. stDataRecord._ver = vtData[i]._ver;
  1334. _todo_of->sync(stDataRecord);
  1335. }
  1336. catch(exception &ex)
  1337. {
  1338. }
  1339. }
  1340. }
  1341. return ret;
  1342. }
  1343. /**
  1344. * 仅设置Key, 内存不够时会自动淘汰老的数据
  1345. * @param k: 关键字
  1346. * @return int:
  1347. * TC_HashMapCompact::RT_READONLY: map只读
  1348. * TC_HashMapCompact::RT_NO_MEMORY: 没有空间(不淘汰数据情况下会出现)
  1349. * TC_HashMapCompact::RT_OK: 设置成功
  1350. * 其他返回值: 错误
  1351. */
  1352. int set(const K& k, uint8_t iVersion = 0)
  1353. {
  1354. tars::TarsOutputStream<BufferWriter> osk;
  1355. k.writeTo(osk);
  1356. string sk(osk.getBuffer(), osk.getLength());
  1357. int ret = TC_HashMapCompact::RT_OK;
  1358. vector<TC_HashMapCompact::BlockData> vtData;
  1359. {
  1360. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1361. ret = this->_t.set(sk, iVersion, vtData);
  1362. }
  1363. //操作淘汰数据
  1364. if(_todo_of)
  1365. {
  1366. for(size_t i = 0; i < vtData.size(); i++)
  1367. {
  1368. K tk;
  1369. V tv;
  1370. try
  1371. {
  1372. tars::TarsInputStream<BufferReader> is;
  1373. is.setBuffer(vtData[i]._key.c_str(), vtData[i]._key.length());
  1374. tk.readFrom(is);
  1375. is.setBuffer(vtData[i]._value.c_str(), vtData[i]._value.length());
  1376. tv.readFrom(is);
  1377. typename ToDoFunctor::DataRecord stDataRecord;
  1378. stDataRecord._key = tk;
  1379. stDataRecord._value = tv;
  1380. stDataRecord._dirty = vtData[i]._dirty;
  1381. stDataRecord._iSyncTime = vtData[i]._synct;
  1382. stDataRecord._expiret = vtData[i]._expiret;
  1383. stDataRecord._ver = vtData[i]._ver;
  1384. _todo_of->sync(stDataRecord);
  1385. }
  1386. catch(exception &ex)
  1387. {
  1388. }
  1389. }
  1390. }
  1391. return ret;
  1392. }
  1393. /**
  1394. * 删除数据
  1395. * 无论cache是否有数据,todo的del都被调用
  1396. *
  1397. * @param k, 关键字
  1398. * @return int:
  1399. * TC_HashMapCompact::RT_READONLY: map只读
  1400. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1401. * TC_HashMapCompact::RT_ONLY_KEY:只有Key, 也删除了
  1402. * TC_HashMapCompact::RT_OK: 删除数据成功
  1403. * 其他返回值: 错误
  1404. */
  1405. int del(const K& k)
  1406. {
  1407. int ret = TC_HashMapCompact::RT_OK;
  1408. TC_HashMapCompact::BlockData data;
  1409. tars::TarsOutputStream<BufferWriter> os;
  1410. k.writeTo(os);
  1411. string sk(os.getBuffer(), os.getLength());
  1412. {
  1413. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1414. ret = this->_t.del(sk, data);
  1415. }
  1416. if(ret != TC_HashMapCompact::RT_OK && ret != TC_HashMapCompact::RT_ONLY_KEY && ret != TC_HashMapCompact::RT_NO_DATA)
  1417. {
  1418. return ret;
  1419. }
  1420. if(_todo_of)
  1421. {
  1422. typename ToDoFunctor::DataRecord stDataRecord;
  1423. stDataRecord._key = k;
  1424. if(ret == TC_HashMapCompact::RT_OK)
  1425. {
  1426. V v;
  1427. tars::TarsInputStream<BufferReader> is;
  1428. is.setBuffer(data._value.c_str(), data._value.length());
  1429. v.readFrom(is);
  1430. stDataRecord._value = v;
  1431. stDataRecord._dirty = data._dirty;
  1432. stDataRecord._iSyncTime = data._synct;
  1433. stDataRecord._expiret = data._expiret;
  1434. stDataRecord._ver = data._ver;
  1435. }
  1436. _todo_of->del((ret == TC_HashMapCompact::RT_OK), stDataRecord);
  1437. }
  1438. return ret;
  1439. }
  1440. /**
  1441. * 删除数据
  1442. * cache有数据,todo的erase被调用
  1443. *
  1444. * @param k, 关键字
  1445. * @return int:
  1446. * TC_HashMapCompact::RT_READONLY: map只读
  1447. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1448. * TC_HashMapCompact::RT_ONLY_KEY:只有Key, 也删除了
  1449. * TC_HashMapCompact::RT_OK: 删除数据成功
  1450. * 其他返回值: 错误
  1451. */
  1452. int erase(const K& k)
  1453. {
  1454. int ret = TC_HashMapCompact::RT_OK;
  1455. TC_HashMapCompact::BlockData data;
  1456. tars::TarsOutputStream<BufferWriter> os;
  1457. k.writeTo(os);
  1458. string sk(os.getBuffer(), os.getLength());
  1459. {
  1460. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1461. ret = this->_t.del(sk, data);
  1462. }
  1463. if(ret != TC_HashMapCompact::RT_OK)
  1464. {
  1465. return ret;
  1466. }
  1467. if(_todo_of)
  1468. {
  1469. V v;
  1470. tars::TarsInputStream<BufferReader> is;
  1471. is.setBuffer(data._value.c_str(), data._value.length());
  1472. v.readFrom(is);
  1473. typename ToDoFunctor::DataRecord stDataRecord;
  1474. stDataRecord._key = k;
  1475. stDataRecord._value = v;
  1476. stDataRecord._dirty = data._dirty;
  1477. stDataRecord._iSyncTime = data._synct;
  1478. stDataRecord._expiret = data._expiret;
  1479. stDataRecord._ver = data._ver;
  1480. _todo_of->erase(stDataRecord);
  1481. }
  1482. return ret;
  1483. }
  1484. /**
  1485. * 强制删除数据,不调用todo的erase被调用
  1486. *
  1487. * @param k, 关键字
  1488. * @return int:
  1489. * TC_HashMapCompact::RT_READONLY: map只读
  1490. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1491. * TC_HashMapCompact::RT_ONLY_KEY:只有Key, 也删除了
  1492. * TC_HashMapCompact::RT_OK: 删除数据成功
  1493. * 其他返回值: 错误
  1494. */
  1495. int eraseByForce(const K& k)
  1496. {
  1497. int ret = TC_HashMapCompact::RT_OK;
  1498. TC_HashMapCompact::BlockData data;
  1499. tars::TarsOutputStream<BufferWriter> os;
  1500. k.writeTo(os);
  1501. string sk(os.getBuffer(), os.getLength());
  1502. {
  1503. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1504. ret = this->_t.del(sk, data);
  1505. }
  1506. if(ret != TC_HashMapCompact::RT_OK)
  1507. {
  1508. return ret;
  1509. }
  1510. return ret;
  1511. }
  1512. /**
  1513. * 根据key, 获取相同hash值的所有数据
  1514. * 注意:c匹配对象操作中, map是加锁的, 需要注意
  1515. * @param h
  1516. * @param vv
  1517. * @param c, 匹配仿函数: bool operator()(K v);
  1518. *
  1519. * @return int, RT_OK
  1520. */
  1521. /**
  1522. * 根据hash,强制删除相同hash值的所有数据,不调用todo的erase被调用
  1523. * 注意:c匹配对象操作中, map是加锁的, 需要注意
  1524. * @param h
  1525. * @param c, 匹配仿函数: bool operator()(K v);
  1526. * @return int:
  1527. * TC_HashMapCompact::RT_READONLY: map只读
  1528. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1529. * TC_HashMapCompact::RT_ONLY_KEY:只有Key, 也删除了
  1530. * TC_HashMapCompact::RT_OK: 删除数据成功
  1531. * 其他返回值: 错误
  1532. */
  1533. template<typename C>
  1534. int eraseHashByForce(size_t h, C c)
  1535. {
  1536. int ret = TC_HashMapCompact::RT_OK;
  1537. vector<string> vDelKey;
  1538. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1539. TC_HashMapCompact::FailureRecover check(&this->_t);
  1540. size_t index = h % this->_t.getHashCount();
  1541. size_t iAddr = this->_t.item(index)->_iBlockAddr;
  1542. TC_HashMapCompact::Block block(&this->_t, iAddr);
  1543. TC_HashMapCompact::BlockData data;
  1544. while(block.getHead() != 0)
  1545. {
  1546. ret = block.getBlockData(data);
  1547. if(ret == TC_HashMapCompact::RT_OK)
  1548. {
  1549. K tk;
  1550. V tv;
  1551. try
  1552. {
  1553. tars::TarsInputStream<BufferReader> is;
  1554. is.setBuffer(data._key.c_str(), data._key.length());
  1555. tk.readFrom(is);
  1556. if(c(tk))
  1557. vDelKey.push_back(data._key);
  1558. }
  1559. catch(exception &ex)
  1560. {
  1561. }
  1562. }
  1563. if(!block.nextBlock())
  1564. {
  1565. break;
  1566. }
  1567. }
  1568. for(size_t i=0; i<vDelKey.size(); ++i)
  1569. {
  1570. ret = this->_t.del(vDelKey[i], data);
  1571. if(ret != TC_HashMapCompact::RT_OK)
  1572. {
  1573. return ret;
  1574. }
  1575. }
  1576. return TC_HashMapCompact::RT_OK;
  1577. }
  1578. /**
  1579. * 根据key, 获取相同hash值的所有数据
  1580. * 注意:c匹配对象操作中, map是加锁的, 需要注意
  1581. * @param h
  1582. * @param vv
  1583. * @param c, 匹配仿函数: bool operator()(K v);
  1584. *
  1585. * @return int, RT_OK
  1586. */
  1587. /**
  1588. * 根据hash,强制删除相同hash值的所有数据,不调用todo的erase被调用
  1589. * 注意:c匹配对象操作中, map是加锁的, 需要注意
  1590. * @param h
  1591. * @param c, 匹配仿函数: bool operator()(K v);
  1592. * @return int:
  1593. * TC_HashMapCompact::RT_READONLY: map只读
  1594. * TC_HashMapCompact::RT_NO_DATA: 没有当前数据
  1595. * TC_HashMapCompact::RT_ONLY_KEY:只有Key, 也删除了
  1596. * TC_HashMapCompact::RT_OK: 删除数据成功
  1597. * 其他返回值: 错误
  1598. */
  1599. template<typename C>
  1600. int eraseHashByForce(size_t h, C c, vector<K>& vDelK)
  1601. {
  1602. int ret = TC_HashMapCompact::RT_OK;
  1603. vector<string> vDelKey;
  1604. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1605. TC_HashMapCompact::FailureRecover check(&this->_t);
  1606. size_t index = h % this->_t.getHashCount();
  1607. size_t iAddr = this->_t.item(index)->_iBlockAddr;
  1608. TC_HashMapCompact::Block block(&this->_t, iAddr);
  1609. TC_HashMapCompact::BlockData data;
  1610. while(block.getHead() != 0)
  1611. {
  1612. ret = block.getBlockData(data);
  1613. if(ret == TC_HashMapCompact::RT_OK)
  1614. {
  1615. K tk;
  1616. try
  1617. {
  1618. tars::TarsInputStream<BufferReader> is;
  1619. is.setBuffer(data._key.c_str(), data._key.length());
  1620. tk.readFrom(is);
  1621. if(c(tk))
  1622. {
  1623. vDelKey.push_back(data._key);
  1624. vDelK.push_back(tk);
  1625. }
  1626. }
  1627. catch(exception &ex)
  1628. {
  1629. }
  1630. }
  1631. if(!block.nextBlock())
  1632. {
  1633. break;
  1634. }
  1635. }
  1636. for(size_t i=0; i<vDelKey.size(); ++i)
  1637. {
  1638. ret = this->_t.del(vDelKey[i], data);
  1639. if(ret != TC_HashMapCompact::RT_OK)
  1640. {
  1641. vDelK.resize(i);
  1642. return ret;
  1643. }
  1644. }
  1645. return TC_HashMapCompact::RT_OK;
  1646. }
  1647. /**
  1648. * 淘汰数据, 根据Get时间淘汰
  1649. * 直到: 元素个数/chunks * 100 < radio,bCheckDirty 为true时,遇到脏数据则淘汰结束
  1650. * @param radio: 共享内存chunks使用比例 0< radio < 100
  1651. * @return int:
  1652. * TC_HashMapCompact::RT_READONLY: map只读
  1653. * TC_HashMapCompact::RT_OK:淘汰完毕
  1654. */
  1655. int erase(int radio, bool bCheckDirty = false)
  1656. {
  1657. while(true)
  1658. {
  1659. int ret;
  1660. TC_HashMapCompact::BlockData data;
  1661. {
  1662. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1663. ret = this->_t.erase(radio, data, bCheckDirty);
  1664. if(ret == TC_HashMapCompact::RT_OK || ret == TC_HashMapCompact::RT_READONLY)
  1665. {
  1666. return ret;
  1667. }
  1668. if(ret != TC_HashMapCompact::RT_ERASE_OK)
  1669. {
  1670. continue;
  1671. }
  1672. }
  1673. if(_todo_of)
  1674. {
  1675. K tk;
  1676. V tv;
  1677. tars::TarsInputStream<BufferReader> is;
  1678. is.setBuffer(data._key.c_str(), data._key.length());
  1679. tk.readFrom(is);
  1680. is.setBuffer(data._value.c_str(), data._value.length());
  1681. tv.readFrom(is);
  1682. typename ToDoFunctor::DataRecord stDataRecord;
  1683. stDataRecord._key = tk;
  1684. stDataRecord._value = tv;
  1685. stDataRecord._dirty = data._dirty;
  1686. stDataRecord._iSyncTime = data._synct;
  1687. stDataRecord._expiret = data._expiret;
  1688. stDataRecord._ver = data._ver;
  1689. _todo_of->erase(stDataRecord);
  1690. }
  1691. }
  1692. return TC_HashMapCompact::RT_OK;
  1693. }
  1694. /**
  1695. * 回写单条记录, 如果记录不存在, 则不做任何处理
  1696. * @param k
  1697. *
  1698. * @return int
  1699. * TC_HashMapCompact::RT_NO_DATA: 没有数据
  1700. * TC_HashMapCompact::RT_ONLY_KEY:只有Key
  1701. * TC_HashMapCompact::RT_OK:获取数据成功
  1702. * TC_HashMapCompact::RT_LOAD_DATA_ERR: load数据失败
  1703. * 其他返回值: 错误
  1704. */
  1705. int sync(const K& k)
  1706. {
  1707. V v;
  1708. uint32_t iSyncTime;
  1709. uint32_t iExpireTime;
  1710. uint8_t iVersion;
  1711. int ret = get(k, v, iSyncTime, iExpireTime, iVersion);
  1712. if(ret == TC_HashMapCompact::RT_OK)
  1713. {
  1714. bool bDirty = (checkDirty(k) == TC_HashMapCompact::RT_DIRTY_DATA);
  1715. if(_todo_of)
  1716. {
  1717. typename ToDoFunctor::DataRecord stDataRecord;
  1718. stDataRecord._key = k;
  1719. stDataRecord._value = v;
  1720. stDataRecord._dirty = bDirty;
  1721. stDataRecord._iSyncTime = iSyncTime;
  1722. stDataRecord._expiret = iExpireTime;
  1723. stDataRecord._ver = iVersion;
  1724. _todo_of->sync(stDataRecord);
  1725. }
  1726. }
  1727. return ret;
  1728. }
  1729. /**
  1730. * 将脏数据且一定时间没有回写的数据全部回写
  1731. * 数据回写时间与当前时间超过_pHead->_iSyncTime(setSyncTime)则需要回写
  1732. *
  1733. * map只读时仍然可以回写
  1734. *
  1735. * @param iNowTime: 回写到什么时间, 通常是当前时间
  1736. * @return int:
  1737. * TC_HashMapCompact::RT_OK: 回写完毕了
  1738. */
  1739. int sync(uint32_t iNowTime)
  1740. {
  1741. {
  1742. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1743. this->_t.sync();
  1744. }
  1745. while(true)
  1746. {
  1747. TC_HashMapCompact::BlockData data;
  1748. int ret;
  1749. {
  1750. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1751. ret = this->_t.sync(iNowTime, data);
  1752. if(ret == TC_HashMapCompact::RT_OK)
  1753. {
  1754. return ret;
  1755. }
  1756. if(ret != TC_HashMapCompact::RT_NEED_SYNC)
  1757. {
  1758. continue;
  1759. }
  1760. }
  1761. if(_todo_of)
  1762. {
  1763. K tk;
  1764. V tv;
  1765. tars::TarsInputStream<BufferReader> is;
  1766. is.setBuffer(data._key.c_str(), data._key.length());
  1767. tk.readFrom(is);
  1768. is.setBuffer(data._value.c_str(), data._value.length());
  1769. tv.readFrom(is);
  1770. typename ToDoFunctor::DataRecord stDataRecord;
  1771. stDataRecord._key = tk;
  1772. stDataRecord._value = tv;
  1773. stDataRecord._dirty = data._dirty;
  1774. stDataRecord._iSyncTime = data._synct;
  1775. stDataRecord._expiret = data._expiret;
  1776. stDataRecord._ver = data._ver;
  1777. _todo_of->sync(stDataRecord);
  1778. }
  1779. }
  1780. return TC_HashMapCompact::RT_OK;
  1781. }
  1782. /**
  1783. *将脏数据尾指针赋给回写尾指针
  1784. */
  1785. void sync()
  1786. {
  1787. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1788. this->_t.sync();
  1789. }
  1790. /**
  1791. * 将脏数据且一定时间没有回写的数据回写,只回写一个脏数据,目的是替代int sync(uint32_t iNowTime)
  1792. * 方法,把由业务控制每次回写数据量,使用时应该先调用void sync()
  1793. * 注意:c条件对象操作中, map是加锁的, 需要注意
  1794. *
  1795. * map只读时仍然可以回写
  1796. *
  1797. * @param iNowTime: 回写到什么时间, 通常是当前时间
  1798. * @param c: 条件仿函数: bool operator()(K v);
  1799. * @return int:
  1800. * TC_HashMapCompact::RT_OK: 回写完毕了
  1801. *
  1802. * 示例:
  1803. * p->sync();
  1804. * while(true) {
  1805. * int iRet = pthis->SyncOnce(tNow);
  1806. * if( iRet == TC_HashMapCompact::RT_OK )
  1807. * break;
  1808. * }
  1809. */
  1810. template<typename C>
  1811. int syncOnce(uint32_t iNowTime, C c)
  1812. {
  1813. TC_HashMapCompact::BlockData data;
  1814. K tk;
  1815. tars::TarsInputStream<BufferReader> is;
  1816. int ret;
  1817. {
  1818. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1819. ret = this->_t.sync(iNowTime, data);
  1820. if(ret == TC_HashMapCompact::RT_OK)
  1821. {
  1822. return ret;
  1823. }
  1824. if(ret != TC_HashMapCompact::RT_NEED_SYNC)
  1825. {
  1826. return ret;
  1827. }
  1828. if(_todo_of)
  1829. {
  1830. is.setBuffer(data._key.c_str(), data._key.length());
  1831. tk.readFrom(is);
  1832. if(!c(tk))
  1833. {
  1834. this->_t.setDirtyAfterSync(data._key);
  1835. return ret;
  1836. }
  1837. }
  1838. else
  1839. {
  1840. return ret;
  1841. }
  1842. }
  1843. V tv;
  1844. is.setBuffer(data._value.c_str(), data._value.length());
  1845. tv.readFrom(is);
  1846. typename ToDoFunctor::DataRecord stDataRecord;
  1847. stDataRecord._key = tk;
  1848. stDataRecord._value = tv;
  1849. stDataRecord._dirty = data._dirty;
  1850. stDataRecord._iSyncTime = data._synct;
  1851. stDataRecord._expiret = data._expiret;
  1852. stDataRecord._ver = data._ver;
  1853. _todo_of->sync(stDataRecord);
  1854. return ret;
  1855. }
  1856. /**
  1857. * 将脏数据且一定时间没有回写的数据回写,只回写一个脏数据,目的是替代int sync(uint32_t iNowTime)
  1858. * 方法,把由业务控制每次回写数据量,使用时应该先调用void sync()
  1859. *
  1860. * 数据回写时间与当前时间超过_pHead->_iSyncTime(setSyncTime)则需要回写
  1861. * map只读时仍然可以回写
  1862. *
  1863. * @param iNowTime: 回写到什么时间, 通常是当前时间
  1864. * @return int:
  1865. * TC_HashMapCompact::RT_OK: 回写完毕了
  1866. *
  1867. * 示例:
  1868. * p->sync();
  1869. * while(true) {
  1870. * int iRet = pthis->SyncOnce(tNow);
  1871. * if( iRet == TC_HashMapCompact::RT_OK )
  1872. * break;
  1873. * }
  1874. */
  1875. int syncOnce(uint32_t iNowTime)
  1876. {
  1877. TC_HashMapCompact::BlockData data;
  1878. int ret;
  1879. {
  1880. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1881. ret = this->_t.sync(iNowTime, data);
  1882. if(ret == TC_HashMapCompact::RT_OK)
  1883. {
  1884. return ret;
  1885. }
  1886. if(ret != TC_HashMapCompact::RT_NEED_SYNC)
  1887. {
  1888. return ret;
  1889. }
  1890. }
  1891. if(_todo_of)
  1892. {
  1893. K tk;
  1894. V tv;
  1895. tars::TarsInputStream<BufferReader> is;
  1896. is.setBuffer(data._key.c_str(), data._key.length());
  1897. tk.readFrom(is);
  1898. is.setBuffer(data._value.c_str(), data._value.length());
  1899. tv.readFrom(is);
  1900. typename ToDoFunctor::DataRecord stDataRecord;
  1901. stDataRecord._key = tk;
  1902. stDataRecord._value = tv;
  1903. stDataRecord._dirty = data._dirty;
  1904. stDataRecord._iSyncTime = data._synct;
  1905. stDataRecord._expiret = data._expiret;
  1906. stDataRecord._ver = data._ver;
  1907. _todo_of->sync(stDataRecord);
  1908. }
  1909. return ret;
  1910. }
  1911. /**
  1912. * 备份数据
  1913. * map只读时仍然可以备份
  1914. * 可以多个线程/进程备份数据,同时备份时bForceFromBegin设置为false效率更高
  1915. *
  1916. * @param bForceFromBegin: 是否强制重头开始备份, 通常为false
  1917. * @return int:
  1918. * TC_HashMapCompact::RT_OK: 备份OK了
  1919. */
  1920. int backup(bool bForceFromBegin = false)
  1921. {
  1922. {
  1923. //开始准备备份
  1924. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1925. this->_t.backup(bForceFromBegin);
  1926. }
  1927. while(true)
  1928. {
  1929. TC_HashMapCompact::BlockData data;
  1930. int ret;
  1931. {
  1932. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  1933. ret = this->_t.backup(data);
  1934. if(ret == TC_HashMapCompact::RT_OK)
  1935. {
  1936. return ret;
  1937. }
  1938. if(ret != TC_HashMapCompact::RT_NEED_BACKUP)
  1939. {
  1940. continue;
  1941. }
  1942. }
  1943. if(_todo_of)
  1944. {
  1945. K tk;
  1946. V tv;
  1947. tars::TarsInputStream<BufferReader> is;
  1948. is.setBuffer(data._key.c_str(), data._key.length());
  1949. tk.readFrom(is);
  1950. is.setBuffer(data._value.c_str(), data._value.length());
  1951. tv.readFrom(is);
  1952. typename ToDoFunctor::DataRecord stDataRecord;
  1953. stDataRecord._key = tk;
  1954. stDataRecord._value = tv;
  1955. stDataRecord._dirty = data._dirty;
  1956. stDataRecord._iSyncTime = data._synct;
  1957. stDataRecord._expiret = data._expiret;
  1958. stDataRecord._ver = data._ver;
  1959. _todo_of->backup(stDataRecord);
  1960. }
  1961. }
  1962. return TC_HashMapCompact::RT_OK;
  1963. }
  1964. /**
  1965. * 描述
  1966. *
  1967. * @return string
  1968. */
  1969. string desc() { return this->_t.desc(); }
  1970. ///////////////////////////////////////////////////////////////////////////////
  1971. /**
  1972. * 尾部
  1973. *
  1974. * @return lock_iterator
  1975. */
  1976. lock_iterator end()
  1977. {
  1978. JhmAutoLockPtr jlock;
  1979. return JhmLockIterator(this->_t.end(), jlock);
  1980. }
  1981. /**
  1982. * 根据Key查找数据
  1983. * @param k
  1984. */
  1985. lock_iterator find(const K& k)
  1986. {
  1987. tars::TarsOutputStream<BufferWriter> os;
  1988. k.writeTo(os);
  1989. string sk(os.getBuffer(), os.getLength());
  1990. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  1991. return JhmLockIterator(this->_t.find(sk), jlock);
  1992. }
  1993. /**
  1994. * block正序
  1995. *
  1996. * @return lock_iterator
  1997. */
  1998. lock_iterator begin()
  1999. {
  2000. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  2001. return JhmLockIterator(this->_t.begin(), jlock);
  2002. }
  2003. /**
  2004. * block逆序
  2005. *
  2006. * @return lock_iterator
  2007. */
  2008. lock_iterator rbegin()
  2009. {
  2010. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  2011. return JhmLockIterator(this->_t.rbegin(), jlock);
  2012. }
  2013. /**
  2014. * 以Set时间排序的迭代器
  2015. * 返回的迭代器++表示按照时间顺序:最近Set-->最久Set
  2016. *
  2017. * @return lock_iterator
  2018. */
  2019. lock_iterator beginSetTime()
  2020. {
  2021. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  2022. return JhmLockIterator(this->_t.beginSetTime(), jlock);
  2023. }
  2024. /**
  2025. * Set时间链逆序的迭代器
  2026. *
  2027. * 返回的迭代器++表示按照时间顺序:最久Set-->最近Set
  2028. *
  2029. * @return lock_iterator
  2030. */
  2031. lock_iterator rbeginSetTime()
  2032. {
  2033. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  2034. return JhmLockIterator(this->_t.rbeginSetTime(), jlock);
  2035. }
  2036. /**
  2037. * 以Get时间排序的迭代器
  2038. * 返回的迭代器++表示按照时间顺序:最近Get-->最久Get
  2039. *
  2040. * @return lock_iterator
  2041. */
  2042. lock_iterator beginGetTime()
  2043. {
  2044. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  2045. return JhmLockIterator(this->_t.beginGetTime(), jlock);
  2046. }
  2047. /**
  2048. * Get时间链逆序的迭代器
  2049. *
  2050. * 返回的迭代器++表示按照时间顺序:最久Get-->最近Get
  2051. *
  2052. * @return lock_iterator
  2053. */
  2054. lock_iterator rbeginGetTime()
  2055. {
  2056. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  2057. return JhmLockIterator(this->_t.rbeginGetTime(), jlock);
  2058. }
  2059. /**
  2060. * 获取脏链表尾部迭代器(最长时间没有Set的脏数据)
  2061. *
  2062. * 返回的迭代器++表示按照时间顺序:最近Set-->最久Set
  2063. * 可能存在干净数据
  2064. *
  2065. * @return lock_iterator
  2066. */
  2067. lock_iterator beginDirty()
  2068. {
  2069. JhmAutoLockPtr jlock(new JhmAutoLock(this->mutex()));
  2070. return JhmLockIterator(this->_t.beginDirty(), jlock);
  2071. }
  2072. /////////////////////////////////////////////////////////////////////////////////////////
  2073. // 以下是遍历map函数, 不需要对map加锁
  2074. /**
  2075. * 根据hash桶遍历
  2076. *
  2077. * @return hash_iterator
  2078. */
  2079. hash_iterator hashBegin()
  2080. {
  2081. JhmLockPtr jlock(new JhmLock(this->mutex()));
  2082. TC_LockT<typename LockPolicy::Mutex> lock(LockPolicy::mutex());
  2083. return JhmIterator(this->_t.hashBegin(), jlock);
  2084. }
  2085. /**
  2086. * 结束
  2087. *
  2088. * @return
  2089. */
  2090. hash_iterator hashEnd()
  2091. {
  2092. JhmLockPtr jlock;
  2093. return JhmIterator(this->_t.hashEnd(), jlock);
  2094. }
  2095. protected:
  2096. /**
  2097. * 删除数据的函数对象
  2098. */
  2099. ToDoFunctor *_todo_of;
  2100. };
  2101. }
  2102. #endif