sqlite_aclk_chart.c 46 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "sqlite_functions.h"
  3. #include "sqlite_aclk_chart.h"
  4. #ifdef ENABLE_ACLK
  5. #include "../../aclk/aclk_charts_api.h"
  6. #include "../../aclk/aclk.h"
  7. static inline int
  8. sql_queue_chart_payload(struct aclk_database_worker_config *wc, void *data, enum aclk_database_opcode opcode)
  9. {
  10. int rc;
  11. if (unlikely(!wc))
  12. return 1;
  13. struct aclk_database_cmd cmd;
  14. memset(&cmd, 0, sizeof(cmd));
  15. cmd.opcode = opcode;
  16. cmd.data = data;
  17. rc = aclk_database_enq_cmd_noblock(wc, &cmd);
  18. return rc;
  19. }
  20. static time_t payload_sent(char *uuid_str, uuid_t *uuid, void *payload, size_t payload_size)
  21. {
  22. static __thread sqlite3_stmt *res = NULL;
  23. int rc;
  24. time_t send_status = 0;
  25. if (unlikely(!res)) {
  26. char sql[ACLK_SYNC_QUERY_SIZE];
  27. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1, "SELECT acl.date_submitted FROM aclk_chart_latest_%s acl, aclk_chart_payload_%s acp "
  28. "WHERE acl.unique_id = acp.unique_id AND acl.uuid = @uuid AND acp.payload = @payload;",
  29. uuid_str, uuid_str);
  30. rc = prepare_statement(db_meta, sql, &res);
  31. if (rc != SQLITE_OK) {
  32. error_report("Failed to prepare statement to check payload data on %s", sql);
  33. return 0;
  34. }
  35. }
  36. rc = sqlite3_bind_blob(res, 1, uuid, sizeof(*uuid), SQLITE_STATIC);
  37. if (unlikely(rc != SQLITE_OK))
  38. goto bind_fail;
  39. rc = sqlite3_bind_blob(res, 2, payload, payload_size, SQLITE_STATIC);
  40. if (unlikely(rc != SQLITE_OK))
  41. goto bind_fail;
  42. while (sqlite3_step(res) == SQLITE_ROW) {
  43. send_status = (time_t) sqlite3_column_int64(res, 0);
  44. }
  45. bind_fail:
  46. if (unlikely(sqlite3_reset(res) != SQLITE_OK))
  47. error_report("Failed to reset statement in check payload, rc = %d", rc);
  48. return send_status;
  49. }
  50. static int aclk_add_chart_payload(
  51. struct aclk_database_worker_config *wc,
  52. uuid_t *uuid,
  53. char *claim_id,
  54. ACLK_PAYLOAD_TYPE payload_type,
  55. void *payload,
  56. size_t payload_size,
  57. time_t *send_status,
  58. int check_sent)
  59. {
  60. static __thread sqlite3_stmt *res_chart = NULL;
  61. int rc;
  62. time_t date_submitted;
  63. if (unlikely(!payload))
  64. return 0;
  65. if (check_sent) {
  66. date_submitted = payload_sent(wc->uuid_str, uuid, payload, payload_size);
  67. if (send_status)
  68. *send_status = date_submitted;
  69. if (date_submitted)
  70. return 0;
  71. }
  72. if (unlikely(!res_chart)) {
  73. char sql[ACLK_SYNC_QUERY_SIZE];
  74. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1,
  75. "INSERT INTO aclk_chart_payload_%s (unique_id, uuid, claim_id, date_created, type, payload) " \
  76. "VALUES (@unique_id, @uuid, @claim_id, unixepoch(), @type, @payload);", wc->uuid_str);
  77. rc = prepare_statement(db_meta, sql, &res_chart);
  78. if (rc != SQLITE_OK) {
  79. error_report("Failed to prepare statement to store chart payload data");
  80. return 1;
  81. }
  82. }
  83. uuid_t unique_uuid;
  84. uuid_generate(unique_uuid);
  85. uuid_t claim_uuid;
  86. if (uuid_parse(claim_id, claim_uuid))
  87. return 1;
  88. rc = sqlite3_bind_blob(res_chart, 1, &unique_uuid, sizeof(unique_uuid), SQLITE_STATIC);
  89. if (unlikely(rc != SQLITE_OK))
  90. goto bind_fail;
  91. rc = sqlite3_bind_blob(res_chart, 2, uuid, sizeof(*uuid), SQLITE_STATIC);
  92. if (unlikely(rc != SQLITE_OK))
  93. goto bind_fail;
  94. rc = sqlite3_bind_blob(res_chart, 3, &claim_uuid, sizeof(claim_uuid), SQLITE_STATIC);
  95. if (unlikely(rc != SQLITE_OK))
  96. goto bind_fail;
  97. rc = sqlite3_bind_int(res_chart, 4, payload_type);
  98. if (unlikely(rc != SQLITE_OK))
  99. goto bind_fail;
  100. rc = sqlite3_bind_blob(res_chart, 5, payload, payload_size, SQLITE_STATIC);
  101. if (unlikely(rc != SQLITE_OK))
  102. goto bind_fail;
  103. rc = execute_insert(res_chart);
  104. if (unlikely(rc != SQLITE_DONE))
  105. error_report("Failed store chart payload event, rc = %d", rc);
  106. else {
  107. wc->chart_payload_count++;
  108. time_t now = now_realtime_sec();
  109. if (wc->rotation_after > now && wc->rotation_after < now + ACLK_DATABASE_ROTATION_DELAY)
  110. wc->rotation_after = now + ACLK_DATABASE_ROTATION_DELAY;
  111. }
  112. bind_fail:
  113. if (unlikely(sqlite3_reset(res_chart) != SQLITE_OK))
  114. error_report("Failed to reset statement in store chart payload, rc = %d", rc);
  115. return (rc != SQLITE_DONE);
  116. }
  117. int aclk_add_chart_event(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  118. {
  119. int rc = 0;
  120. CHECK_SQLITE_CONNECTION(db_meta);
  121. char *claim_id = get_agent_claimid();
  122. RRDSET *st = cmd.data;
  123. if (likely(claim_id)) {
  124. struct chart_instance_updated chart_payload;
  125. memset(&chart_payload, 0, sizeof(chart_payload));
  126. chart_payload.config_hash = get_str_from_uuid(&st->state->hash_id);
  127. chart_payload.update_every = st->update_every;
  128. chart_payload.memory_mode = st->rrd_memory_mode;
  129. chart_payload.name = (char *)st->name;
  130. chart_payload.node_id = wc->node_id;
  131. chart_payload.claim_id = claim_id;
  132. chart_payload.id = strdupz(st->id);
  133. chart_payload.chart_labels = rrdlabels_create();
  134. rrdlabels_copy(chart_payload.chart_labels, st->state->chart_labels);
  135. size_t size;
  136. char *payload = generate_chart_instance_updated(&size, &chart_payload);
  137. if (likely(payload))
  138. rc = aclk_add_chart_payload(wc, st->chart_uuid, claim_id, ACLK_PAYLOAD_CHART, (void *) payload, size, NULL, 1);
  139. freez(payload);
  140. chart_instance_updated_destroy(&chart_payload);
  141. }
  142. return rc;
  143. }
  144. static inline int aclk_upd_dimension_event(struct aclk_database_worker_config *wc, char *claim_id, uuid_t *dim_uuid,
  145. const char *dim_id, const char *dim_name, const char *chart_type_id, time_t first_time, time_t last_time,
  146. time_t *send_status)
  147. {
  148. int rc = 0;
  149. size_t size;
  150. if (unlikely(!dim_uuid || !dim_id || !dim_name || !chart_type_id))
  151. return 0;
  152. struct chart_dimension_updated dim_payload;
  153. memset(&dim_payload, 0, sizeof(dim_payload));
  154. #ifdef NETDATA_INTERNAL_CHECKS
  155. if (!first_time)
  156. info("Host %s (node %s) deleting dimension id=[%s] name=[%s] chart=[%s]",
  157. wc->host_guid, wc->node_id, dim_id, dim_name, chart_type_id);
  158. if (last_time)
  159. info("Host %s (node %s) stopped collecting dimension id=[%s] name=[%s] chart=[%s] %ld seconds ago at %ld",
  160. wc->host_guid, wc->node_id, dim_id, dim_name, chart_type_id, now_realtime_sec() - last_time, last_time);
  161. #endif
  162. dim_payload.node_id = wc->node_id;
  163. dim_payload.claim_id = claim_id;
  164. dim_payload.name = dim_name;
  165. dim_payload.id = dim_id;
  166. dim_payload.chart_id = chart_type_id;
  167. dim_payload.created_at.tv_sec = first_time;
  168. dim_payload.last_timestamp.tv_sec = last_time;
  169. char *payload = generate_chart_dimension_updated(&size, &dim_payload);
  170. if (likely(payload))
  171. rc = aclk_add_chart_payload(wc, dim_uuid, claim_id, ACLK_PAYLOAD_DIMENSION, (void *)payload, size, send_status, 1);
  172. freez(payload);
  173. return rc;
  174. }
  175. void aclk_process_dimension_deletion(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  176. {
  177. int rc = 0;
  178. sqlite3_stmt *res = NULL;
  179. if (!aclk_connected)
  180. return;
  181. if (unlikely(!db_meta))
  182. return;
  183. uuid_t host_id;
  184. if (uuid_parse(wc->host_guid, host_id))
  185. return;
  186. char *claim_id = get_agent_claimid();
  187. if (!claim_id)
  188. return;
  189. rc = sqlite3_prepare_v2(
  190. db_meta,
  191. "DELETE FROM dimension_delete where host_id = @host_id "
  192. "RETURNING dimension_id, dimension_name, chart_type_id, dim_id LIMIT 10;",
  193. -1,
  194. &res,
  195. 0);
  196. if (rc != SQLITE_OK) {
  197. error_report("Failed to prepare statement when trying to delete dimension deletes");
  198. freez(claim_id);
  199. return;
  200. }
  201. rc = sqlite3_bind_blob(res, 1, &host_id, sizeof(host_id), SQLITE_STATIC);
  202. if (unlikely(rc != SQLITE_OK))
  203. goto bind_fail;
  204. unsigned count = 0;
  205. while (sqlite3_step(res) == SQLITE_ROW) {
  206. (void) aclk_upd_dimension_event(
  207. wc,
  208. claim_id,
  209. (uuid_t *)sqlite3_column_text(res, 3),
  210. (const char *)sqlite3_column_text(res, 0),
  211. (const char *)sqlite3_column_text(res, 1),
  212. (const char *)sqlite3_column_text(res, 2),
  213. 0,
  214. 0,
  215. NULL);
  216. count++;
  217. }
  218. if (count) {
  219. memset(&cmd, 0, sizeof(cmd));
  220. cmd.opcode = ACLK_DATABASE_DIM_DELETION;
  221. if (aclk_database_enq_cmd_noblock(wc, &cmd))
  222. info("Failed to queue a dimension deletion message");
  223. }
  224. bind_fail:
  225. rc = sqlite3_finalize(res);
  226. if (unlikely(rc != SQLITE_OK))
  227. error_report("Failed to finalize statement when adding dimension deletion events, rc = %d", rc);
  228. freez(claim_id);
  229. return;
  230. }
  231. int aclk_add_dimension_event(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  232. {
  233. int rc = 1;
  234. CHECK_SQLITE_CONNECTION(db_meta);
  235. struct aclk_chart_dimension_data *aclk_cd_data = cmd.data;
  236. char *claim_id = get_agent_claimid();
  237. if (!claim_id)
  238. goto cleanup;
  239. rc = aclk_add_chart_payload(wc, &aclk_cd_data->uuid, claim_id, ACLK_PAYLOAD_DIMENSION,
  240. (void *) aclk_cd_data->payload, aclk_cd_data->payload_size, NULL, aclk_cd_data->check_payload);
  241. freez(claim_id);
  242. cleanup:
  243. freez(aclk_cd_data->payload);
  244. freez(aclk_cd_data);
  245. return rc;
  246. }
  247. void aclk_send_chart_event(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  248. {
  249. int rc;
  250. wc->chart_pending = 0;
  251. if (unlikely(!wc->chart_updates)) {
  252. log_access(
  253. "ACLK STA [%s (%s)]: Ignoring chart push event, updates have been turned off for this node.",
  254. wc->node_id,
  255. wc->host ? wc->host->hostname : "N/A");
  256. return;
  257. }
  258. char *claim_id = get_agent_claimid();
  259. if (unlikely(!claim_id))
  260. return;
  261. uuid_t claim_uuid;
  262. if (uuid_parse(claim_id, claim_uuid))
  263. return;
  264. int limit = cmd.count > 0 ? cmd.count : 1;
  265. uint64_t first_sequence;
  266. uint64_t last_sequence;
  267. time_t last_timestamp = 0;
  268. char sql[ACLK_SYNC_QUERY_SIZE];
  269. static __thread sqlite3_stmt *res = NULL;
  270. if (unlikely(!res)) {
  271. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1,"SELECT ac.sequence_id, acp.payload, ac.date_created, ac.type, ac.uuid " \
  272. "FROM aclk_chart_%s ac, aclk_chart_payload_%s acp " \
  273. "WHERE ac.date_submitted IS NULL AND ac.unique_id = acp.unique_id AND ac.update_count > 0 " \
  274. "AND acp.claim_id = @claim_id ORDER BY ac.sequence_id ASC LIMIT %d;", wc->uuid_str, wc->uuid_str, limit);
  275. rc = prepare_statement(db_meta, sql, &res);
  276. if (rc != SQLITE_OK) {
  277. error_report("Failed to prepare statement when trying to send a chart update via ACLK");
  278. freez(claim_id);
  279. return;
  280. }
  281. }
  282. rc = sqlite3_bind_blob(res, 1, claim_uuid, sizeof(claim_uuid), SQLITE_STATIC);
  283. if (unlikely(rc != SQLITE_OK))
  284. goto bind_fail;
  285. char **payload_list = callocz(limit + 1, sizeof(char *));
  286. size_t *payload_list_size = callocz(limit + 1, sizeof(size_t));
  287. size_t *payload_list_max_size = callocz(limit + 1, sizeof(size_t));
  288. struct aclk_message_position *position_list = callocz(limit + 1, sizeof(*position_list));
  289. int *is_dim = callocz(limit + 1, sizeof(*is_dim));
  290. int loop = cmd.param1;
  291. uint64_t start_sequence_id = wc->chart_sequence_id;
  292. while (loop > 0) {
  293. uint64_t previous_sequence_id = wc->chart_sequence_id;
  294. int count = 0;
  295. first_sequence = 0;
  296. last_sequence = 0;
  297. while (count < limit && sqlite3_step(res) == SQLITE_ROW) {
  298. size_t payload_size = sqlite3_column_bytes(res, 1);
  299. if (payload_list_max_size[count] < payload_size) {
  300. freez(payload_list[count]);
  301. payload_list_max_size[count] = payload_size;
  302. payload_list[count] = mallocz(payload_size);
  303. }
  304. payload_list_size[count] = payload_size;
  305. memcpy(payload_list[count], sqlite3_column_blob(res, 1), payload_size);
  306. position_list[count].sequence_id = (uint64_t)sqlite3_column_int64(res, 0);
  307. position_list[count].previous_sequence_id = previous_sequence_id;
  308. position_list[count].seq_id_creation_time.tv_sec = sqlite3_column_int64(res, 2);
  309. position_list[count].seq_id_creation_time.tv_usec = 0;
  310. if (!first_sequence)
  311. first_sequence = position_list[count].sequence_id;
  312. last_sequence = position_list[count].sequence_id;
  313. last_timestamp = position_list[count].seq_id_creation_time.tv_sec;
  314. previous_sequence_id = last_sequence;
  315. is_dim[count] = sqlite3_column_int(res, 3) > 0;
  316. count++;
  317. if (wc->chart_payload_count)
  318. wc->chart_payload_count--;
  319. }
  320. freez(payload_list[count]);
  321. payload_list_max_size[count] = 0;
  322. payload_list[count] = NULL;
  323. rc = sqlite3_reset(res);
  324. if (unlikely(rc != SQLITE_OK))
  325. error_report("Failed to reset statement when pushing chart events, rc = %d", rc);
  326. if (likely(first_sequence)) {
  327. db_lock();
  328. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1, "UPDATE aclk_chart_%s SET status = NULL, date_submitted=unixepoch() "
  329. "WHERE date_submitted IS NULL AND sequence_id BETWEEN %" PRIu64 " AND %" PRIu64 ";",
  330. wc->uuid_str, first_sequence, last_sequence);
  331. db_execute(sql);
  332. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1, "INSERT OR REPLACE INTO aclk_chart_latest_%s (uuid, unique_id, date_submitted) "
  333. " SELECT uuid, unique_id, date_submitted FROM aclk_chart_%s s "
  334. " WHERE date_submitted IS NOT NULL AND sequence_id BETWEEN %" PRIu64 " AND %" PRIu64
  335. " ;",
  336. wc->uuid_str, wc->uuid_str, first_sequence, last_sequence);
  337. db_execute(sql);
  338. db_unlock();
  339. aclk_chart_inst_and_dim_update(payload_list, payload_list_size, is_dim, position_list, wc->batch_id);
  340. log_access(
  341. "ACLK RES [%s (%s)]: CHARTS SENT from %" PRIu64 " to %" PRIu64 " batch=%" PRIu64,
  342. wc->node_id,
  343. wc->hostname ? wc->hostname : "N/A",
  344. first_sequence,
  345. last_sequence,
  346. wc->batch_id);
  347. wc->chart_sequence_id = last_sequence;
  348. wc->chart_timestamp = last_timestamp;
  349. } else
  350. break;
  351. --loop;
  352. }
  353. if (start_sequence_id != wc->chart_sequence_id) {
  354. time_t now = now_realtime_sec();
  355. if (wc->rotation_after > now && wc->rotation_after < now + ACLK_DATABASE_ROTATION_DELAY)
  356. wc->rotation_after = now + ACLK_DATABASE_ROTATION_DELAY;
  357. } else {
  358. wc->chart_payload_count = sql_get_pending_count(wc);
  359. if (!wc->chart_payload_count)
  360. log_access(
  361. "ACLK STA [%s (%s)]: Sync of charts and dimensions done in %ld seconds.",
  362. wc->node_id,
  363. wc->hostname ? wc->hostname : "N/A",
  364. now_realtime_sec() - wc->startup_time);
  365. }
  366. for (int i = 0; i <= limit; ++i)
  367. freez(payload_list[i]);
  368. freez(payload_list);
  369. freez(payload_list_size);
  370. freez(payload_list_max_size);
  371. freez(position_list);
  372. freez(is_dim);
  373. bind_fail:
  374. rc = sqlite3_reset(res);
  375. if (unlikely(rc != SQLITE_OK))
  376. error_report("Failed to reset statement when pushing chart events, rc = %d", rc);
  377. freez(claim_id);
  378. return;
  379. }
  380. // Push one chart config to the cloud
  381. int aclk_send_chart_config(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  382. {
  383. UNUSED(wc);
  384. CHECK_SQLITE_CONNECTION(db_meta);
  385. sqlite3_stmt *res = NULL;
  386. int rc = 0;
  387. char *hash_id = (char *) cmd.data_param;
  388. uuid_t hash_uuid;
  389. rc = uuid_parse(hash_id, hash_uuid);
  390. if (unlikely(rc)) {
  391. freez((char *) cmd.data_param);
  392. return 1;
  393. }
  394. BUFFER *sql = buffer_create(1024);
  395. buffer_sprintf(sql, "SELECT type, family, context, title, priority, plugin, module, unit, chart_type " \
  396. "FROM chart_hash WHERE hash_id = @hash_id;");
  397. rc = sqlite3_prepare_v2(db_meta, buffer_tostring(sql), -1, &res, 0);
  398. if (rc != SQLITE_OK) {
  399. error_report("Failed to prepare statement when trying to fetch a chart hash configuration");
  400. goto fail;
  401. }
  402. rc = sqlite3_bind_blob(res, 1, &hash_uuid , sizeof(hash_uuid), SQLITE_STATIC);
  403. if (unlikely(rc != SQLITE_OK))
  404. goto bind_fail;
  405. struct chart_config_updated chart_config;
  406. chart_config.config_hash = NULL;
  407. while (sqlite3_step(res) == SQLITE_ROW) {
  408. chart_config.type = strdupz((char *)sqlite3_column_text(res, 0));
  409. chart_config.family = strdupz((char *)sqlite3_column_text(res, 1));
  410. chart_config.context = strdupz((char *)sqlite3_column_text(res, 2));
  411. chart_config.title = strdupz((char *)sqlite3_column_text(res, 3));
  412. chart_config.priority = sqlite3_column_int64(res, 4);
  413. chart_config.plugin = strdupz((char *)sqlite3_column_text(res, 5));
  414. chart_config.module = sqlite3_column_bytes(res, 6) > 0 ? strdupz((char *)sqlite3_column_text(res, 6)) : NULL;
  415. chart_config.chart_type = (RRDSET_TYPE) sqlite3_column_int(res,8);
  416. chart_config.units = strdupz((char *)sqlite3_column_text(res, 7));
  417. chart_config.config_hash = strdupz(hash_id);
  418. }
  419. if (likely(chart_config.config_hash)) {
  420. log_access(
  421. "ACLK REQ [%s (%s)]: Sending chart config for %s.",
  422. wc->node_id,
  423. wc->host ? wc->host->hostname : "N/A",
  424. hash_id);
  425. aclk_chart_config_updated(&chart_config, 1);
  426. destroy_chart_config_updated(&chart_config);
  427. } else
  428. log_access(
  429. "ACLK STA [%s (%s)]: Chart config for %s not found.",
  430. wc->node_id,
  431. wc->host ? wc->host->hostname : "N/A",
  432. hash_id);
  433. bind_fail:
  434. rc = sqlite3_finalize(res);
  435. if (unlikely(rc != SQLITE_OK))
  436. error_report("Failed to reset statement when pushing chart config hash, rc = %d", rc);
  437. fail:
  438. freez((char *)cmd.data_param);
  439. buffer_free(sql);
  440. return rc;
  441. }
  442. void aclk_receive_chart_ack(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  443. {
  444. int rc;
  445. sqlite3_stmt *res = NULL;
  446. char sql[ACLK_SYNC_QUERY_SIZE];
  447. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1,"UPDATE aclk_chart_%s SET date_updated=unixepoch() WHERE sequence_id <= @sequence_id "
  448. "AND date_submitted IS NOT NULL AND date_updated IS NULL;", wc->uuid_str);
  449. rc = sqlite3_prepare_v2(db_meta, sql, -1, &res, 0);
  450. if (rc != SQLITE_OK) {
  451. error_report("Failed to prepare statement to ack chart sequence ids");
  452. return;
  453. }
  454. rc = sqlite3_bind_int64(res, 1, (uint64_t) cmd.param1);
  455. if (unlikely(rc != SQLITE_OK))
  456. goto bind_fail;
  457. rc = execute_insert(res);
  458. if (rc != SQLITE_DONE)
  459. error_report("Failed to ACK sequence id, rc = %d", rc);
  460. else
  461. log_access(
  462. "ACLK STA [%s (%s)]: CHARTS ACKNOWLEDGED IN THE DATABASE UP TO %" PRIu64,
  463. wc->node_id,
  464. wc->host ? wc->host->hostname : "N/A",
  465. cmd.param1);
  466. bind_fail:
  467. if (unlikely(sqlite3_finalize(res) != SQLITE_OK))
  468. error_report("Failed to finalize statement to ACK older sequence ids, rc = %d", rc);
  469. return;
  470. }
  471. void aclk_receive_chart_reset(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  472. {
  473. BUFFER *sql = buffer_create(1024);
  474. buffer_sprintf(
  475. sql,
  476. "UPDATE aclk_chart_%s SET status = NULL, date_submitted = NULL WHERE sequence_id >= %" PRIu64 ";",
  477. wc->uuid_str,
  478. cmd.param1);
  479. db_execute(buffer_tostring(sql));
  480. if (cmd.param1 == 1) {
  481. buffer_flush(sql);
  482. log_access("ACLK REQ [%s (%s)]: Received chart full resync.", wc->node_id, wc->hostname ? wc->hostname: "N/A");
  483. buffer_sprintf(sql, "DELETE FROM aclk_chart_payload_%s; DELETE FROM aclk_chart_%s; " \
  484. "DELETE FROM aclk_chart_latest_%s;", wc->uuid_str, wc->uuid_str, wc->uuid_str);
  485. db_lock();
  486. db_execute("BEGIN TRANSACTION;");
  487. db_execute(buffer_tostring(sql));
  488. db_execute("COMMIT TRANSACTION;");
  489. db_unlock();
  490. wc->chart_sequence_id = 0;
  491. wc->chart_timestamp = 0;
  492. wc->chart_payload_count = 0;
  493. RRDHOST *host = wc->host;
  494. if (likely(host)) {
  495. rrdhost_rdlock(host);
  496. RRDSET *st;
  497. rrdset_foreach_read(st, host)
  498. {
  499. rrdset_rdlock(st);
  500. rrdset_flag_clear(st, RRDSET_FLAG_ACLK);
  501. RRDDIM *rd;
  502. rrddim_foreach_read(rd, st)
  503. {
  504. rrddim_flag_clear(rd, RRDDIM_FLAG_ACLK);
  505. rd->aclk_live_status = (rd->aclk_live_status == 0);
  506. }
  507. rrdset_unlock(st);
  508. }
  509. rrdhost_unlock(host);
  510. } else
  511. error_report("ACLK synchronization thread for %s is not linked to HOST", wc->host_guid);
  512. } else {
  513. log_access(
  514. "ACLK STA [%s (%s)]: RESTARTING CHART SYNC FROM SEQUENCE %" PRIu64,
  515. wc->node_id,
  516. wc->hostname ? wc->hostname : "N/A",
  517. cmd.param1);
  518. wc->chart_payload_count = sql_get_pending_count(wc);
  519. sql_get_last_chart_sequence(wc);
  520. }
  521. buffer_free(sql);
  522. wc->chart_updates = 1;
  523. return;
  524. }
  525. //
  526. // Functions called directly from ACLK threads and will queue commands
  527. //
  528. void aclk_get_chart_config(char **hash_id)
  529. {
  530. struct aclk_database_worker_config *wc = (struct aclk_database_worker_config *)localhost->dbsync_worker;
  531. if (unlikely(!wc || !hash_id))
  532. return;
  533. struct aclk_database_cmd cmd;
  534. memset(&cmd, 0, sizeof(cmd));
  535. cmd.opcode = ACLK_DATABASE_PUSH_CHART_CONFIG;
  536. for (int i = 0; hash_id[i]; ++i) {
  537. // TODO: Verify that we have a valid hash_id
  538. log_access(
  539. "ACLK REQ [%s (%s)]: Request %d for chart config with hash %s received.",
  540. wc->node_id,
  541. wc->host ? wc->host->hostname : "N/A",
  542. i,
  543. hash_id[i]);
  544. cmd.data_param = (void *)strdupz(hash_id[i]);
  545. aclk_database_enq_cmd(wc, &cmd);
  546. }
  547. return;
  548. }
  549. // Send a command to a node_id
  550. // Need to discover the thread that will handle the request
  551. // if thread not in active hosts, then try to find in the queue
  552. static void aclk_submit_param_command(char *node_id, enum aclk_database_opcode aclk_command, uint64_t param)
  553. {
  554. if (unlikely(!node_id))
  555. return;
  556. struct aclk_database_worker_config *wc = NULL;
  557. struct aclk_database_cmd cmd;
  558. memset(&cmd, 0, sizeof(cmd));
  559. cmd.opcode = aclk_command;
  560. cmd.param1 = param;
  561. rrd_rdlock();
  562. RRDHOST *host = find_host_by_node_id(node_id);
  563. if (likely(host))
  564. wc = (struct aclk_database_worker_config *)host->dbsync_worker;
  565. rrd_unlock();
  566. if (wc)
  567. aclk_database_enq_cmd(wc, &cmd);
  568. else {
  569. if (aclk_worker_enq_cmd(node_id, &cmd))
  570. log_access("ACLK STA [%s (N/A)]: ACLK synchronization thread is not active.", node_id);
  571. }
  572. return;
  573. }
  574. void aclk_ack_chart_sequence_id(char *node_id, uint64_t last_sequence_id)
  575. {
  576. if (unlikely(!node_id))
  577. return;
  578. char *hostname = get_hostname_by_node_id(node_id);
  579. log_access("ACLK REQ [%s (%s)]: CHARTS ACKNOWLEDGED upto %" PRIu64, node_id, hostname ? hostname : "N/A",
  580. last_sequence_id);
  581. freez(hostname);
  582. aclk_submit_param_command(node_id, ACLK_DATABASE_CHART_ACK, last_sequence_id);
  583. return;
  584. }
  585. // Start streaming charts / dimensions for node_id
  586. void aclk_start_streaming(char *node_id, uint64_t sequence_id, time_t created_at, uint64_t batch_id)
  587. {
  588. UNUSED(created_at);
  589. if (unlikely(!node_id))
  590. return;
  591. uuid_t node_uuid;
  592. if (uuid_parse(node_id, node_uuid)) {
  593. log_access("ACLK REQ [%s (N/A)]: CHARTS STREAM ignored, invalid node id", node_id);
  594. return;
  595. }
  596. struct aclk_database_worker_config *wc = find_inactive_wc_by_node_id(node_id);
  597. rrd_rdlock();
  598. RRDHOST *host = localhost;
  599. while(host) {
  600. if (wc || (host->node_id && !(uuid_compare(*host->node_id, node_uuid)))) {
  601. rrd_unlock();
  602. if (!wc)
  603. wc = (struct aclk_database_worker_config *)host->dbsync_worker ?
  604. (struct aclk_database_worker_config *)host->dbsync_worker :
  605. (struct aclk_database_worker_config *)find_inactive_wc_by_node_id(node_id);
  606. if (likely(wc)) {
  607. wc->chart_reset_count++;
  608. __sync_synchronize();
  609. wc->chart_updates = 0;
  610. wc->batch_id = batch_id;
  611. __sync_synchronize();
  612. wc->batch_created = now_realtime_sec();
  613. log_access(
  614. "ACLK REQ [%s (%s)]: CHARTS STREAM from %"PRIu64" (LOCAL %"PRIu64") t=%ld resets=%d" ,
  615. wc->node_id,
  616. wc->hostname ? wc->hostname : "N/A",
  617. sequence_id + 1,
  618. wc->chart_sequence_id,
  619. wc->chart_timestamp,
  620. wc->chart_reset_count);
  621. if (sequence_id > wc->chart_sequence_id || wc->chart_reset_count > 10) {
  622. log_access(
  623. "ACLK RES [%s (%s)]: CHARTS FULL RESYNC REQUEST "
  624. "remote_seq=%" PRIu64 " local_seq=%" PRIu64 " resets=%d ",
  625. wc->node_id,
  626. wc->hostname ? wc->hostname : "N/A",
  627. sequence_id,
  628. wc->chart_sequence_id,
  629. wc->chart_reset_count);
  630. chart_reset_t chart_reset;
  631. chart_reset.claim_id = get_agent_claimid();
  632. if (chart_reset.claim_id) {
  633. chart_reset.node_id = node_id;
  634. chart_reset.reason = SEQ_ID_NOT_EXISTS;
  635. aclk_chart_reset(chart_reset);
  636. freez(chart_reset.claim_id);
  637. wc->chart_reset_count = -1;
  638. }
  639. } else {
  640. struct aclk_database_cmd cmd;
  641. memset(&cmd, 0, sizeof(cmd));
  642. // TODO: handle timestamp
  643. if (sequence_id < wc->chart_sequence_id ||
  644. !sequence_id) { // || created_at != wc->chart_timestamp) {
  645. log_access(
  646. "ACLK REQ [%s (%s)]: CHART RESET from %" PRIu64 " t=%ld batch=%" PRIu64,
  647. wc->node_id,
  648. wc->hostname ? wc->hostname : "N/A",
  649. sequence_id + 1,
  650. wc->chart_timestamp,
  651. wc->batch_id);
  652. cmd.opcode = ACLK_DATABASE_RESET_CHART;
  653. cmd.param1 = sequence_id + 1;
  654. cmd.completion = NULL;
  655. aclk_database_enq_cmd(wc, &cmd);
  656. } else {
  657. wc->chart_reset_count = 0;
  658. wc->chart_updates = 1;
  659. }
  660. }
  661. } else {
  662. log_access("ACLK STA [%s (%s)]: ACLK synchronization thread is not active.", node_id, wc->hostname ? wc->hostname : "N/A");
  663. }
  664. return;
  665. }
  666. host = host->next;
  667. }
  668. rrd_unlock();
  669. return;
  670. }
  671. #define SQL_SELECT_HOST_MEMORY_MODE "SELECT memory_mode FROM chart WHERE host_id = @host_id LIMIT 1;"
  672. static RRD_MEMORY_MODE sql_get_host_memory_mode(uuid_t *host_id)
  673. {
  674. int rc;
  675. RRD_MEMORY_MODE memory_mode = RRD_MEMORY_MODE_RAM;
  676. sqlite3_stmt *res = NULL;
  677. rc = sqlite3_prepare_v2(db_meta, SQL_SELECT_HOST_MEMORY_MODE, -1, &res, 0);
  678. if (unlikely(rc != SQLITE_OK)) {
  679. error_report("Failed to prepare statement to read host memory mode");
  680. return memory_mode;
  681. }
  682. rc = sqlite3_bind_blob(res, 1, host_id, sizeof(*host_id), SQLITE_STATIC);
  683. if (unlikely(rc != SQLITE_OK)) {
  684. error_report("Failed to bind host parameter to fetch host memory mode");
  685. goto failed;
  686. }
  687. while (sqlite3_step(res) == SQLITE_ROW) {
  688. memory_mode = (RRD_MEMORY_MODE)sqlite3_column_int(res, 0);
  689. }
  690. failed:
  691. rc = sqlite3_finalize(res);
  692. if (unlikely(rc != SQLITE_OK))
  693. error_report("Failed to finalize the prepared statement when reading host memory mode");
  694. return memory_mode;
  695. }
  696. #define SELECT_HOST_DIMENSION_LIST \
  697. "SELECT d.dim_id, c.update_every, c.type||'.'||c.id, d.id, d.name FROM chart c, dimension d " \
  698. "WHERE d.chart_id = c.chart_id AND c.host_id = @host_id ORDER BY c.update_every ASC;"
  699. #define SELECT_HOST_CHART_LIST \
  700. "SELECT distinct h.host_id, c.update_every, c.type||'.'||c.id FROM chart c, host h " \
  701. "WHERE c.host_id = h.host_id AND c.host_id = @host_id ORDER BY c.update_every ASC;"
  702. void aclk_update_retention(struct aclk_database_worker_config *wc)
  703. {
  704. int rc;
  705. if (!aclk_connected)
  706. return;
  707. char *claim_id = get_agent_claimid();
  708. if (unlikely(!claim_id))
  709. return;
  710. sqlite3_stmt *res = NULL;
  711. RRD_MEMORY_MODE memory_mode;
  712. uuid_t host_uuid;
  713. rc = uuid_parse(wc->host_guid, host_uuid);
  714. if (unlikely(rc)) {
  715. freez(claim_id);
  716. return;
  717. }
  718. if (wc->host)
  719. memory_mode = wc->host->rrd_memory_mode;
  720. else
  721. memory_mode = sql_get_host_memory_mode(&host_uuid);
  722. if (memory_mode == RRD_MEMORY_MODE_DBENGINE)
  723. rc = sqlite3_prepare_v2(db_meta, SELECT_HOST_DIMENSION_LIST, -1, &res, 0);
  724. else
  725. rc = sqlite3_prepare_v2(db_meta, SELECT_HOST_CHART_LIST, -1, &res, 0);
  726. if (unlikely(rc != SQLITE_OK)) {
  727. error_report("Failed to prepare statement to fetch host dimensions");
  728. freez(claim_id);
  729. return;
  730. }
  731. rc = sqlite3_bind_blob(res, 1, &host_uuid, sizeof(host_uuid), SQLITE_STATIC);
  732. if (unlikely(rc != SQLITE_OK)) {
  733. error_report("Failed to bind host parameter to fetch host dimensions");
  734. goto failed;
  735. }
  736. time_t start_time = LONG_MAX;
  737. time_t first_entry_t;
  738. time_t last_entry_t;
  739. uint32_t update_every = 0;
  740. uint32_t dimension_update_count = 0;
  741. uint32_t total_checked = 0;
  742. uint32_t total_deleted= 0;
  743. uint32_t total_stopped= 0;
  744. time_t send_status;
  745. struct retention_updated rotate_data;
  746. memset(&rotate_data, 0, sizeof(rotate_data));
  747. int max_intervals = 32;
  748. rotate_data.interval_duration_count = 0;
  749. rotate_data.interval_durations = callocz(max_intervals, sizeof(*rotate_data.interval_durations));
  750. now_realtime_timeval(&rotate_data.rotation_timestamp);
  751. rotate_data.memory_mode = memory_mode;
  752. rotate_data.claim_id = claim_id;
  753. rotate_data.node_id = strdupz(wc->node_id);
  754. time_t now = now_realtime_sec();
  755. while (sqlite3_step(res) == SQLITE_ROW && dimension_update_count < ACLK_MAX_DIMENSION_CLEANUP) {
  756. if (unlikely(netdata_exit))
  757. break;
  758. if (!update_every || update_every != (uint32_t)sqlite3_column_int(res, 1)) {
  759. if (update_every) {
  760. debug(D_ACLK_SYNC, "Update %s for %u oldest time = %ld", wc->host_guid, update_every, start_time);
  761. if (start_time == LONG_MAX)
  762. rotate_data.interval_durations[rotate_data.interval_duration_count].retention = 0;
  763. else
  764. rotate_data.interval_durations[rotate_data.interval_duration_count].retention =
  765. rotate_data.rotation_timestamp.tv_sec - start_time;
  766. rotate_data.interval_duration_count++;
  767. }
  768. update_every = (uint32_t)sqlite3_column_int(res, 1);
  769. rotate_data.interval_durations[rotate_data.interval_duration_count].update_every = update_every;
  770. start_time = LONG_MAX;
  771. }
  772. #ifdef ENABLE_DBENGINE
  773. if (memory_mode == RRD_MEMORY_MODE_DBENGINE)
  774. rc =
  775. rrdeng_metric_latest_time_by_uuid((uuid_t *)sqlite3_column_blob(res, 0), &first_entry_t, &last_entry_t, 0);
  776. else
  777. #endif
  778. {
  779. if (wc->host) {
  780. RRDSET *st = NULL;
  781. rc = (st = rrdset_find(wc->host, (const char *)sqlite3_column_text(res, 2))) ? 0 : 1;
  782. if (!rc) {
  783. first_entry_t = rrdset_first_entry_t(st);
  784. last_entry_t = rrdset_last_entry_t(st);
  785. }
  786. } else {
  787. rc = 0;
  788. first_entry_t = rotate_data.rotation_timestamp.tv_sec;
  789. }
  790. }
  791. if (likely(!rc && first_entry_t))
  792. start_time = MIN(start_time, first_entry_t);
  793. if (memory_mode == RRD_MEMORY_MODE_DBENGINE && wc->chart_updates && (dimension_update_count < ACLK_MAX_DIMENSION_CLEANUP)) {
  794. int live = ((now - last_entry_t) < (RRDSET_MINIMUM_DIM_LIVE_MULTIPLIER * update_every));
  795. if (rc) {
  796. first_entry_t = 0;
  797. last_entry_t = 0;
  798. live = 0;
  799. }
  800. if (!wc->host || !first_entry_t) {
  801. if (!first_entry_t) {
  802. delete_dimension_uuid((uuid_t *)sqlite3_column_blob(res, 0));
  803. total_deleted++;
  804. dimension_update_count++;
  805. }
  806. else {
  807. (void)aclk_upd_dimension_event(
  808. wc,
  809. claim_id,
  810. (uuid_t *)sqlite3_column_blob(res, 0),
  811. (const char *)(const char *)sqlite3_column_text(res, 3),
  812. (const char *)(const char *)sqlite3_column_text(res, 4),
  813. (const char *)(const char *)sqlite3_column_text(res, 2),
  814. first_entry_t,
  815. live ? 0 : last_entry_t,
  816. &send_status);
  817. if (!send_status) {
  818. if (last_entry_t)
  819. total_stopped++;
  820. dimension_update_count++;
  821. }
  822. }
  823. }
  824. }
  825. total_checked++;
  826. }
  827. if (update_every) {
  828. debug(D_ACLK_SYNC, "Update %s for %u oldest time = %ld", wc->host_guid, update_every, start_time);
  829. if (start_time == LONG_MAX)
  830. rotate_data.interval_durations[rotate_data.interval_duration_count].retention = 0;
  831. else
  832. rotate_data.interval_durations[rotate_data.interval_duration_count].retention =
  833. rotate_data.rotation_timestamp.tv_sec - start_time;
  834. rotate_data.interval_duration_count++;
  835. }
  836. if (dimension_update_count < ACLK_MAX_DIMENSION_CLEANUP && !netdata_exit)
  837. log_access("ACLK STA [%s (%s)]: UPDATES %d RETENTION MESSAGE SENT. CHECKED %u DIMENSIONS. %u DELETED, %u STOPPED COLLECTING",
  838. wc->node_id, wc->hostname ? wc->hostname : "N/A", wc->chart_updates, total_checked, total_deleted, total_stopped);
  839. else
  840. log_access("ACLK STA [%s (%s)]: UPDATES %d RETENTION MESSAGE NOT SENT. CHECKED %u DIMENSIONS. %u DELETED, %u STOPPED COLLECTING",
  841. wc->node_id, wc->hostname ? wc->hostname : "N/A", wc->chart_updates, total_checked, total_deleted, total_stopped);
  842. #ifdef NETDATA_INTERNAL_CHECKS
  843. info("Retention update for %s (chart updates = %d)", wc->host_guid, wc->chart_updates);
  844. for (int i = 0; i < rotate_data.interval_duration_count; ++i)
  845. info(
  846. "Update for host %s (node %s) for %u Retention = %u",
  847. wc->host_guid,
  848. wc->node_id,
  849. rotate_data.interval_durations[i].update_every,
  850. rotate_data.interval_durations[i].retention);
  851. #endif
  852. if (dimension_update_count < ACLK_MAX_DIMENSION_CLEANUP && !netdata_exit)
  853. aclk_retention_updated(&rotate_data);
  854. freez(rotate_data.node_id);
  855. freez(rotate_data.interval_durations);
  856. failed:
  857. freez(claim_id);
  858. rc = sqlite3_finalize(res);
  859. if (unlikely(rc != SQLITE_OK))
  860. error_report("Failed to finalize the prepared statement when reading host dimensions");
  861. return;
  862. }
  863. uint32_t sql_get_pending_count(struct aclk_database_worker_config *wc)
  864. {
  865. char sql[ACLK_SYNC_QUERY_SIZE];
  866. static __thread sqlite3_stmt *res = NULL;
  867. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1, "SELECT count(1) FROM aclk_chart_%s ac WHERE ac.date_submitted IS NULL;", wc->uuid_str);
  868. int rc;
  869. uint32_t chart_payload_count = 0;
  870. if (unlikely(!res)) {
  871. rc = prepare_statement(db_meta, sql, &res);
  872. if (rc != SQLITE_OK) {
  873. error_report("Failed to prepare statement to count pending messages");
  874. return 0;
  875. }
  876. }
  877. while (sqlite3_step(res) == SQLITE_ROW)
  878. chart_payload_count = (uint32_t) sqlite3_column_int(res, 0);
  879. rc = sqlite3_reset(res);
  880. if (unlikely(rc != SQLITE_OK))
  881. error_report("Failed to reset statement when fetching pending messages, rc = %d", rc);
  882. return chart_payload_count;
  883. }
  884. void sql_get_last_chart_sequence(struct aclk_database_worker_config *wc)
  885. {
  886. char sql[ACLK_SYNC_QUERY_SIZE];
  887. snprintfz(sql,ACLK_SYNC_QUERY_SIZE-1, "SELECT ac.sequence_id, ac.date_created FROM aclk_chart_%s ac " \
  888. "WHERE ac.date_submitted IS NOT NULL ORDER BY ac.sequence_id DESC LIMIT 1;", wc->uuid_str);
  889. int rc;
  890. sqlite3_stmt *res = NULL;
  891. rc = sqlite3_prepare_v2(db_meta, sql, -1, &res, 0);
  892. if (rc != SQLITE_OK) {
  893. error_report("Failed to prepare statement to find last chart sequence id");
  894. return;
  895. }
  896. wc->chart_sequence_id = 0;
  897. wc->chart_timestamp = 0;
  898. while (sqlite3_step(res) == SQLITE_ROW) {
  899. wc->chart_sequence_id = (uint64_t)sqlite3_column_int64(res, 0);
  900. wc->chart_timestamp = (time_t)sqlite3_column_int64(res, 1);
  901. }
  902. debug(D_ACLK_SYNC, "Node %s reports last sequence_id=%" PRIu64, wc->node_id, wc->chart_sequence_id);
  903. rc = sqlite3_finalize(res);
  904. if (unlikely(rc != SQLITE_OK))
  905. error_report("Failed to reset statement when fetching chart sequence info, rc = %d", rc);
  906. return;
  907. }
  908. void queue_dimension_to_aclk(RRDDIM *rd, time_t last_updated)
  909. {
  910. int live = !last_updated;
  911. if (likely(rd->aclk_live_status == live))
  912. return;
  913. time_t created_at = rd->tiers[0]->query_ops.oldest_time(rd->tiers[0]->db_metric_handle);
  914. if (unlikely(!created_at && rd->updated))
  915. created_at = rd->last_collected_time.tv_sec;
  916. rd->aclk_live_status = live;
  917. struct aclk_database_worker_config *wc = rd->rrdset->rrdhost->dbsync_worker;
  918. if (unlikely(!wc))
  919. return;
  920. char *claim_id = get_agent_claimid();
  921. if (unlikely(!claim_id))
  922. return;
  923. struct chart_dimension_updated dim_payload;
  924. memset(&dim_payload, 0, sizeof(dim_payload));
  925. dim_payload.node_id = wc->node_id;
  926. dim_payload.claim_id = claim_id;
  927. dim_payload.name = rd->name;
  928. dim_payload.id = rd->id;
  929. dim_payload.chart_id = rd->rrdset->id;
  930. dim_payload.created_at.tv_sec = created_at;
  931. dim_payload.last_timestamp.tv_sec = last_updated;
  932. size_t size = 0;
  933. char *payload = generate_chart_dimension_updated(&size, &dim_payload);
  934. freez(claim_id);
  935. if (unlikely(!payload))
  936. return;
  937. struct aclk_chart_dimension_data *aclk_cd_data = mallocz(sizeof(*aclk_cd_data));
  938. uuid_copy(aclk_cd_data->uuid, rd->metric_uuid);
  939. aclk_cd_data->payload = payload;
  940. aclk_cd_data->payload_size = size;
  941. aclk_cd_data->check_payload = 1;
  942. struct aclk_database_cmd cmd;
  943. memset(&cmd, 0, sizeof(cmd));
  944. cmd.opcode = ACLK_DATABASE_ADD_DIMENSION;
  945. cmd.data = aclk_cd_data;
  946. int rc = aclk_database_enq_cmd_noblock(wc, &cmd);
  947. if (unlikely(rc)) {
  948. freez(aclk_cd_data->payload);
  949. freez(aclk_cd_data);
  950. rd->aclk_live_status = !live;
  951. }
  952. return;
  953. }
  954. void aclk_send_dimension_update(RRDDIM *rd)
  955. {
  956. char *claim_id = get_agent_claimid();
  957. if (unlikely(!claim_id))
  958. return;
  959. time_t first_entry_t = rrddim_first_entry_t(rd);
  960. time_t last_entry_t = rrddim_last_entry_t(rd);
  961. time_t now = now_realtime_sec();
  962. int live = ((now - rd->last_collected_time.tv_sec) < (RRDSET_MINIMUM_DIM_LIVE_MULTIPLIER * rd->update_every));
  963. if (!live || rd->aclk_live_status != live || !first_entry_t) {
  964. (void)aclk_upd_dimension_event(
  965. rd->rrdset->rrdhost->dbsync_worker,
  966. claim_id,
  967. &rd->metric_uuid,
  968. rd->id,
  969. rd->name,
  970. rd->rrdset->id,
  971. first_entry_t,
  972. live ? 0 : last_entry_t,
  973. NULL);
  974. if (!first_entry_t)
  975. debug(
  976. D_ACLK_SYNC,
  977. "%s: Update dimension chart=%s dim=%s live=%d (%ld, %ld)",
  978. rd->rrdset->rrdhost->hostname,
  979. rd->rrdset->name,
  980. rd->name,
  981. live,
  982. first_entry_t,
  983. last_entry_t);
  984. else
  985. debug(
  986. D_ACLK_SYNC,
  987. "%s: Update dimension chart=%s dim=%s live=%d (%ld, %ld) collected %ld seconds ago",
  988. rd->rrdset->rrdhost->hostname,
  989. rd->rrdset->name,
  990. rd->name,
  991. live,
  992. first_entry_t,
  993. last_entry_t,
  994. now - last_entry_t);
  995. rd->aclk_live_status = live;
  996. }
  997. freez(claim_id);
  998. return;
  999. }
  1000. #define SQL_SEQ_NULL(result, n) sqlite3_column_type(result, n) == SQLITE_NULL ? 0 : sqlite3_column_int64(result, n)
  1001. struct aclk_chart_sync_stats *aclk_get_chart_sync_stats(RRDHOST *host)
  1002. {
  1003. struct aclk_chart_sync_stats *aclk_statistics = NULL;
  1004. struct aclk_database_worker_config *wc = NULL;
  1005. wc = (struct aclk_database_worker_config *)host->dbsync_worker;
  1006. if (!wc)
  1007. return NULL;
  1008. aclk_statistics = callocz(1, sizeof(struct aclk_chart_sync_stats));
  1009. aclk_statistics->updates = wc->chart_updates;
  1010. aclk_statistics->batch_id = wc->batch_id;
  1011. char host_uuid_fixed[GUID_LEN + 1];
  1012. strncpy(host_uuid_fixed, host->machine_guid, GUID_LEN);
  1013. host_uuid_fixed[GUID_LEN] = 0;
  1014. host_uuid_fixed[8] = '_';
  1015. host_uuid_fixed[13] = '_';
  1016. host_uuid_fixed[18] = '_';
  1017. host_uuid_fixed[23] = '_';
  1018. sqlite3_stmt *res = NULL;
  1019. BUFFER *sql = buffer_create(1024);
  1020. buffer_sprintf(sql, "SELECT min(sequence_id), max(sequence_id), 0 FROM aclk_chart_%s;", host_uuid_fixed);
  1021. buffer_sprintf(sql, "SELECT min(sequence_id), max(sequence_id), 0 FROM aclk_chart_%s WHERE date_submitted IS NULL;", host_uuid_fixed);
  1022. buffer_sprintf(sql, "SELECT min(sequence_id), max(sequence_id), 0 FROM aclk_chart_%s WHERE date_submitted IS NOT NULL;", host_uuid_fixed);
  1023. buffer_sprintf(sql, "SELECT min(sequence_id), max(sequence_id), 0 FROM aclk_chart_%s WHERE date_updated IS NOT NULL;", host_uuid_fixed);
  1024. buffer_sprintf(sql, "SELECT max(date_created), max(date_submitted), max(date_updated), 0 FROM aclk_chart_%s;", host_uuid_fixed);
  1025. int rc = sqlite3_prepare_v2(db_meta, buffer_tostring(sql), -1, &res, 0);
  1026. if (rc != SQLITE_OK) {
  1027. buffer_free(sql);
  1028. freez(aclk_statistics);
  1029. return NULL;
  1030. }
  1031. rc = sqlite3_step(res);
  1032. if (rc == SQLITE_ROW) {
  1033. aclk_statistics->min_seqid = SQL_SEQ_NULL(res, 0);
  1034. aclk_statistics->max_seqid = SQL_SEQ_NULL(res, 1);
  1035. }
  1036. rc = sqlite3_step(res);
  1037. if (rc == SQLITE_ROW) {
  1038. aclk_statistics->min_seqid_pend = SQL_SEQ_NULL(res, 0);
  1039. aclk_statistics->max_seqid_pend = SQL_SEQ_NULL(res, 1);
  1040. }
  1041. rc = sqlite3_step(res);
  1042. if (rc == SQLITE_ROW) {
  1043. aclk_statistics->min_seqid_sent = SQL_SEQ_NULL(res, 0);
  1044. aclk_statistics->max_seqid_sent = SQL_SEQ_NULL(res, 1);
  1045. }
  1046. rc = sqlite3_step(res);
  1047. if (rc == SQLITE_ROW) {
  1048. aclk_statistics->min_seqid_ack = SQL_SEQ_NULL(res, 0);
  1049. aclk_statistics->max_seqid_ack = SQL_SEQ_NULL(res, 1);
  1050. }
  1051. rc = sqlite3_step(res);
  1052. if (rc == SQLITE_ROW) {
  1053. aclk_statistics->min_seqid_ack = SQL_SEQ_NULL(res, 0);
  1054. aclk_statistics->max_seqid_ack = SQL_SEQ_NULL(res, 1);
  1055. }
  1056. rc = sqlite3_step(res);
  1057. if (rc == SQLITE_ROW) {
  1058. aclk_statistics->max_date_created = (time_t) SQL_SEQ_NULL(res, 0);
  1059. aclk_statistics->max_date_submitted = (time_t) SQL_SEQ_NULL(res, 1);
  1060. aclk_statistics->max_date_ack = (time_t) SQL_SEQ_NULL(res, 2);
  1061. }
  1062. rc = sqlite3_finalize(res);
  1063. if (unlikely(rc != SQLITE_OK))
  1064. error_report("Failed to finalize statement when fetching aclk sync statistics, rc = %d", rc);
  1065. buffer_free(sql);
  1066. return aclk_statistics;
  1067. }
  1068. void sql_check_chart_liveness(RRDSET *st) {
  1069. RRDDIM *rd;
  1070. if (unlikely(st->state->is_ar_chart))
  1071. return;
  1072. rrdset_rdlock(st);
  1073. if (unlikely(!rrdset_flag_check(st, RRDSET_FLAG_ACLK))) {
  1074. rrdset_unlock(st);
  1075. return;
  1076. }
  1077. if (unlikely(!rrdset_flag_check(st, RRDSET_FLAG_ACLK))) {
  1078. if (likely(st->dimensions && st->counter_done && !queue_chart_to_aclk(st))) {
  1079. debug(D_ACLK_SYNC,"Check chart liveness [%s] submit chart definition", st->name);
  1080. rrdset_flag_set(st, RRDSET_FLAG_ACLK);
  1081. }
  1082. }
  1083. else
  1084. debug(D_ACLK_SYNC,"Check chart liveness [%s] chart definition already submitted", st->name);
  1085. time_t mark = now_realtime_sec();
  1086. debug(D_ACLK_SYNC,"Check chart liveness [%s] scanning dimensions", st->name);
  1087. rrddim_foreach_read(rd, st) {
  1088. if (!rrddim_flag_check(rd, RRDDIM_FLAG_HIDDEN))
  1089. queue_dimension_to_aclk(rd, calc_dimension_liveness(rd, mark));
  1090. }
  1091. rrdset_unlock(st);
  1092. }
  1093. // ST is read locked
  1094. int queue_chart_to_aclk(RRDSET *st)
  1095. {
  1096. return sql_queue_chart_payload((struct aclk_database_worker_config *) st->rrdhost->dbsync_worker,
  1097. st, ACLK_DATABASE_ADD_CHART);
  1098. }
  1099. #endif //ENABLE_ACLK