sqlite_aclk_chart.c 47 KB

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