sqlite_aclk.c 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "sqlite_functions.h"
  3. #include "sqlite_aclk.h"
  4. #include "sqlite_aclk_chart.h"
  5. #include "sqlite_aclk_node.h"
  6. #ifdef ENABLE_ACLK
  7. #include "../../aclk/aclk.h"
  8. #endif
  9. void sanity_check(void) {
  10. // make sure the compiler will stop on misconfigurations
  11. BUILD_BUG_ON(WORKER_UTILIZATION_MAX_JOB_TYPES < ACLK_MAX_ENUMERATIONS_DEFINED);
  12. }
  13. const char *aclk_sync_config[] = {
  14. "CREATE TABLE IF NOT EXISTS dimension_delete (dimension_id blob, dimension_name text, chart_type_id text, "
  15. "dim_id blob, chart_id blob, host_id blob, date_created);",
  16. "CREATE INDEX IF NOT EXISTS ind_h1 ON dimension_delete (host_id);",
  17. "CREATE TRIGGER IF NOT EXISTS tr_dim_del AFTER DELETE ON dimension BEGIN INSERT INTO dimension_delete "
  18. "(dimension_id, dimension_name, chart_type_id, dim_id, chart_id, host_id, date_created)"
  19. " select old.id, old.name, c.type||\".\"||c.id, old.dim_id, old.chart_id, c.host_id, unixepoch() FROM"
  20. " chart c WHERE c.chart_id = old.chart_id; END;",
  21. "DELETE FROM dimension_delete WHERE host_id NOT IN"
  22. " (SELECT host_id FROM host) OR unixepoch() - date_created > 604800;",
  23. NULL,
  24. };
  25. uv_mutex_t aclk_async_lock;
  26. struct aclk_database_worker_config *aclk_thread_head = NULL;
  27. int retention_running = 0;
  28. #ifdef ENABLE_ACLK
  29. static void stop_retention_run()
  30. {
  31. uv_mutex_lock(&aclk_async_lock);
  32. retention_running = 0;
  33. uv_mutex_unlock(&aclk_async_lock);
  34. }
  35. static int request_retention_run()
  36. {
  37. int rc = 0;
  38. uv_mutex_lock(&aclk_async_lock);
  39. if (unlikely(retention_running))
  40. rc = 1;
  41. else
  42. retention_running = 1;
  43. uv_mutex_unlock(&aclk_async_lock);
  44. return rc;
  45. }
  46. #endif
  47. int claimed()
  48. {
  49. int rc;
  50. rrdhost_aclk_state_lock(localhost);
  51. rc = (localhost->aclk_state.claimed_id != NULL);
  52. rrdhost_aclk_state_unlock(localhost);
  53. return rc;
  54. }
  55. void aclk_add_worker_thread(struct aclk_database_worker_config *wc)
  56. {
  57. if (unlikely(!wc))
  58. return;
  59. uv_mutex_lock(&aclk_async_lock);
  60. if (unlikely(!wc->host)) {
  61. wc->next = aclk_thread_head;
  62. aclk_thread_head = wc;
  63. }
  64. uv_mutex_unlock(&aclk_async_lock);
  65. return;
  66. }
  67. void aclk_del_worker_thread(struct aclk_database_worker_config *wc)
  68. {
  69. if (unlikely(!wc))
  70. return;
  71. uv_mutex_lock(&aclk_async_lock);
  72. struct aclk_database_worker_config **tmp = &aclk_thread_head;
  73. while (*tmp && (*tmp) != wc)
  74. tmp = &(*tmp)->next;
  75. if (*tmp)
  76. *tmp = wc->next;
  77. uv_mutex_unlock(&aclk_async_lock);
  78. return;
  79. }
  80. int aclk_worker_thread_exists(char *guid)
  81. {
  82. int rc = 0;
  83. uv_mutex_lock(&aclk_async_lock);
  84. struct aclk_database_worker_config *tmp = aclk_thread_head;
  85. while (tmp && !rc) {
  86. rc = strcmp(tmp->uuid_str, guid) == 0;
  87. tmp = tmp->next;
  88. }
  89. uv_mutex_unlock(&aclk_async_lock);
  90. return rc;
  91. }
  92. void aclk_database_init_cmd_queue(struct aclk_database_worker_config *wc)
  93. {
  94. wc->cmd_queue.head = wc->cmd_queue.tail = 0;
  95. wc->queue_size = 0;
  96. fatal_assert(0 == uv_cond_init(&wc->cmd_cond));
  97. fatal_assert(0 == uv_mutex_init(&wc->cmd_mutex));
  98. }
  99. int aclk_database_enq_cmd_noblock(struct aclk_database_worker_config *wc, struct aclk_database_cmd *cmd)
  100. {
  101. unsigned queue_size;
  102. /* wait for free space in queue */
  103. uv_mutex_lock(&wc->cmd_mutex);
  104. if ((queue_size = wc->queue_size) == ACLK_DATABASE_CMD_Q_MAX_SIZE || wc->is_shutting_down) {
  105. uv_mutex_unlock(&wc->cmd_mutex);
  106. return 1;
  107. }
  108. fatal_assert(queue_size < ACLK_DATABASE_CMD_Q_MAX_SIZE);
  109. /* enqueue command */
  110. wc->cmd_queue.cmd_array[wc->cmd_queue.tail] = *cmd;
  111. wc->cmd_queue.tail = wc->cmd_queue.tail != ACLK_DATABASE_CMD_Q_MAX_SIZE - 1 ?
  112. wc->cmd_queue.tail + 1 : 0;
  113. wc->queue_size = queue_size + 1;
  114. uv_mutex_unlock(&wc->cmd_mutex);
  115. return 0;
  116. }
  117. void aclk_database_enq_cmd(struct aclk_database_worker_config *wc, struct aclk_database_cmd *cmd)
  118. {
  119. unsigned queue_size;
  120. /* wait for free space in queue */
  121. uv_mutex_lock(&wc->cmd_mutex);
  122. if (wc->is_shutting_down) {
  123. uv_mutex_unlock(&wc->cmd_mutex);
  124. return;
  125. }
  126. while ((queue_size = wc->queue_size) == ACLK_DATABASE_CMD_Q_MAX_SIZE) {
  127. uv_cond_wait(&wc->cmd_cond, &wc->cmd_mutex);
  128. }
  129. fatal_assert(queue_size < ACLK_DATABASE_CMD_Q_MAX_SIZE);
  130. /* enqueue command */
  131. wc->cmd_queue.cmd_array[wc->cmd_queue.tail] = *cmd;
  132. wc->cmd_queue.tail = wc->cmd_queue.tail != ACLK_DATABASE_CMD_Q_MAX_SIZE - 1 ?
  133. wc->cmd_queue.tail + 1 : 0;
  134. wc->queue_size = queue_size + 1;
  135. uv_mutex_unlock(&wc->cmd_mutex);
  136. /* wake up event loop */
  137. int rc = uv_async_send(&wc->async);
  138. if (unlikely(rc))
  139. debug(D_ACLK_SYNC, "Failed to wake up event loop");
  140. }
  141. struct aclk_database_cmd aclk_database_deq_cmd(struct aclk_database_worker_config* wc)
  142. {
  143. struct aclk_database_cmd ret;
  144. unsigned queue_size;
  145. uv_mutex_lock(&wc->cmd_mutex);
  146. queue_size = wc->queue_size;
  147. if (queue_size == 0 || wc->is_shutting_down) {
  148. memset(&ret, 0, sizeof(ret));
  149. ret.opcode = ACLK_DATABASE_NOOP;
  150. ret.completion = NULL;
  151. if (wc->is_shutting_down)
  152. uv_cond_signal(&wc->cmd_cond);
  153. } else {
  154. /* dequeue command */
  155. ret = wc->cmd_queue.cmd_array[wc->cmd_queue.head];
  156. if (queue_size == 1) {
  157. wc->cmd_queue.head = wc->cmd_queue.tail = 0;
  158. } else {
  159. wc->cmd_queue.head = wc->cmd_queue.head != ACLK_DATABASE_CMD_Q_MAX_SIZE - 1 ?
  160. wc->cmd_queue.head + 1 : 0;
  161. }
  162. wc->queue_size = queue_size - 1;
  163. /* wake up producers */
  164. uv_cond_signal(&wc->cmd_cond);
  165. }
  166. uv_mutex_unlock(&wc->cmd_mutex);
  167. return ret;
  168. }
  169. int aclk_worker_enq_cmd(char *node_id, struct aclk_database_cmd *cmd)
  170. {
  171. if (unlikely(!node_id || !cmd))
  172. return 0;
  173. uv_mutex_lock(&aclk_async_lock);
  174. struct aclk_database_worker_config *wc = aclk_thread_head;
  175. while (wc) {
  176. if (!strcmp(wc->node_id, node_id))
  177. break;
  178. wc = wc->next;
  179. }
  180. uv_mutex_unlock(&aclk_async_lock);
  181. if (wc)
  182. aclk_database_enq_cmd(wc, cmd);
  183. return (wc == NULL);
  184. }
  185. struct aclk_database_worker_config *find_inactive_wc_by_node_id(char *node_id)
  186. {
  187. if (unlikely(!node_id))
  188. return NULL;
  189. uv_mutex_lock(&aclk_async_lock);
  190. struct aclk_database_worker_config *wc = aclk_thread_head;
  191. while (wc) {
  192. if (!strcmp(wc->node_id, node_id))
  193. break;
  194. wc = wc->next;
  195. }
  196. uv_mutex_unlock(&aclk_async_lock);
  197. return (wc);
  198. }
  199. void aclk_sync_exit_all()
  200. {
  201. rrd_rdlock();
  202. RRDHOST *host = localhost;
  203. while(host) {
  204. struct aclk_database_worker_config *wc = host->dbsync_worker;
  205. if (wc) {
  206. wc->is_shutting_down = 1;
  207. (void) aclk_database_deq_cmd(wc);
  208. uv_cond_signal(&wc->cmd_cond);
  209. }
  210. host = host->next;
  211. }
  212. rrd_unlock();
  213. uv_mutex_lock(&aclk_async_lock);
  214. struct aclk_database_worker_config *wc = aclk_thread_head;
  215. while (wc) {
  216. wc->is_shutting_down = 1;
  217. wc = wc->next;
  218. }
  219. uv_mutex_unlock(&aclk_async_lock);
  220. }
  221. #ifdef ENABLE_ACLK
  222. enum {
  223. IDX_HOST_ID,
  224. IDX_HOSTNAME,
  225. IDX_REGISTRY,
  226. IDX_UPDATE_EVERY,
  227. IDX_OS,
  228. IDX_TIMEZONE,
  229. IDX_TAGS,
  230. IDX_HOPS,
  231. IDX_MEMORY_MODE,
  232. IDX_ABBREV_TIMEZONE,
  233. IDX_UTC_OFFSET,
  234. IDX_PROGRAM_NAME,
  235. IDX_PROGRAM_VERSION,
  236. IDX_ENTRIES,
  237. IDX_HEALTH_ENABLED,
  238. };
  239. static int create_host_callback(void *data, int argc, char **argv, char **column)
  240. {
  241. UNUSED(data);
  242. UNUSED(argc);
  243. UNUSED(column);
  244. char guid[UUID_STR_LEN];
  245. uuid_unparse_lower(*(uuid_t *)argv[IDX_HOST_ID], guid);
  246. struct rrdhost_system_info *system_info = callocz(1, sizeof(struct rrdhost_system_info));
  247. system_info->hops = str2i((const char *) argv[IDX_HOPS]);
  248. sql_build_host_system_info((uuid_t *)argv[IDX_HOST_ID], system_info);
  249. RRDHOST *host = rrdhost_find_or_create(
  250. (const char *) argv[IDX_HOSTNAME]
  251. , (const char *) argv[IDX_REGISTRY]
  252. , guid
  253. , (const char *) argv[IDX_OS]
  254. , (const char *) argv[IDX_TIMEZONE]
  255. , (const char *) argv[IDX_ABBREV_TIMEZONE]
  256. , argv[IDX_UTC_OFFSET] ? str2uint32_t(argv[IDX_UTC_OFFSET]) : 0
  257. , (const char *) argv[IDX_TAGS]
  258. , (const char *) (argv[IDX_PROGRAM_NAME] ? argv[IDX_PROGRAM_NAME] : "unknown")
  259. , (const char *) (argv[IDX_PROGRAM_VERSION] ? argv[IDX_PROGRAM_VERSION] : "unknown")
  260. , argv[3] ? str2i(argv[IDX_UPDATE_EVERY]) : 1
  261. , argv[13] ? str2i(argv[IDX_ENTRIES]) : 0
  262. , RRD_MEMORY_MODE_DBENGINE
  263. , 0 // health
  264. , 0 // rrdpush enabled
  265. , NULL //destination
  266. , NULL // api key
  267. , NULL // send charts matching
  268. , system_info
  269. , 1
  270. );
  271. if (likely(host))
  272. host->host_labels = sql_load_host_labels((uuid_t *)argv[IDX_HOST_ID]);
  273. #ifdef NETDATA_INTERNAL_CHECKS
  274. char node_str[UUID_STR_LEN] = "<none>";
  275. if (likely(host->node_id))
  276. uuid_unparse_lower(*host->node_id, node_str);
  277. internal_error(true, "Adding archived host \"%s\" with GUID \"%s\" node id = \"%s\"", host->hostname, host->machine_guid, node_str);
  278. #endif
  279. return 0;
  280. }
  281. #endif
  282. int aclk_start_sync_thread(void *data, int argc, char **argv, char **column)
  283. {
  284. char uuid_str[GUID_LEN + 1];
  285. UNUSED(data);
  286. UNUSED(argc);
  287. UNUSED(column);
  288. uuid_unparse_lower(*((uuid_t *) argv[0]), uuid_str);
  289. RRDHOST *host = rrdhost_find_by_guid(uuid_str, 0);
  290. if (host == localhost)
  291. return 0;
  292. sql_create_aclk_table(host, (uuid_t *) argv[0], (uuid_t *) argv[1]);
  293. return 0;
  294. }
  295. void sql_aclk_sync_init(void)
  296. {
  297. #ifdef ENABLE_ACLK
  298. char *err_msg = NULL;
  299. int rc;
  300. if (unlikely(!db_meta)) {
  301. if (default_rrd_memory_mode != RRD_MEMORY_MODE_DBENGINE) {
  302. return;
  303. }
  304. error_report("Database has not been initialized");
  305. return;
  306. }
  307. info("SQLite aclk sync initialization");
  308. for (int i = 0; aclk_sync_config[i]; i++) {
  309. debug(D_ACLK_SYNC, "Executing %s", aclk_sync_config[i]);
  310. rc = sqlite3_exec(db_meta, aclk_sync_config[i], 0, 0, &err_msg);
  311. if (rc != SQLITE_OK) {
  312. error_report("SQLite error aclk sync initialization setup, rc = %d (%s)", rc, err_msg);
  313. error_report("SQLite failed statement %s", aclk_sync_config[i]);
  314. sqlite3_free(err_msg);
  315. return;
  316. }
  317. }
  318. info("SQLite aclk sync initialization completed");
  319. fatal_assert(0 == uv_mutex_init(&aclk_async_lock));
  320. if (likely(rrdcontext_enabled == CONFIG_BOOLEAN_YES)) {
  321. rc = sqlite3_exec(db_meta, "SELECT host_id, hostname, registry_hostname, update_every, os, "
  322. "timezone, tags, hops, memory_mode, abbrev_timezone, utc_offset, program_name, "
  323. "program_version, entries, health_enabled FROM host WHERE hops >0;",
  324. create_host_callback, NULL, &err_msg);
  325. if (rc != SQLITE_OK) {
  326. error_report("SQLite error when loading archived hosts, rc = %d (%s)", rc, err_msg);
  327. sqlite3_free(err_msg);
  328. }
  329. }
  330. rc = sqlite3_exec(db_meta, "SELECT ni.host_id, ni.node_id FROM host h, node_instance ni WHERE "
  331. "h.host_id = ni.host_id AND ni.node_id IS NOT NULL;", aclk_start_sync_thread, NULL, &err_msg);
  332. if (rc != SQLITE_OK) {
  333. error_report("SQLite error when starting ACLK sync threads, rc = %d (%s)", rc, err_msg);
  334. sqlite3_free(err_msg);
  335. }
  336. #endif
  337. return;
  338. }
  339. static void async_cb(uv_async_t *handle)
  340. {
  341. uv_stop(handle->loop);
  342. uv_update_time(handle->loop);
  343. debug(D_ACLK_SYNC, "%s called, active=%d.", __func__, uv_is_active((uv_handle_t *)handle));
  344. }
  345. #define TIMER_PERIOD_MS (1000)
  346. static void timer_cb(uv_timer_t* handle)
  347. {
  348. uv_stop(handle->loop);
  349. uv_update_time(handle->loop);
  350. #ifdef ENABLE_ACLK
  351. struct aclk_database_worker_config *wc = handle->data;
  352. struct aclk_database_cmd cmd;
  353. memset(&cmd, 0, sizeof(cmd));
  354. cmd.opcode = ACLK_DATABASE_TIMER;
  355. aclk_database_enq_cmd_noblock(wc, &cmd);
  356. time_t now = now_realtime_sec();
  357. if (wc->cleanup_after && wc->cleanup_after < now) {
  358. cmd.opcode = ACLK_DATABASE_CLEANUP;
  359. if (!aclk_database_enq_cmd_noblock(wc, &cmd))
  360. wc->cleanup_after += ACLK_DATABASE_CLEANUP_INTERVAL;
  361. }
  362. if (aclk_connected) {
  363. if (wc->rotation_after && wc->rotation_after < now) {
  364. cmd.opcode = ACLK_DATABASE_UPD_RETENTION;
  365. if (!aclk_database_enq_cmd_noblock(wc, &cmd))
  366. wc->rotation_after += ACLK_DATABASE_ROTATION_INTERVAL;
  367. }
  368. if (wc->chart_updates && !wc->chart_pending && wc->chart_payload_count) {
  369. cmd.opcode = ACLK_DATABASE_PUSH_CHART;
  370. cmd.count = ACLK_MAX_CHART_BATCH;
  371. cmd.param1 = ACLK_MAX_CHART_BATCH_COUNT;
  372. if (!aclk_database_enq_cmd_noblock(wc, &cmd)) {
  373. if (wc->retry_count)
  374. info("Queued chart/dimension payload command %s, retry count = %u", wc->host_guid, wc->retry_count);
  375. wc->chart_pending = 1;
  376. wc->retry_count = 0;
  377. } else {
  378. wc->retry_count++;
  379. if (wc->retry_count % 100 == 0)
  380. error_report("Failed to queue chart/dimension payload command %s, retry count = %u",
  381. wc->host_guid,
  382. wc->retry_count);
  383. }
  384. }
  385. if (wc->alert_updates && !wc->pause_alert_updates) {
  386. cmd.opcode = ACLK_DATABASE_PUSH_ALERT;
  387. cmd.count = ACLK_MAX_ALERT_UPDATES;
  388. aclk_database_enq_cmd_noblock(wc, &cmd);
  389. }
  390. }
  391. #endif
  392. }
  393. #ifdef ENABLE_ACLK
  394. void after_send_retention(uv_work_t *req, int status)
  395. {
  396. struct aclk_database_worker_config *wc = req->data;
  397. (void)status;
  398. stop_retention_run();
  399. wc->retention_running = 0;
  400. struct aclk_database_cmd cmd;
  401. memset(&cmd, 0, sizeof(cmd));
  402. cmd.opcode = ACLK_DATABASE_DIM_DELETION;
  403. if (aclk_database_enq_cmd_noblock(wc, &cmd))
  404. info("Failed to queue a dimension deletion message");
  405. cmd.opcode = ACLK_DATABASE_NODE_INFO;
  406. if (aclk_database_enq_cmd_noblock(wc, &cmd))
  407. info("Failed to queue a node update info message");
  408. }
  409. static void send_retention(uv_work_t *req)
  410. {
  411. struct aclk_database_worker_config *wc = req->data;
  412. if (unlikely(wc->is_shutting_down))
  413. return;
  414. aclk_update_retention(wc);
  415. }
  416. #endif
  417. #define MAX_CMD_BATCH_SIZE (256)
  418. void aclk_database_worker(void *arg)
  419. {
  420. worker_register("ACLKSYNC");
  421. worker_register_job_name(ACLK_DATABASE_NOOP, "noop");
  422. worker_register_job_name(ACLK_DATABASE_ADD_CHART, "chart add");
  423. worker_register_job_name(ACLK_DATABASE_ADD_DIMENSION, "dimension add");
  424. worker_register_job_name(ACLK_DATABASE_PUSH_CHART, "chart push");
  425. worker_register_job_name(ACLK_DATABASE_PUSH_CHART_CONFIG, "chart conf push");
  426. worker_register_job_name(ACLK_DATABASE_RESET_CHART, "chart reset");
  427. worker_register_job_name(ACLK_DATABASE_CHART_ACK, "chart ack");
  428. worker_register_job_name(ACLK_DATABASE_UPD_RETENTION, "retention check");
  429. worker_register_job_name(ACLK_DATABASE_DIM_DELETION, "dimension delete");
  430. worker_register_job_name(ACLK_DATABASE_ORPHAN_HOST, "node orphan");
  431. worker_register_job_name(ACLK_DATABASE_ALARM_HEALTH_LOG, "alert log");
  432. worker_register_job_name(ACLK_DATABASE_CLEANUP, "cleanup");
  433. worker_register_job_name(ACLK_DATABASE_DELETE_HOST, "node delete");
  434. worker_register_job_name(ACLK_DATABASE_NODE_INFO, "node info");
  435. worker_register_job_name(ACLK_DATABASE_NODE_COLLECTORS, "node collectors");
  436. worker_register_job_name(ACLK_DATABASE_PUSH_ALERT, "alert push");
  437. worker_register_job_name(ACLK_DATABASE_PUSH_ALERT_CONFIG, "alert conf push");
  438. worker_register_job_name(ACLK_DATABASE_PUSH_ALERT_SNAPSHOT, "alert snapshot");
  439. worker_register_job_name(ACLK_DATABASE_QUEUE_REMOVED_ALERTS, "alerts check");
  440. worker_register_job_name(ACLK_DATABASE_TIMER, "timer");
  441. struct aclk_database_worker_config *wc = arg;
  442. uv_loop_t *loop;
  443. int ret;
  444. enum aclk_database_opcode opcode;
  445. uv_timer_t timer_req;
  446. struct aclk_database_cmd cmd;
  447. unsigned cmd_batch_size;
  448. //aclk_database_init_cmd_queue(wc);
  449. char threadname[NETDATA_THREAD_NAME_MAX+1];
  450. if (wc->host)
  451. snprintfz(threadname, NETDATA_THREAD_NAME_MAX, "AS_%s", wc->host->hostname);
  452. else {
  453. snprintfz(threadname, NETDATA_THREAD_NAME_MAX, "AS_%s", wc->uuid_str);
  454. threadname[11] = '\0';
  455. }
  456. uv_thread_set_name_np(wc->thread, threadname);
  457. loop = wc->loop = mallocz(sizeof(uv_loop_t));
  458. ret = uv_loop_init(loop);
  459. if (ret) {
  460. error("uv_loop_init(): %s", uv_strerror(ret));
  461. goto error_after_loop_init;
  462. }
  463. loop->data = wc;
  464. ret = uv_async_init(wc->loop, &wc->async, async_cb);
  465. if (ret) {
  466. error("uv_async_init(): %s", uv_strerror(ret));
  467. goto error_after_async_init;
  468. }
  469. wc->async.data = wc;
  470. ret = uv_timer_init(loop, &timer_req);
  471. if (ret) {
  472. error("uv_timer_init(): %s", uv_strerror(ret));
  473. goto error_after_timer_init;
  474. }
  475. timer_req.data = wc;
  476. fatal_assert(0 == uv_timer_start(&timer_req, timer_cb, TIMER_PERIOD_MS, TIMER_PERIOD_MS));
  477. // wc->retry_count = 0;
  478. wc->node_info_send = 1;
  479. // aclk_add_worker_thread(wc);
  480. info("Starting ACLK sync thread for host %s -- scratch area %lu bytes", wc->host_guid, sizeof(*wc));
  481. memset(&cmd, 0, sizeof(cmd));
  482. #ifdef ENABLE_ACLK
  483. uv_work_t retention_work;
  484. sql_get_last_chart_sequence(wc);
  485. wc->chart_payload_count = sql_get_pending_count(wc);
  486. if (!wc->chart_payload_count)
  487. info("%s: No pending charts and dimensions detected during startup", wc->host_guid);
  488. #endif
  489. wc->startup_time = now_realtime_sec();
  490. wc->cleanup_after = wc->startup_time + ACLK_DATABASE_CLEANUP_FIRST;
  491. wc->rotation_after = wc->startup_time + ACLK_DATABASE_ROTATION_DELAY;
  492. debug(D_ACLK_SYNC,"Node %s reports pending message count = %u", wc->node_id, wc->chart_payload_count);
  493. while (likely(!netdata_exit)) {
  494. worker_is_idle();
  495. uv_run(loop, UV_RUN_DEFAULT);
  496. /* wait for commands */
  497. cmd_batch_size = 0;
  498. do {
  499. if (unlikely(cmd_batch_size >= MAX_CMD_BATCH_SIZE))
  500. break;
  501. cmd = aclk_database_deq_cmd(wc);
  502. if (netdata_exit)
  503. break;
  504. opcode = cmd.opcode;
  505. ++cmd_batch_size;
  506. if(likely(opcode != ACLK_DATABASE_NOOP))
  507. worker_is_busy(opcode);
  508. switch (opcode) {
  509. case ACLK_DATABASE_NOOP:
  510. /* the command queue was empty, do nothing */
  511. break;
  512. // MAINTENANCE
  513. case ACLK_DATABASE_CLEANUP:
  514. debug(D_ACLK_SYNC, "Database cleanup for %s", wc->host_guid);
  515. sql_maint_aclk_sync_database(wc, cmd);
  516. if (wc->host == localhost)
  517. sql_check_aclk_table_list(wc);
  518. break;
  519. case ACLK_DATABASE_DELETE_HOST:
  520. debug(D_ACLK_SYNC,"Cleaning ACLK tables for %s", (char *) cmd.data);
  521. sql_delete_aclk_table_list(wc, cmd);
  522. break;
  523. // CHART / DIMENSION OPERATIONS
  524. #ifdef ENABLE_ACLK
  525. case ACLK_DATABASE_ADD_CHART:
  526. debug(D_ACLK_SYNC, "Adding chart event for %s", wc->host_guid);
  527. aclk_add_chart_event(wc, cmd);
  528. break;
  529. case ACLK_DATABASE_ADD_DIMENSION:
  530. debug(D_ACLK_SYNC, "Adding dimension event for %s", wc->host_guid);
  531. aclk_add_dimension_event(wc, cmd);
  532. break;
  533. case ACLK_DATABASE_PUSH_CHART:
  534. debug(D_ACLK_SYNC, "Pushing chart info to the cloud for node %s", wc->host_guid);
  535. aclk_send_chart_event(wc, cmd);
  536. break;
  537. case ACLK_DATABASE_PUSH_CHART_CONFIG:
  538. debug(D_ACLK_SYNC, "Pushing chart config info to the cloud for node %s", wc->host_guid);
  539. aclk_send_chart_config(wc, cmd);
  540. break;
  541. case ACLK_DATABASE_CHART_ACK:
  542. debug(D_ACLK_SYNC, "ACK chart SEQ for %s to %"PRIu64, wc->uuid_str, (uint64_t) cmd.param1);
  543. aclk_receive_chart_ack(wc, cmd);
  544. break;
  545. case ACLK_DATABASE_RESET_CHART:
  546. debug(D_ACLK_SYNC, "RESET chart SEQ for %s to %"PRIu64, wc->uuid_str, (uint64_t) cmd.param1);
  547. aclk_receive_chart_reset(wc, cmd);
  548. break;
  549. #endif
  550. // ALERTS
  551. case ACLK_DATABASE_PUSH_ALERT_CONFIG:
  552. debug(D_ACLK_SYNC,"Pushing chart config info to the cloud for %s", wc->host_guid);
  553. aclk_push_alert_config_event(wc, cmd);
  554. break;
  555. case ACLK_DATABASE_PUSH_ALERT:
  556. debug(D_ACLK_SYNC, "Pushing alert info to the cloud for %s", wc->host_guid);
  557. aclk_push_alert_event(wc, cmd);
  558. break;
  559. case ACLK_DATABASE_ALARM_HEALTH_LOG:
  560. debug(D_ACLK_SYNC, "Pushing alarm health log to the cloud for %s", wc->host_guid);
  561. aclk_push_alarm_health_log(wc, cmd);
  562. break;
  563. case ACLK_DATABASE_PUSH_ALERT_SNAPSHOT:
  564. debug(D_ACLK_SYNC, "Pushing alert snapshot to the cloud for node %s", wc->host_guid);
  565. aclk_push_alert_snapshot_event(wc, cmd);
  566. break;
  567. case ACLK_DATABASE_QUEUE_REMOVED_ALERTS:
  568. debug(D_ACLK_SYNC, "Queueing removed alerts for node %s", wc->host_guid);
  569. sql_process_queue_removed_alerts_to_aclk(wc, cmd);
  570. break;
  571. // NODE OPERATIONS
  572. case ACLK_DATABASE_NODE_INFO:
  573. debug(D_ACLK_SYNC,"Sending node info for %s", wc->uuid_str);
  574. sql_build_node_info(wc, cmd);
  575. break;
  576. case ACLK_DATABASE_NODE_COLLECTORS:
  577. debug(D_ACLK_SYNC,"Sending node collectors info for %s", wc->uuid_str);
  578. sql_build_node_collectors(wc);
  579. break;
  580. #ifdef ENABLE_ACLK
  581. case ACLK_DATABASE_DIM_DELETION:
  582. debug(D_ACLK_SYNC,"Sending dimension deletion information %s", wc->uuid_str);
  583. aclk_process_dimension_deletion(wc, cmd);
  584. break;
  585. case ACLK_DATABASE_UPD_RETENTION:
  586. if (unlikely(wc->retention_running))
  587. break;
  588. if (unlikely(request_retention_run())) {
  589. wc->rotation_after = now_realtime_sec() + ACLK_DATABASE_RETENTION_RETRY;
  590. break;
  591. }
  592. debug(D_ACLK_SYNC,"Sending retention info for %s", wc->uuid_str);
  593. retention_work.data = wc;
  594. wc->retention_running = 1;
  595. if (unlikely(uv_queue_work(loop, &retention_work, send_retention, after_send_retention))) {
  596. wc->retention_running = 0;
  597. stop_retention_run();
  598. }
  599. break;
  600. // NODE_INSTANCE DETECTION
  601. case ACLK_DATABASE_ORPHAN_HOST:
  602. wc->host = NULL;
  603. wc->is_orphan = 1;
  604. aclk_add_worker_thread(wc);
  605. break;
  606. #endif
  607. case ACLK_DATABASE_TIMER:
  608. if (unlikely(localhost && !wc->host && !wc->is_orphan)) {
  609. if (claimed()) {
  610. wc->host = rrdhost_find_by_guid(wc->host_guid, 0);
  611. if (wc->host) {
  612. info("HOST %s (%s) detected as active", wc->host->hostname, wc->host_guid);
  613. snprintfz(threadname, NETDATA_THREAD_NAME_MAX, "AS_%s", wc->host->hostname);
  614. uv_thread_set_name_np(wc->thread, threadname);
  615. wc->host->dbsync_worker = wc;
  616. if (unlikely(!wc->hostname))
  617. wc->hostname = strdupz(wc->host->hostname);
  618. aclk_del_worker_thread(wc);
  619. wc->node_info_send = 1;
  620. }
  621. }
  622. }
  623. if (wc->node_info_send && localhost && claimed() && aclk_connected) {
  624. cmd.opcode = ACLK_DATABASE_NODE_INFO;
  625. cmd.completion = NULL;
  626. wc->node_info_send = aclk_database_enq_cmd_noblock(wc, &cmd);
  627. }
  628. if (wc->node_collectors_send && wc->node_collectors_send + 30 < now_realtime_sec()) {
  629. cmd.opcode = ACLK_DATABASE_NODE_COLLECTORS;
  630. cmd.completion = NULL;
  631. wc->node_collectors_send = aclk_database_enq_cmd_noblock(wc, &cmd);
  632. }
  633. if (localhost == wc->host)
  634. (void) sqlite3_wal_checkpoint(db_meta, NULL);
  635. break;
  636. default:
  637. debug(D_ACLK_SYNC, "%s: default.", __func__);
  638. break;
  639. }
  640. if (cmd.completion)
  641. aclk_complete(cmd.completion);
  642. } while (opcode != ACLK_DATABASE_NOOP);
  643. }
  644. if (!uv_timer_stop(&timer_req))
  645. uv_close((uv_handle_t *)&timer_req, NULL);
  646. /* cleanup operations of the event loop */
  647. //info("Shutting down ACLK sync event loop for %s", wc->host_guid);
  648. /*
  649. * uv_async_send after uv_close does not seem to crash in linux at the moment,
  650. * it is however undocumented behaviour we need to be aware if this becomes
  651. * an issue in the future.
  652. */
  653. uv_close((uv_handle_t *)&wc->async, NULL);
  654. uv_run(loop, UV_RUN_DEFAULT);
  655. info("Shutting down ACLK sync event loop complete for host %s", wc->host_guid);
  656. /* TODO: don't let the API block by waiting to enqueue commands */
  657. uv_cond_destroy(&wc->cmd_cond);
  658. /* uv_mutex_destroy(&wc->cmd_mutex); */
  659. //fatal_assert(0 == uv_loop_close(loop));
  660. int rc;
  661. do {
  662. rc = uv_loop_close(loop);
  663. } while (rc != UV_EBUSY);
  664. freez(loop);
  665. rrd_rdlock();
  666. if (likely(wc->host))
  667. wc->host->dbsync_worker = NULL;
  668. freez(wc->hostname);
  669. freez(wc);
  670. rrd_unlock();
  671. worker_unregister();
  672. return;
  673. error_after_timer_init:
  674. uv_close((uv_handle_t *)&wc->async, NULL);
  675. error_after_async_init:
  676. fatal_assert(0 == uv_loop_close(loop));
  677. error_after_loop_init:
  678. freez(loop);
  679. worker_unregister();
  680. }
  681. // -------------------------------------------------------------
  682. void sql_create_aclk_table(RRDHOST *host, uuid_t *host_uuid, uuid_t *node_id)
  683. {
  684. #ifdef ENABLE_ACLK
  685. char uuid_str[GUID_LEN + 1];
  686. char host_guid[GUID_LEN + 1];
  687. uuid_unparse_lower_fix(host_uuid, uuid_str);
  688. if (aclk_worker_thread_exists(uuid_str))
  689. return;
  690. uuid_unparse_lower(*host_uuid, host_guid);
  691. BUFFER *sql = buffer_create(ACLK_SYNC_QUERY_SIZE);
  692. buffer_sprintf(sql, TABLE_ACLK_CHART, uuid_str);
  693. db_execute(buffer_tostring(sql));
  694. buffer_flush(sql);
  695. buffer_sprintf(sql, TABLE_ACLK_CHART_PAYLOAD, uuid_str);
  696. db_execute(buffer_tostring(sql));
  697. buffer_flush(sql);
  698. buffer_sprintf(sql, TABLE_ACLK_CHART_LATEST, uuid_str);
  699. db_execute(buffer_tostring(sql));
  700. buffer_flush(sql);
  701. buffer_sprintf(sql, INDEX_ACLK_CHART, uuid_str, uuid_str);
  702. db_execute(buffer_tostring(sql));
  703. buffer_flush(sql);
  704. buffer_sprintf(sql, INDEX_ACLK_CHART_LATEST, uuid_str, uuid_str);
  705. db_execute(buffer_tostring(sql));
  706. buffer_flush(sql);
  707. buffer_sprintf(sql, TRIGGER_ACLK_CHART_PAYLOAD, uuid_str, uuid_str, uuid_str);
  708. db_execute(buffer_tostring(sql));
  709. buffer_flush(sql);
  710. buffer_sprintf(sql, TABLE_ACLK_ALERT, uuid_str);
  711. db_execute(buffer_tostring(sql));
  712. buffer_flush(sql);
  713. buffer_sprintf(sql, INDEX_ACLK_ALERT, uuid_str, uuid_str);
  714. db_execute(buffer_tostring(sql));
  715. buffer_free(sql);
  716. if (likely(host) && unlikely(host->dbsync_worker))
  717. return;
  718. struct aclk_database_worker_config *wc = callocz(1, sizeof(struct aclk_database_worker_config));
  719. if (node_id && !uuid_is_null(*node_id))
  720. uuid_unparse_lower(*node_id, wc->node_id);
  721. if (likely(host)) {
  722. host->dbsync_worker = (void *)wc;
  723. wc->hostname = strdupz(host->hostname);
  724. }
  725. else
  726. wc->hostname = get_hostname_by_node_id(wc->node_id);
  727. wc->host = host;
  728. strcpy(wc->uuid_str, uuid_str);
  729. strcpy(wc->host_guid, host_guid);
  730. wc->chart_updates = 0;
  731. wc->alert_updates = 0;
  732. wc->retry_count = 0;
  733. aclk_database_init_cmd_queue(wc);
  734. aclk_add_worker_thread(wc);
  735. fatal_assert(0 == uv_thread_create(&(wc->thread), aclk_database_worker, wc));
  736. #else
  737. UNUSED(host);
  738. UNUSED(host_uuid);
  739. UNUSED(node_id);
  740. #endif
  741. return;
  742. }
  743. void sql_maint_aclk_sync_database(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  744. {
  745. UNUSED(cmd);
  746. debug(D_ACLK, "Checking database for %s", wc->host_guid);
  747. BUFFER *sql = buffer_create(ACLK_SYNC_QUERY_SIZE);
  748. buffer_sprintf(sql,"DELETE FROM aclk_chart_%s WHERE date_submitted IS NOT NULL AND "
  749. "CAST(date_updated AS INT) < unixepoch()-%d;", wc->uuid_str, ACLK_DELETE_ACK_INTERNAL);
  750. db_execute(buffer_tostring(sql));
  751. buffer_flush(sql);
  752. buffer_sprintf(sql,"DELETE FROM aclk_chart_payload_%s WHERE unique_id NOT IN "
  753. "(SELECT unique_id FROM aclk_chart_%s) AND unique_id NOT IN (SELECT unique_id FROM aclk_chart_latest_%s);",
  754. wc->uuid_str, wc->uuid_str, wc->uuid_str);
  755. db_execute(buffer_tostring(sql));
  756. buffer_flush(sql);
  757. buffer_sprintf(sql,"DELETE FROM aclk_alert_%s WHERE date_submitted IS NOT NULL AND "
  758. "CAST(date_cloud_ack AS INT) < unixepoch()-%d;", wc->uuid_str, ACLK_DELETE_ACK_ALERTS_INTERNAL);
  759. db_execute(buffer_tostring(sql));
  760. buffer_flush(sql);
  761. buffer_sprintf(sql,"UPDATE aclk_chart_%s SET status = NULL, date_submitted=unixepoch() WHERE "
  762. "date_submitted IS NULL AND CAST(date_created AS INT) < unixepoch()-%d;", wc->uuid_str, ACLK_AUTO_MARK_SUBMIT_INTERVAL);
  763. db_execute(buffer_tostring(sql));
  764. buffer_flush(sql);
  765. buffer_sprintf(sql,"UPDATE aclk_chart_%s SET date_updated = unixepoch() WHERE date_updated IS NULL"
  766. " AND date_submitted IS NOT NULL AND CAST(date_submitted AS INT) < unixepoch()-%d;",
  767. wc->uuid_str, ACLK_AUTO_MARK_UPDATED_INTERVAL);
  768. db_execute(buffer_tostring(sql));
  769. buffer_free(sql);
  770. return;
  771. }
  772. #define SQL_SELECT_HOST_BY_UUID "SELECT host_id FROM host WHERE host_id = @host_id;"
  773. static int is_host_available(uuid_t *host_id)
  774. {
  775. sqlite3_stmt *res = NULL;
  776. int rc;
  777. if (unlikely(!db_meta)) {
  778. if (default_rrd_memory_mode == RRD_MEMORY_MODE_DBENGINE)
  779. error_report("Database has not been initialized");
  780. return 1;
  781. }
  782. rc = sqlite3_prepare_v2(db_meta, SQL_SELECT_HOST_BY_UUID, -1, &res, 0);
  783. if (unlikely(rc != SQLITE_OK)) {
  784. error_report("Failed to prepare statement to select node instance information for a node");
  785. return 1;
  786. }
  787. rc = sqlite3_bind_blob(res, 1, host_id, sizeof(*host_id), SQLITE_STATIC);
  788. if (unlikely(rc != SQLITE_OK)) {
  789. error_report("Failed to bind host_id parameter to select node instance information");
  790. goto failed;
  791. }
  792. rc = sqlite3_step(res);
  793. failed:
  794. if (unlikely(sqlite3_finalize(res) != SQLITE_OK))
  795. error_report("Failed to finalize the prepared statement when checking host existence");
  796. return (rc == SQLITE_ROW);
  797. }
  798. // OPCODE: ACLK_DATABASE_DELETE_HOST
  799. void sql_delete_aclk_table_list(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  800. {
  801. UNUSED(wc);
  802. char uuid_str[GUID_LEN + 1];
  803. char host_str[GUID_LEN + 1];
  804. int rc;
  805. uuid_t host_uuid;
  806. char *host_guid = (char *)cmd.data;
  807. if (unlikely(!host_guid))
  808. return;
  809. rc = uuid_parse(host_guid, host_uuid);
  810. freez(host_guid);
  811. if (rc)
  812. return;
  813. uuid_unparse_lower(host_uuid, host_str);
  814. uuid_unparse_lower_fix(&host_uuid, uuid_str);
  815. debug(D_ACLK_SYNC, "Checking if I should delete aclk tables for node %s", host_str);
  816. if (is_host_available(&host_uuid)) {
  817. debug(D_ACLK_SYNC, "Host %s exists, not deleting aclk sync tables", host_str);
  818. return;
  819. }
  820. debug(D_ACLK_SYNC, "Host %s does NOT exist, can delete aclk sync tables", host_str);
  821. sqlite3_stmt *res = NULL;
  822. BUFFER *sql = buffer_create(ACLK_SYNC_QUERY_SIZE);
  823. buffer_sprintf(sql,"SELECT 'drop '||type||' IF EXISTS '||name||';' FROM sqlite_schema " \
  824. "WHERE name LIKE 'aclk_%%_%s' AND type IN ('table', 'trigger', 'index');", uuid_str);
  825. rc = sqlite3_prepare_v2(db_meta, buffer_tostring(sql), -1, &res, 0);
  826. if (rc != SQLITE_OK) {
  827. error_report("Failed to prepare statement to clean up aclk tables");
  828. goto fail;
  829. }
  830. buffer_flush(sql);
  831. while (sqlite3_step(res) == SQLITE_ROW)
  832. buffer_strcat(sql, (char *) sqlite3_column_text(res, 0));
  833. rc = sqlite3_finalize(res);
  834. if (unlikely(rc != SQLITE_OK))
  835. error_report("Failed to finalize statement to clean up aclk tables, rc = %d", rc);
  836. db_execute(buffer_tostring(sql));
  837. fail:
  838. buffer_free(sql);
  839. return;
  840. }
  841. static int sql_check_aclk_table(void *data, int argc, char **argv, char **column)
  842. {
  843. struct aclk_database_worker_config *wc = data;
  844. UNUSED(argc);
  845. UNUSED(column);
  846. debug(D_ACLK_SYNC,"Scheduling aclk sync table check for node %s", (char *) argv[0]);
  847. struct aclk_database_cmd cmd;
  848. memset(&cmd, 0, sizeof(cmd));
  849. cmd.opcode = ACLK_DATABASE_DELETE_HOST;
  850. cmd.data = strdupz((char *) argv[0]);
  851. aclk_database_enq_cmd_noblock(wc, &cmd);
  852. return 0;
  853. }
  854. #define SQL_SELECT_ACLK_ACTIVE_LIST "SELECT REPLACE(SUBSTR(name,19),'_','-') FROM sqlite_schema " \
  855. "WHERE name LIKE 'aclk_chart_latest_%' AND type IN ('table');"
  856. void sql_check_aclk_table_list(struct aclk_database_worker_config *wc)
  857. {
  858. char *err_msg = NULL;
  859. debug(D_ACLK_SYNC,"Cleaning tables for nodes that do not exist");
  860. int rc = sqlite3_exec(db_meta, SQL_SELECT_ACLK_ACTIVE_LIST, sql_check_aclk_table, (void *) wc, &err_msg);
  861. if (rc != SQLITE_OK) {
  862. error_report("Query failed when trying to check for obsolete ACLK sync tables, %s", err_msg);
  863. sqlite3_free(err_msg);
  864. }
  865. db_execute("DELETE FROM dimension_delete WHERE host_id NOT IN (SELECT host_id FROM host) "
  866. " OR unixepoch() - date_created > 604800;");
  867. return;
  868. }
  869. void aclk_data_rotated(void)
  870. {
  871. #ifdef ENABLE_ACLK
  872. if (!aclk_connected)
  873. return;
  874. time_t next_rotation_time = now_realtime_sec()+ACLK_DATABASE_ROTATION_DELAY;
  875. rrd_rdlock();
  876. RRDHOST *this_host = localhost;
  877. while (this_host) {
  878. struct aclk_database_worker_config *wc = this_host->dbsync_worker;
  879. if (wc)
  880. wc->rotation_after = next_rotation_time;
  881. this_host = this_host->next;
  882. }
  883. rrd_unlock();
  884. struct aclk_database_worker_config *tmp = aclk_thread_head;
  885. uv_mutex_lock(&aclk_async_lock);
  886. while (tmp) {
  887. tmp->rotation_after = next_rotation_time;
  888. tmp = tmp->next;
  889. }
  890. uv_mutex_unlock(&aclk_async_lock);
  891. #endif
  892. return;
  893. }