log_persistent_manager.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779
  1. #include "log_persistent_manager.h"
  2. #include "log_producer_manager.h"
  3. #include "inner_log.h"
  4. #include "log_builder.h"
  5. #include "sds.h"
  6. #ifndef _WIN32
  7. #include <sys/types.h>
  8. #include <dirent.h>
  9. #include <sys/stat.h>
  10. #endif //NOT _WIN32
  11. #define DB_TABLE_NAME "RvcLogTable"
  12. #if defined(_MSC_VER)
  13. #define DB_PATH "RVC.LogSdk\\"
  14. #else
  15. #define DB_PATH "RVC.LogSdk/"
  16. #endif //_MSC_VER
  17. #define SLASH '/'
  18. #define BACK_SLASH '\\'
  19. #define MB_SLASH_STR "/"
  20. #define MB_BACK_SLASH_STR "\\"
  21. #define W_SLASH_STR L"/"
  22. #define W_BACK_SLASH_STR L"\\"
  23. #if defined(UNICODE) || defined(_UNICODE)
  24. #define SLASH_STR W_SLASH_STR
  25. #define BACK_SLASH_STR W_BACK_SLASH_STR
  26. #else
  27. #define SLASH_STR MB_SLASH_STR
  28. #define BACK_SLASH_STR MB_BACK_SLASH_STR
  29. #endif
  30. #ifdef _WIN32
  31. #define SPLIT_SLASH BACK_SLASH
  32. #define SPLIT_SLASH_STR BACK_SLASH_STR
  33. #define MB_SPLIT_SLASH_STR MB_BACK_SLASH_STR
  34. #define W_SPLIT_SLASH_STR W_BACK_SLASH_STR
  35. #define LINE_BREAK_STR "\r\n"
  36. #else
  37. #define SPLIT_SLASH SLASH
  38. #define SPLIT_SLASH_STR SLASH_STR
  39. #define MB_SPLIT_SLASH_STR MB_SLASH_STR
  40. #define W_SPLIT_SLASH_STR W_SLASH_STR
  41. #define LINE_BREAK_STR "\n"
  42. #endif
  43. #ifndef _WIN32
  44. #ifndef CLOCK_MONOTONIC_RAW
  45. #define CLOCK_MONOTONIC_RAW 4
  46. #endif
  47. uint32_t GetTickCount(void)
  48. {
  49. uint32_t ticks = 0;
  50. struct timespec ts;
  51. if (!clock_gettime(CLOCK_MONOTONIC_RAW, &ts))
  52. ticks = (ts.tv_sec * 1000) + (ts.tv_nsec / 1000000);
  53. return ticks;
  54. }
  55. #endif // _WIN32
  56. static void _copy_config_string(const char* value, sds* src_value)
  57. {
  58. int strLen;
  59. if (value == NULL || src_value == NULL)
  60. {
  61. return;
  62. }
  63. strLen = strlen(value);
  64. if (*src_value == NULL)
  65. {
  66. *src_value = sdsnewEmpty(strLen);
  67. }
  68. else {
  69. sdsfree(*src_value);
  70. }
  71. *src_value = sdscpylen(*src_value, value, strLen);
  72. }
  73. static log_db_manager* create_db_manager(log_persistent_manager* manager, char * modular) {
  74. int ret = 0;
  75. log_db_manager* guard_db_manager = NULL;
  76. char tmpFilePath[MAX_PATH] = { 0 };
  77. int len = strlen(manager->config->persistentFilePath);
  78. strcpy(tmpFilePath, manager->config->persistentFilePath);
  79. if (manager->config->persistentFilePath[len - 1] == '\\' || manager->config->persistentFilePath[len - 1] == '/') {
  80. strcat(tmpFilePath, DB_PATH);
  81. }
  82. else {
  83. strcat(tmpFilePath, SPLIT_SLASH_STR);
  84. strcat(tmpFilePath, DB_PATH);
  85. }
  86. guard_db_manager = create_log_db(manager->config, tmpFilePath, modular, DB_TABLE_NAME);
  87. if (guard_db_manager != NULL) {
  88. ret = open_db(guard_db_manager);
  89. if (ret <= 0) {
  90. destroy_log_db(guard_db_manager);
  91. guard_db_manager = NULL;
  92. }
  93. }
  94. return guard_db_manager;
  95. }
  96. static int log_persistent_manager_update_guard_log_status(log_persistent_manager* manager,
  97. char* modular, int send_count, char uuid[MAX_LOG_COUNT][MAX_UUID_LEN], log_db_status_e status) {
  98. log_db_manager* guard_db_manager = NULL;
  99. int i, op_count = 0;
  100. guard_db_manager = create_db_manager(manager, modular);
  101. if (guard_db_manager != NULL) {
  102. db_transaction_begin(guard_db_manager);
  103. for (i = 0; i < send_count; i++) {
  104. op_count += db_update_status(guard_db_manager, uuid[i], status);
  105. }
  106. db_transaction_commit(guard_db_manager);
  107. destroy_log_db(guard_db_manager);
  108. }
  109. else {
  110. return FALSE;
  111. }
  112. return op_count;
  113. }
  114. static int log_persistent_manager_delete_guard_log(log_persistent_manager* manager,
  115. char* modular, int send_count, char uuid[MAX_LOG_COUNT][MAX_UUID_LEN]) {
  116. log_db_manager* guard_db_manager = NULL;
  117. int ret, i, op_count = 0;
  118. guard_db_manager = create_db_manager(manager, modular);
  119. if (guard_db_manager != NULL) {
  120. db_transaction_begin(guard_db_manager);
  121. for (i = 0; i < send_count; i++) {
  122. op_count += db_delete_one(guard_db_manager, uuid[i]);
  123. }
  124. db_transaction_commit(guard_db_manager);
  125. destroy_log_db(guard_db_manager);
  126. }
  127. else {
  128. return FALSE;
  129. }
  130. return op_count;
  131. }
  132. void on_log_persistent_manager_send_done_uuid(
  133. log_producer_result result,
  134. int log_bytes,
  135. int compressed_bytes,
  136. const char* req_id,
  137. const char* error_message,
  138. const unsigned char* raw_buffer,
  139. void* persistent_manager,
  140. int send_count,
  141. char uuid[MAX_LOG_COUNT][MAX_UUID_LEN],
  142. char* modular)
  143. {
  144. int i;
  145. char tmpFileName[MAX_PATH] = { 0 };
  146. log_persistent_manager* manager = (log_persistent_manager*)persistent_manager;
  147. if (manager == NULL)
  148. {
  149. return;
  150. }
  151. strcat(tmpFileName, "RvcLogSdk"/*manager->config->skyeyeEntityName*/);
  152. strcat(tmpFileName, ".db");
  153. /*如果异常,则将条目恢复成待发送状态*/
  154. if (result != LOG_PRODUCER_OK && result != LOG_PRODUCER_DROP_ERROR && result != LOG_PRODUCER_INVALID)
  155. {
  156. CS_ENTER(manager->lock);
  157. log_persistent_manager_updata_log_status(manager, modular, send_count, uuid, LOG_DB_STATUS_WAITING);
  158. CS_LEAVE(manager->lock);
  159. return;
  160. }
  161. CS_ENTER(manager->lock);
  162. if (strcmp(modular,tmpFileName)==0) {
  163. db_transaction_begin(manager->db_manager);
  164. for (i = 0; i < send_count; i++) {
  165. log_persistent_manager_delete_log(manager, uuid[i]);
  166. }
  167. db_transaction_commit(manager->db_manager);
  168. }
  169. else {
  170. log_persistent_manager_delete_guard_log(manager, modular, send_count, uuid);
  171. }
  172. CS_LEAVE(manager->lock);
  173. }
  174. static int log_persistent_manager_init(log_persistent_manager* manager, log_producer_config* config)
  175. {
  176. int ret;
  177. int len;
  178. char tmpFilePath[MAX_PATH] = {0};
  179. char tmpFileName[MAX_PATH] = { 0 };
  180. if (config->persistentFilePath == NULL) {
  181. return -1;
  182. }
  183. memset(manager, 0, sizeof(log_persistent_manager));
  184. manager->config = config;
  185. manager->lock = CreateCriticalSection();
  186. len = strlen(config->persistentFilePath);
  187. strcpy(tmpFilePath, config->persistentFilePath);
  188. if (config->persistentFilePath[len - 1] == '\\' || config->persistentFilePath[len - 1] == '/') {
  189. strcat(tmpFilePath, DB_PATH);
  190. }
  191. else {
  192. strcat(tmpFilePath, SPLIT_SLASH_STR);
  193. strcat(tmpFilePath, DB_PATH);
  194. }
  195. strcat(tmpFileName, "RvcLogSdk");//config->skyeyeEntityName);
  196. strcat(tmpFileName, ".db");
  197. manager->db_manager = create_log_db(config, tmpFilePath, tmpFileName, DB_TABLE_NAME);
  198. _copy_config_string(tmpFileName, &manager->db_name);
  199. if (manager->db_manager == NULL) {
  200. return -1;
  201. }
  202. ret = open_db(manager->db_manager);
  203. if (ret <= 0) {
  204. return -1;
  205. }
  206. if (config->openDB == 0)
  207. close_logdb(manager->db_manager);
  208. return 0;
  209. }
  210. static void log_persistent_manager_clear(log_persistent_manager* manager)
  211. {
  212. ReleaseCriticalSection(manager->lock);
  213. close_logdb(manager->db_manager);
  214. }
  215. log_persistent_manager* create_log_persistent_manager(log_producer_config* config)
  216. {
  217. int rst;
  218. log_persistent_manager* manager;
  219. if (!log_producer_persistent_config_is_enabled(config))
  220. {
  221. return NULL;
  222. }
  223. manager = (log_persistent_manager*)malloc(sizeof(log_persistent_manager));
  224. rst = log_persistent_manager_init(manager, config);
  225. if (rst < 0) {
  226. free(manager);
  227. return NULL;
  228. }
  229. return manager;
  230. }
  231. void destroy_log_persistent_manager(log_persistent_manager* manager)
  232. {
  233. if (manager == NULL)
  234. {
  235. return;
  236. }
  237. log_persistent_manager_clear(manager);
  238. sdsfree(manager->db_name);
  239. free(manager);
  240. manager = NULL;
  241. }
  242. int log_persistent_manager_save_log_group(log_persistent_manager* manager, log_group_builder* builder)
  243. {
  244. unsigned long end_time;
  245. unsigned long start_time = GetTickCount();
  246. int rst = db_insert_group(manager->db_manager, builder);
  247. end_time = GetTickCount();
  248. //aos_debug_log((LB, "write db log cust %d", end_time - start_time));
  249. if (rst <= 0) {
  250. aos_error_log((LB, "write db log failed, rst %d",
  251. rst));
  252. return LOG_PRODUCER_PERSISTENT_ERROR;
  253. }
  254. else {
  255. return LOG_PRODUCER_OK;
  256. }
  257. }
  258. int log_persistent_manager_save_log(log_persistent_manager* manager, serialize_buf* buf)
  259. {
  260. //unsigned long end_time;
  261. //unsigned long start_time = GetTickCount();
  262. int rst = db_insert_one(manager->db_manager, buf);
  263. //end_time = GetTickCount();
  264. //aos_debug_log((LB, "write db log cust %d", end_time - start_time));
  265. if (rst <= 0) {
  266. aos_error_log((LB, "write db log failed, rst %d",
  267. rst));
  268. return LOG_PRODUCER_PERSISTENT_ERROR;
  269. }
  270. else {
  271. return LOG_PRODUCER_OK;
  272. }
  273. }
  274. static log_group_builder* log_persistent_manager_get_guard_log(log_persistent_manager* manager)
  275. {
  276. #if defined(_MSC_VER)
  277. static int read_file_index = 0;
  278. char srcFilePath[MAX_PATH] = { 0 };
  279. WIN32_FIND_DATA FindFileData;
  280. HANDLE hFind;
  281. BOOL fFinished = FALSE;
  282. log_db_manager* guard_db_manager = NULL;
  283. int i;
  284. char tmpFilePath[MAX_PATH] = { 0 };
  285. int len = strlen(manager->config->persistentFilePath);
  286. strcpy(tmpFilePath, manager->config->persistentFilePath);
  287. if (manager->config->persistentFilePath[len - 1] == '\\' || manager->config->persistentFilePath[len - 1] == '/') {
  288. strcat(tmpFilePath, DB_PATH);
  289. } else {
  290. strcat(tmpFilePath, "\\");
  291. strcat(tmpFilePath, DB_PATH);
  292. }
  293. sprintf_s(srcFilePath, MAX_PATH, "%s*.db", tmpFilePath);
  294. hFind = FindFirstFile(srcFilePath, &FindFileData);
  295. if (INVALID_HANDLE_VALUE != hFind) {
  296. for (i = 0; !fFinished; i++) {
  297. if ((FILE_ATTRIBUTE_DIRECTORY & FindFileData.dwFileAttributes) || (i < read_file_index)) {
  298. if (!FindNextFile(hFind, &FindFileData)) {
  299. if (GetLastError() == ERROR_NO_MORE_FILES) {
  300. read_file_index = 0;
  301. fFinished = TRUE;
  302. } else {
  303. break;
  304. }
  305. }
  306. continue;
  307. }
  308. aos_debug_log((LB, "guard read log db name %s", FindFileData.cFileName));
  309. if (strcmp(manager->db_name, FindFileData.cFileName) != 0) {
  310. guard_db_manager = create_db_manager(manager, FindFileData.cFileName);
  311. if (guard_db_manager != NULL) {
  312. log_group_builder* builder = db_read_table_last_logs(guard_db_manager, manager->config->logCountPerPackage);
  313. destroy_log_db(guard_db_manager);
  314. if (builder != NULL) {
  315. read_file_index++;
  316. FindClose(hFind);
  317. return builder;
  318. }
  319. }
  320. } else {
  321. log_group_builder* builder = db_read_table_last_logs(manager->db_manager, manager->config->logCountPerPackage);
  322. if (builder != NULL) {
  323. read_file_index++;
  324. FindClose(hFind);
  325. return builder;
  326. }
  327. }
  328. if (!FindNextFile(hFind, &FindFileData)) {
  329. if (GetLastError() == ERROR_NO_MORE_FILES) {
  330. read_file_index = 0;
  331. fFinished = TRUE;
  332. } else {
  333. break;
  334. }
  335. }
  336. }
  337. FindClose(hFind);
  338. }
  339. return NULL;
  340. #else
  341. static int read_file_index = 0;
  342. DIR* d = NULL;
  343. struct dirent* dp = NULL;
  344. struct stat st;
  345. char srcFilePath[MAX_PATH] = { 0 };
  346. BOOL fFinished = FALSE;
  347. log_db_manager* guard_db_manager = NULL;
  348. int ret, i;
  349. char tmpFilePath[MAX_PATH] = { 0 };
  350. const int len = strlen(manager->config->persistentFilePath);
  351. strcpy(tmpFilePath, manager->config->persistentFilePath);
  352. if (manager->config->persistentFilePath[len - 1] == '\\' || manager->config->persistentFilePath[len - 1] == '/') {
  353. strcat(tmpFilePath, DB_PATH);
  354. tmpFilePath[strlen(tmpFilePath) - 1] = '\0';
  355. } else {
  356. strcat(tmpFilePath, SPLIT_SLASH_STR);
  357. strcat(tmpFilePath, DB_PATH);
  358. tmpFilePath[strlen(tmpFilePath) - 1] = '\0';
  359. }
  360. if (stat(tmpFilePath, &st) < 0 || !S_ISDIR(st.st_mode)) {
  361. return NULL;
  362. }
  363. if (!(d = opendir(tmpFilePath))) {
  364. return NULL;
  365. }
  366. while ((dp = readdir(d)) != NULL) {
  367. if ((!strncmp(dp->d_name, ".", 1)) || (!strncmp(dp->d_name, "..", 2))) {
  368. continue;
  369. }
  370. int isdir = 0;
  371. struct stat st1;
  372. sprintf(srcFilePath, "%s/%s", tmpFilePath, dp->d_name);
  373. if (stat(srcFilePath, &st1) < 0) {
  374. continue;
  375. }
  376. isdir = !!(S_ISDIR(st1.st_mode));
  377. if (isdir) {
  378. continue;
  379. }
  380. if (strstr(dp->d_name, ".db") == NULL) {
  381. continue;
  382. }
  383. if (i < read_file_index) {
  384. i++;
  385. continue;
  386. }
  387. aos_debug_log((LB, "guard read log db name %s", dp->d_name));
  388. if (strcmp(manager->db_name, dp->d_name) != 0) {
  389. guard_db_manager = create_db_manager(manager, dp->d_name);
  390. if (guard_db_manager != NULL) {
  391. log_group_builder* builder = db_read_table_last_logs(guard_db_manager, manager->config->logCountPerPackage);
  392. destroy_log_db(guard_db_manager);
  393. if (builder != NULL) {
  394. read_file_index++;
  395. return builder;
  396. }
  397. }
  398. } else {
  399. log_group_builder* builder = db_read_table_last_logs(manager->db_manager, manager->config->logCountPerPackage);
  400. if (builder != NULL) {
  401. read_file_index++;
  402. return builder;
  403. }
  404. }
  405. }
  406. closedir(d);
  407. read_file_index = 0;
  408. return NULL;
  409. #endif //_MSC_VER
  410. }
  411. log_group_builder* log_persistent_manager_get_log(log_persistent_manager* manager) {
  412. unsigned long end_time;
  413. unsigned long start_time = GetTickCount();
  414. log_group_builder* builder = db_read_table_last_logs(manager->db_manager, manager->config->logCountPerPackage);
  415. end_time = GetTickCount();
  416. //aos_debug_log((LB, "read db log cust %d", end_time - start_time));
  417. if (builder == NULL) {
  418. aos_error_log((LB, "get db log failed"));
  419. //add for Guarder upload
  420. if (manager->config->enableGuarder) {
  421. return log_persistent_manager_get_guard_log(manager);
  422. }
  423. }
  424. return builder;
  425. }
  426. int log_persistent_manager_updata_log_status(log_persistent_manager* manager, char* modular, int count, char uuid[][MAX_UUID_LEN], log_db_status_e status) {
  427. int i = 0, op_count = 0;
  428. unsigned long end_time;
  429. unsigned long start_time = GetTickCount();
  430. if (strcmp(manager->db_name, modular) == 0) {
  431. db_transaction_begin(manager->db_manager);
  432. for (i = 0; i < count; i++) {
  433. op_count += db_update_status(manager->db_manager, uuid[i], status);
  434. }
  435. db_transaction_commit(manager->db_manager);
  436. end_time = GetTickCount();
  437. //aos_debug_log((LB, "updata db log status cust %d, op count %d", end_time - start_time, op_count));
  438. if (op_count <= 0) {
  439. aos_error_log((LB, "updata_log_status failed"));
  440. return LOG_PRODUCER_PERSISTENT_ERROR;
  441. }
  442. else {
  443. return LOG_PRODUCER_OK;
  444. }
  445. }
  446. else {
  447. op_count = log_persistent_manager_update_guard_log_status(manager, modular, count, uuid, status);
  448. if (op_count <= 0) {
  449. aos_error_log((LB, "updata_log_status failed"));
  450. return LOG_PRODUCER_PERSISTENT_ERROR;
  451. }
  452. else {
  453. return LOG_PRODUCER_OK;
  454. }
  455. }
  456. }
  457. int log_persistent_manager_delete_log(log_persistent_manager* manager, char* uuid) {
  458. unsigned long end_time;
  459. unsigned long start_time = GetTickCount();
  460. int rst = db_delete_one(manager->db_manager, uuid);
  461. end_time = GetTickCount();
  462. //aos_debug_log((LB, "delete db log cust %d", end_time - start_time));
  463. if (rst <= 0) {
  464. aos_error_log((LB, "delete_log failed, rst %d",
  465. rst));
  466. return LOG_PRODUCER_PERSISTENT_ERROR;
  467. }
  468. else {
  469. return LOG_PRODUCER_OK;
  470. }
  471. }
  472. /*将数据恢复成为待上传状态*/
  473. int log_persistent_manager_recover_inner(log_persistent_manager* manager, log_db_manager* log_db_manager) {
  474. unsigned long end_time;
  475. unsigned long start_time = GetTickCount();
  476. int rst = db_update_status(log_db_manager, NULL, LOG_DB_STATUS_WAITING);
  477. end_time = GetTickCount();
  478. aos_debug_log((LB, "recover db log cust %d", end_time - start_time));
  479. if (rst <= 0) {
  480. aos_error_log((LB, "persistent_manager recover failed, rst %d",
  481. rst));
  482. return LOG_PRODUCER_PERSISTENT_ERROR;
  483. }
  484. else {
  485. return LOG_PRODUCER_OK;
  486. }
  487. }
  488. int log_persistent_manager_recover(log_persistent_manager* manager) {
  489. #if defined(_MSC_VER)
  490. char srcFilePath[MAX_PATH] = { 0 };
  491. WIN32_FIND_DATA FindFileData;
  492. HANDLE hFind;
  493. BOOL fFinished = FALSE;
  494. log_db_manager* guard_db_manager = NULL;
  495. int i;
  496. char tmpFilePath[MAX_PATH] = { 0 };
  497. int len = strlen(manager->config->persistentFilePath);
  498. strcpy(tmpFilePath, manager->config->persistentFilePath);
  499. if (manager->config->persistentFilePath[len - 1] == '\\' || manager->config->persistentFilePath[len - 1] == '/') {
  500. strcat(tmpFilePath, DB_PATH);
  501. } else {
  502. strcat(tmpFilePath, "\\");
  503. strcat(tmpFilePath, DB_PATH);
  504. }
  505. sprintf_s(srcFilePath, MAX_PATH, "%s*.db", tmpFilePath);
  506. hFind = FindFirstFile(srcFilePath, &FindFileData);
  507. if (INVALID_HANDLE_VALUE != hFind) {
  508. for (i = 0; !fFinished; i++) {
  509. if ((FILE_ATTRIBUTE_DIRECTORY & FindFileData.dwFileAttributes)) {
  510. if (!FindNextFile(hFind, &FindFileData)) {
  511. if (GetLastError() == ERROR_NO_MORE_FILES) {
  512. fFinished = TRUE;
  513. } else {
  514. break;
  515. }
  516. }
  517. continue;
  518. }
  519. aos_debug_log((LB, "recover guard log db name %s", FindFileData.cFileName));
  520. //clean
  521. //log_persistent_manager_clean(manager, FindFileData.cFileName);
  522. //recover
  523. if (strcmp(manager->db_name, FindFileData.cFileName) != 0) {
  524. guard_db_manager = create_db_manager(manager, FindFileData.cFileName);
  525. if (guard_db_manager != NULL) {
  526. log_persistent_manager_recover_inner(manager, guard_db_manager);
  527. destroy_log_db(guard_db_manager);
  528. }
  529. } else {
  530. log_persistent_manager_recover_inner(manager, manager->db_manager);
  531. }
  532. if (!FindNextFile(hFind, &FindFileData)) {
  533. if (GetLastError() == ERROR_NO_MORE_FILES) {
  534. fFinished = TRUE;
  535. } else {
  536. break;
  537. }
  538. }
  539. }
  540. FindClose(hFind);
  541. }
  542. return LOG_PRODUCER_OK;
  543. #else
  544. DIR* d = NULL;
  545. struct dirent* dp = NULL;
  546. struct stat st;
  547. char srcFilePath[MAX_PATH] = { 0 };
  548. BOOL fFinished = FALSE;
  549. log_db_manager* guard_db_manager = NULL;
  550. int ret, i;
  551. char tmpFilePath[MAX_PATH] = { 0 };
  552. int len = strlen(manager->config->persistentFilePath);
  553. strcpy(tmpFilePath, manager->config->persistentFilePath);
  554. if (manager->config->persistentFilePath[len - 1] == '\\' || manager->config->persistentFilePath[len - 1] == '/') {
  555. strcat(tmpFilePath, DB_PATH);
  556. tmpFilePath[strlen(tmpFilePath) - 1] = '\0';
  557. } else {
  558. strcat(tmpFilePath, SPLIT_SLASH_STR);
  559. strcat(tmpFilePath, DB_PATH);
  560. tmpFilePath[strlen(tmpFilePath) - 1] = '\0';
  561. }
  562. aos_debug_log((LB, "recover log db path %s", tmpFilePath));
  563. if (stat(tmpFilePath, &st) < 0 || !S_ISDIR(st.st_mode)) {
  564. aos_warn_log((LB, "stat %s failed: %d", tmpFilePath, errno));
  565. return NULL;
  566. }
  567. if (!(d = opendir(tmpFilePath))) {
  568. aos_warn_log((LB, "opendir %s failed: %d", tmpFilePath, errno));
  569. return NULL;
  570. }
  571. while ((dp = readdir(d)) != NULL) {
  572. if ((!strncmp(dp->d_name, ".", 1)) || (!strncmp(dp->d_name, "..", 2))) {
  573. continue;
  574. }
  575. int isdir = 0;
  576. struct stat st1;
  577. sprintf(srcFilePath, "%s/%s", tmpFilePath, dp->d_name);
  578. if (stat(srcFilePath, &st1) < 0) {
  579. aos_warn_log((LB, "stat %s failed: %d", srcFilePath, errno));
  580. continue;
  581. }
  582. isdir = !!(S_ISDIR(st1.st_mode));
  583. if (isdir) {
  584. continue;
  585. }
  586. if (strstr(dp->d_name, ".db") == NULL) {
  587. continue;
  588. }
  589. aos_debug_log((LB, "recover log db name %s", dp->d_name));
  590. //clean
  591. //log_persistent_manager_clean(manager, FindFileData.cFileName);
  592. //recover
  593. if (strcmp(manager->db_name, dp->d_name) != 0) {
  594. guard_db_manager = create_db_manager(manager, dp->d_name);
  595. if (guard_db_manager != NULL) {
  596. log_persistent_manager_recover_inner(manager, guard_db_manager);
  597. destroy_log_db(guard_db_manager);
  598. }
  599. } else {
  600. log_persistent_manager_recover_inner(manager, manager->db_manager);
  601. }
  602. }
  603. closedir(d);
  604. return LOG_PRODUCER_OK;
  605. #endif //_MSC_VER
  606. }
  607. int log_persistent_manager_clean(log_persistent_manager* manager, char* modular) {
  608. if (strcmp(manager->db_name, modular) == 0) {
  609. int rst = 1;
  610. unsigned long end_time;
  611. unsigned long start_time = GetTickCount();
  612. int count = db_get_count(manager->db_manager);
  613. if (count > manager->config->maxPersistentLogCount) {//超过最大值,然后清理 2*10000
  614. rst = db_delete_old_logs(manager->db_manager, count - manager->config->maxPersistentLogCount);
  615. }
  616. db_vacuum(manager->db_manager);
  617. end_time = GetTickCount();
  618. aos_debug_log((LB, "clean db log cust %d", end_time - start_time));
  619. if (rst <= 0) {
  620. aos_error_log((LB, "persistent_manager clean failed, rst %d",
  621. rst));
  622. return LOG_PRODUCER_PERSISTENT_ERROR;
  623. }
  624. else {
  625. return LOG_PRODUCER_OK;
  626. }
  627. }
  628. else {
  629. log_db_manager* guard_db_manager = create_db_manager(manager, modular);
  630. if (guard_db_manager != NULL) {
  631. int rst = 1;
  632. unsigned long end_time;
  633. unsigned long start_time = GetTickCount();
  634. int count = db_get_count(guard_db_manager);
  635. if (count > manager->config->maxPersistentLogCount) {
  636. rst = db_delete_old_logs(guard_db_manager, count - manager->config->maxPersistentLogCount);
  637. }
  638. db_vacuum(guard_db_manager);
  639. destroy_log_db(guard_db_manager);
  640. end_time = GetTickCount();
  641. aos_debug_log((LB, "clean guard db log cust %d, modular:%s", end_time - start_time, modular));
  642. if (rst <= 0) {
  643. aos_error_log((LB, "persistent_manager guard clean failed, rst %d",
  644. rst));
  645. return LOG_PRODUCER_PERSISTENT_ERROR;
  646. }
  647. else {
  648. return LOG_PRODUCER_OK;
  649. }
  650. }
  651. }
  652. return LOG_PRODUCER_OK;
  653. }
  654. int log_persistent_manager_transaction_begin(log_persistent_manager* manager) {
  655. int rst = 1;
  656. unsigned long end_time;
  657. unsigned long start_time = GetTickCount();
  658. rst = db_transaction_begin(manager->db_manager);
  659. end_time = GetTickCount();
  660. aos_debug_log((LB, "begin db log cust %d", end_time - start_time));
  661. if (rst <= 0) {
  662. aos_error_log((LB, "persistent_manager begin db failed, rst %d",
  663. rst));
  664. return LOG_PRODUCER_PERSISTENT_ERROR;
  665. }
  666. else {
  667. return LOG_PRODUCER_OK;
  668. }
  669. }
  670. int log_persistent_manager_transaction_commit(log_persistent_manager* manager) {
  671. int rst = 1;
  672. unsigned long end_time;
  673. unsigned long start_time = GetTickCount();
  674. rst = db_transaction_commit(manager->db_manager);
  675. end_time = GetTickCount();
  676. aos_debug_log((LB, "commit db log cust %d", end_time - start_time));
  677. if (rst <= 0) {
  678. aos_error_log((LB, "persistent_manager commit db failed, rst %d",
  679. rst));
  680. return LOG_PRODUCER_PERSISTENT_ERROR;
  681. }
  682. else {
  683. return LOG_PRODUCER_OK;
  684. }
  685. }