开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
DBTable.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：8k
源码类别：

模拟服务器

开发平台：

C/C++

DBTable.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "DBTable.h"
  3. #ifndef WIN32
  4. #include <sys/stat.h>
  5. #else
  6. #include <direct.h>
  7. #endif
  8. #include <stdio.h>
  9. #include <string.h>
  10. #include <stdlib.h>
  13. ZDBTable::ZDBTable(const char *path, const char *name) {
  14. #ifdef WIN32
  15. getcwd(env_path, MAX_TABLE_NAME);
  16. #else
  17. #endif
  18. strcat(env_path, "\");
  19. strcat(env_path, path);
  20. #ifdef WIN32
  21. int ret = mkdir(env_path);
  22. #else
  23.         int ret = mkdir(env_path, 0);
  24. #endif
  25. if(!db_env_create(&dbenv, 0)) {
  26. dbenv->set_errpfx(dbenv, "index_db");
  27.             dbenv->set_lg_regionmax(dbenv, 512 * 1024);
  28.             dbenv->set_lg_max(dbenv, 16 * 1024 * 1024);
  29.             dbenv->set_lg_bsize(dbenv, 2 * 1024 * 1024);
  30.             dbenv->set_cachesize(dbenv, 0, 64 * 1024 * 1024, 1);
  32. if(!dbenv->open(dbenv, env_path, DB_CREATE | DB_INIT_LOG | DB_INIT_LOCK | DB_INIT_MPOOL | DB_INIT_TXN | DB_RECOVER | DB_THREAD | DB_PRIVATE, 0)) {
  33. ret = dbenv->set_flags(dbenv, DB_AUTO_COMMIT | DB_TXN_NOSYNC, 1);
  34. index_number = 0;
  35. strcpy(table_name, name);
  36. return; //成功了
  37. }
  38. dbenv->close(dbenv, 0);
  39. }
  40. dbenv = NULL;
  41. }
  43. ZDBTable::~ZDBTable() {
  44. if(dbenv) dbenv->close(dbenv, 0);
  45. }
  47. int ZDBTable::addIndex(GetIndexFunc func, bool isUnique) {
  48. if(!dbenv) return -1;
  49. if(index_number + 1 >= MAX_INDEX) return index_number;
  50. get_index_funcs[index_number] = func;
  51. is_index_unique[index_number] = isUnique;
  52. return index_number++;
  53. }
  55. bool ZDBTable::open() {
  56. if(!dbenv) return false;
  57. bStop = false;
  58. char index_table_name[MAX_TABLE_NAME];
  59.     int index;
  60. int ret;
  61. if(!db_create(&primary_db, dbenv, 0)) {
  62. if(!primary_db->open(primary_db, NULL, table_name, NULL, DB_BTREE, DB_CREATE| DB_AUTO_COMMIT | DB_THREAD, 0664)) { //打开主数据库
  63. for(index = 0; index < index_number; index++) {
  64. sprintf(index_table_name, "%s.%d", table_name, index);
  65. if(!db_create(&index_db[index], dbenv, 0)) {
  66. if(!is_index_unique[index]) {
  67. if(index_db[index]->set_flags(index_db[index], DB_DUP | DB_DUPSORT)) break;
  68. }
  69. if(index_db[index]->open(index_db[index], NULL, index_table_name, NULL, DB_BTREE, DB_CREATE | DB_AUTO_COMMIT | DB_THREAD, 0664)) break;
  70. if(ret = primary_db->associate(primary_db, NULL, index_db[index], get_index_funcs[index], DB_AUTO_COMMIT)) {
  71. index_db[index]->close(index_db[index], 0);
  72. break;
  73. }
  74. }
  75. else break;
  76. }
  77. if(index == index_number) return true; //成功了
  78. else while(--index) (index_db[index])->close(index_db[index], 0); //出错，关闭前面的索引表
  79. primary_db->close(primary_db, 0);
  80. }
  81. }
  82. return false;
  83. }
  85. void ZDBTable::close() {
  86. if(!dbenv) return;
  87. primary_db->close(primary_db, 0);
  88. for(int index = 0; index < index_number; index++) index_db[index]->close(index_db[index], 0);
  89. }
  91. bool ZDBTable::commit() {
  92. return true;
  93. }
  95. bool ZDBTable::add(const char *key_ptr, int key_size, const char *data_ptr, int data_size) {
  96. DBT data, key;
  97. memset(&key, 0, sizeof(DBT));
  98. memset(&data, 0, sizeof(DBT));
  99. key.data = (void *)key_ptr;
  100. key.size = key_size;
  101. data.data = (void *)data_ptr;
  102. data.size = data_size;
  104. int ret;
  105. int retry = 0;
  106. RETRY:
  107. ret = primary_db->put(primary_db, NULL, &key, &data, DB_AUTO_COMMIT);
  108. if(ret == DB_LOCK_DEADLOCK && ++retry < MAX_RETRY) {
  109. if(bStop) return false;
  110. goto RETRY;
  111. }
  112. if(ret) return false;
  113. else return true;
  114. }
  116. bool ZDBTable::remove(const char *key_ptr, int key_size, int index) {
  117. DBT data, key;
  118. memset(&key, 0, sizeof(DBT));
  119. memset(&data, 0, sizeof(DBT));
  120. key.data = (void *)key_ptr;
  121. key.size = key_size;
  122. int ret;
  123. int retry = 0;
  124. RETRY:
  125. ret = primary_db->del(primary_db, NULL, &key, DB_AUTO_COMMIT);
  126. if(ret == DB_LOCK_DEADLOCK && ++retry < MAX_RETRY) {
  127. if(bStop) return false;
  128. goto RETRY;
  129. }
  130. if(ret) return false;
  131. else return true;
  132. }
  134. ZCursor *ZDBTable::_search(bool bKey, const char *key_ptr, int key_size, int index) {
  135. DBT key, data, pkey;
  136. if(index < -1 || index >= index_number) return NULL;
  137. memset(&key, 0, sizeof(key));
  138. memset(&data, 0, sizeof(data));
  139. memset(&pkey, 0, sizeof(pkey));
  140. key.flags = DB_DBT_MALLOC;
  141. data.flags = DB_DBT_MALLOC;
  142. pkey.flags = DB_DBT_MALLOC;
  143. key.data = (void *)key_ptr;
  144. key.size = key_size;
  145. DBC *dbcp;
  146. if(!key_ptr || !key_size) { //没有设定索引值，要求遍历数据库
  147. if(index_db[index]->cursor(index_db[index], NULL, &dbcp, 0)) {
  148. return NULL;
  149. }
  150. if(dbcp->c_get(dbcp, &key, &data, DB_FIRST)) {
  151. dbcp->c_close(dbcp);
  152. return NULL;
  153. }
  154. }
  155. else {
  156. if(index == -1) { //主键搜索
  157. if(primary_db->get(primary_db, NULL, &key, &data, 0)) return NULL;
  158. }
  159. else if(is_index_unique[index]) { //没有重复索引
  160. if(bKey) {
  161. if(index_db[index]->pget(index_db[index], NULL, &key, &pkey, &data, 0)) return NULL;
  162. }
  163. else {
  164. if(index_db[index]->get(index_db[index], NULL, &key, &data, 0)) return NULL;
  165. }
  166. }
  167. else { //打开游标
  168. if(index_db[index]->cursor(index_db[index], NULL, &dbcp, 0)) {
  169. return NULL;
  170. }
  171. if(bKey) {
  172. if(dbcp->c_pget(dbcp, &key, &pkey, &data, DB_SET)) {
  173. dbcp->c_close(dbcp);
  174. return NULL;
  175. }
  176. }
  177. else {
  178. if(dbcp->c_get(dbcp, &key, &data, DB_SET)) {
  179. dbcp->c_close(dbcp);
  180. return NULL;
  181. }
  182. }
  183. }
  184. }
  185. ZCursor *result = new ZCursor;
  186. result->index = 0;
  187. result->dbcp = dbcp;
  188. if (bKey) 
  189.     {
  190. result->data = (char *)pkey.data;
  191. result->size = pkey.size;
  193.         pkey.data    = NULL;
  194. }
  195. else 
  196.     {
  197. result->data = (char *)data.data;
  198. result->size = data.size;
  200.         data.data    = NULL;
  201. }
  203.     if (!key_ptr || !key_size) 
  204.     {
  205. result->key = (char *)key.data;
  206. result->key_size = key.size;
  207. result->bTravel = true;
  209.         key.data = NULL;
  210. }
  211. else 
  212.     {
  213.         result->bTravel = false;
  214.     }
  216.     if(pkey.data) free(pkey.data);
  217. if(data.data) free(data.data);
  219. return result;
  220. }
  222. bool ZDBTable::_next(bool bKey, ZCursor *cursor) {
  223. DBT key, data, pkey;
  224. memset(&key, 0, sizeof(key));
  225. memset(&data, 0, sizeof(data));
  226. memset(&pkey, 0, sizeof(pkey));
  227. key.flags = DB_DBT_MALLOC;
  228. data.flags = DB_DBT_MALLOC;
  229. pkey.flags = DB_DBT_MALLOC;
  230. if(!cursor || !cursor->dbcp) return false;
  231. free(cursor->data);
  232. cursor->index++;
  233. if(cursor->bTravel) {
  234. free(cursor->key);
  235. if(cursor->dbcp->c_get(cursor->dbcp, &key, &data, DB_NEXT)) {
  236. cursor->dbcp->c_close(cursor->dbcp);
  237. delete cursor;
  238. return false;
  239. }
  240. }
  241. else {
  242. if(bKey) {
  243. if(cursor->dbcp->c_pget(cursor->dbcp, &key, &pkey, &data, DB_NEXT_DUP)) {
  244. cursor->dbcp->c_close(cursor->dbcp);
  245. delete cursor;
  246. return false;
  247. }
  248. }
  249. else {
  250. if(cursor->dbcp->c_get(cursor->dbcp, &key, &data, DB_NEXT_DUP)) {
  251. cursor->dbcp->c_close(cursor->dbcp);
  252. delete cursor;
  253. return false;
  254. }
  255. }
  256. }
  257. if (bKey) 
  258.     {
  259. cursor->data = (char *)pkey.data;
  260. cursor->size = pkey.size;
  262.         pkey.data = NULL;
  263. }
  264. else {
  265. cursor->data = (char *)data.data;
  266. cursor->size = data.size;
  268.         data.data = NULL;
  269. }
  271.     if (cursor->bTravel) 
  272.     {
  273. cursor->key = (char *)key.data;
  274. cursor->key_size = key.size;
  276.         key.data = NULL;
  277. }
  279. if (key.data) free(key.data);
  280. if (pkey.data) free(pkey.data);
  281. if (data.data) free(data.data);
  283.     return true;
  284. }
  286. void ZDBTable::removeLog() { //清除日志文件
  287. int ret;
  288. char **begin, **list;
  289. if((ret = dbenv->txn_checkpoint(dbenv, 0, 0, 0)) != 0) return;
  291. if((ret = dbenv->log_archive(dbenv, &list, DB_ARCH_ABS)) != 0) {
  292. return ;
  293. }
  294. if (list != NULL) {
  295. for(begin = list; *list != NULL; ++list)
  296. if((ret = ::remove(*list)) != 0) {
  297. return;
  298. }
  299. free(begin);
  300. }
  301. }