sqlite_aclk_alert.c 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "sqlite_functions.h"
  3. #include "sqlite_aclk_alert.h"
  4. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  5. #include "../../aclk/aclk_alarm_api.h"
  6. #include "../../aclk/aclk.h"
  7. #endif
  8. // will replace call to aclk_update_alarm in health/health_log.c
  9. // and handle both cases
  10. int sql_queue_alarm_to_aclk(RRDHOST *host, ALARM_ENTRY *ae)
  11. {
  12. //check aclk architecture and handle old json alarm update to cloud
  13. //include also the valid statuses for this case
  14. #ifdef ENABLE_ACLK
  15. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  16. if (!aclk_use_new_cloud_arch && aclk_connected) {
  17. #endif
  18. if ((ae->new_status == RRDCALC_STATUS_WARNING || ae->new_status == RRDCALC_STATUS_CRITICAL) ||
  19. ((ae->old_status == RRDCALC_STATUS_WARNING || ae->old_status == RRDCALC_STATUS_CRITICAL))) {
  20. aclk_update_alarm(host, ae);
  21. }
  22. #endif
  23. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  24. }
  25. if (!claimed())
  26. return 0;
  27. if (ae->flags & HEALTH_ENTRY_FLAG_ACLK_QUEUED)
  28. return 0;
  29. if (ae->new_status == RRDCALC_STATUS_REMOVED || ae->new_status == RRDCALC_STATUS_UNINITIALIZED)
  30. return 0;
  31. if (unlikely(!host->dbsync_worker))
  32. return 1;
  33. if (unlikely(uuid_is_null(ae->config_hash_id)))
  34. return 0;
  35. int rc = 0;
  36. CHECK_SQLITE_CONNECTION(db_meta);
  37. sqlite3_stmt *res_alert = NULL;
  38. char uuid_str[GUID_LEN + 1];
  39. uuid_unparse_lower_fix(&host->host_uuid, uuid_str);
  40. BUFFER *sql = buffer_create(1024);
  41. buffer_sprintf(
  42. sql,
  43. "INSERT INTO aclk_alert_%s (alert_unique_id, date_created) "
  44. "VALUES (@alert_unique_id, strftime('%%s')) on conflict (alert_unique_id) do nothing; ",
  45. uuid_str);
  46. rc = sqlite3_prepare_v2(db_meta, buffer_tostring(sql), -1, &res_alert, 0);
  47. if (unlikely(rc != SQLITE_OK)) {
  48. error_report("Failed to prepare statement to store alert event");
  49. buffer_free(sql);
  50. return 1;
  51. }
  52. rc = sqlite3_bind_int(res_alert, 1, ae->unique_id);
  53. if (unlikely(rc != SQLITE_OK))
  54. goto bind_fail;
  55. rc = execute_insert(res_alert);
  56. if (unlikely(rc != SQLITE_DONE)) {
  57. error_report("Failed to store alert event %u, rc = %d", ae->unique_id, rc);
  58. goto bind_fail;
  59. }
  60. ae->flags |= HEALTH_ENTRY_FLAG_ACLK_QUEUED;
  61. bind_fail:
  62. if (unlikely(sqlite3_finalize(res_alert) != SQLITE_OK))
  63. error_report("Failed to reset statement in store alert event, rc = %d", rc);
  64. buffer_free(sql);
  65. return 0;
  66. #else
  67. UNUSED(host);
  68. UNUSED(ae);
  69. #endif
  70. return 0;
  71. }
  72. int rrdcalc_status_to_proto_enum(RRDCALC_STATUS status)
  73. {
  74. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  75. switch(status) {
  76. case RRDCALC_STATUS_REMOVED:
  77. return ALARM_STATUS_REMOVED;
  78. case RRDCALC_STATUS_UNDEFINED:
  79. return ALARM_STATUS_NOT_A_NUMBER;
  80. case RRDCALC_STATUS_CLEAR:
  81. return ALARM_STATUS_CLEAR;
  82. case RRDCALC_STATUS_WARNING:
  83. return ALARM_STATUS_WARNING;
  84. case RRDCALC_STATUS_CRITICAL:
  85. return ALARM_STATUS_CRITICAL;
  86. default:
  87. return ALARM_STATUS_UNKNOWN;
  88. }
  89. #else
  90. UNUSED(status);
  91. return 1;
  92. #endif
  93. }
  94. void aclk_push_alert_event(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  95. {
  96. #ifndef ENABLE_NEW_CLOUD_PROTOCOL
  97. UNUSED(wc);
  98. UNUSED(cmd);
  99. #else
  100. int rc;
  101. if (unlikely(!wc->alert_updates)) {
  102. log_access("AC [%s (%s)]: Ignoring alert push event, updates have been turned off for this node.", wc->node_id, wc->host ? wc->host->hostname : "N/A");
  103. return;
  104. }
  105. char *claim_id = is_agent_claimed();
  106. if (unlikely(!claim_id))
  107. return;
  108. BUFFER *sql = buffer_create(1024);
  109. if (wc->alerts_start_seq_id != 0) {
  110. buffer_sprintf(
  111. sql,
  112. "UPDATE aclk_alert_%s SET date_submitted = NULL, date_cloud_ack = NULL WHERE sequence_id >= %"PRIu64
  113. "; UPDATE aclk_alert_%s SET date_cloud_ack = strftime('%%s','now') WHERE sequence_id < %"PRIu64
  114. " and date_cloud_ack is null "
  115. "; UPDATE aclk_alert_%s SET date_submitted = strftime('%%s','now') WHERE sequence_id < %"PRIu64
  116. " and date_submitted is null",
  117. wc->uuid_str,
  118. wc->alerts_start_seq_id,
  119. wc->uuid_str,
  120. wc->alerts_start_seq_id,
  121. wc->uuid_str,
  122. wc->alerts_start_seq_id);
  123. db_execute(buffer_tostring(sql));
  124. buffer_reset(sql);
  125. wc->alerts_start_seq_id = 0;
  126. }
  127. int limit = cmd.count > 0 ? cmd.count : 1;
  128. sqlite3_stmt *res = NULL;
  129. buffer_sprintf(sql, "select aa.sequence_id, hl.unique_id, hl.alarm_id, hl.config_hash_id, hl.updated_by_id, hl.when_key, \
  130. hl.duration, hl.non_clear_duration, hl.flags, hl.exec_run_timestamp, hl.delay_up_to_timestamp, hl.name, \
  131. hl.chart, hl.family, hl.exec, hl.recipient, hl.source, hl.units, hl.info, hl.exec_code, hl.new_status, \
  132. hl.old_status, hl.delay, hl.new_value, hl.old_value, hl.last_repeat \
  133. from health_log_%s hl, aclk_alert_%s aa \
  134. where hl.unique_id = aa.alert_unique_id and aa.date_submitted is null \
  135. order by aa.sequence_id asc limit %d;", wc->uuid_str, wc->uuid_str, limit);
  136. rc = sqlite3_prepare_v2(db_meta, buffer_tostring(sql), -1, &res, 0);
  137. if (rc != SQLITE_OK) {
  138. error_report("Failed to prepare statement when trying to send an alert update via ACLK");
  139. buffer_free(sql);
  140. freez(claim_id);
  141. return;
  142. }
  143. char uuid_str[GUID_LEN + 1];
  144. uint64_t first_sequence_id = 0;
  145. uint64_t last_sequence_id = 0;
  146. static __thread uint64_t log_first_sequence_id = 0;
  147. static __thread uint64_t log_last_sequence_id = 0;
  148. while (sqlite3_step(res) == SQLITE_ROW) {
  149. struct alarm_log_entry alarm_log;
  150. char old_value_string[100 + 1];
  151. char new_value_string[100 + 1];
  152. alarm_log.node_id = wc->node_id;
  153. alarm_log.claim_id = claim_id;
  154. alarm_log.chart = strdupz((char *)sqlite3_column_text(res, 12));
  155. alarm_log.name = strdupz((char *)sqlite3_column_text(res, 11));
  156. alarm_log.family = sqlite3_column_bytes(res, 13) > 0 ? strdupz((char *)sqlite3_column_text(res, 13)) : NULL;
  157. alarm_log.batch_id = wc->alerts_batch_id;
  158. alarm_log.sequence_id = (uint64_t) sqlite3_column_int64(res, 0);
  159. alarm_log.when = (time_t) sqlite3_column_int64(res, 5);
  160. uuid_unparse_lower(*((uuid_t *) sqlite3_column_blob(res, 3)), uuid_str);
  161. alarm_log.config_hash = strdupz((char *)uuid_str);
  162. alarm_log.utc_offset = wc->host->utc_offset;
  163. alarm_log.timezone = strdupz((char *)wc->host->abbrev_timezone);
  164. alarm_log.exec_path = sqlite3_column_bytes(res, 14) > 0 ? strdupz((char *)sqlite3_column_text(res, 14)) :
  165. strdupz((char *)wc->host->health_default_exec);
  166. alarm_log.conf_source = strdupz((char *)sqlite3_column_text(res, 16));
  167. char *edit_command = sqlite3_column_bytes(res, 16) > 0 ?
  168. health_edit_command_from_source((char *)sqlite3_column_text(res, 16)) :
  169. strdupz("UNKNOWN=0");
  170. alarm_log.command = strdupz(edit_command);
  171. alarm_log.duration = (time_t) sqlite3_column_int64(res, 6);
  172. alarm_log.non_clear_duration = (time_t) sqlite3_column_int64(res, 7);
  173. alarm_log.status = rrdcalc_status_to_proto_enum((RRDCALC_STATUS) sqlite3_column_int(res, 20));
  174. alarm_log.old_status = rrdcalc_status_to_proto_enum((RRDCALC_STATUS) sqlite3_column_int(res, 21));
  175. alarm_log.delay = (int) sqlite3_column_int(res, 22);
  176. alarm_log.delay_up_to_timestamp = (time_t) sqlite3_column_int64(res, 10);
  177. alarm_log.last_repeat = (time_t) sqlite3_column_int64(res, 25);
  178. alarm_log.silenced = ((sqlite3_column_int64(res, 8) & HEALTH_ENTRY_FLAG_SILENCED) ||
  179. (sqlite3_column_type(res, 15) != SQLITE_NULL &&
  180. !strncmp((char *)sqlite3_column_text(res, 15), "silent", 6))) ?
  181. 1 :
  182. 0;
  183. alarm_log.value_string =
  184. sqlite3_column_type(res, 23) == SQLITE_NULL ?
  185. strdupz((char *)"-") :
  186. strdupz((char *)format_value_and_unit(
  187. new_value_string, 100, sqlite3_column_double(res, 23), (char *)sqlite3_column_text(res, 17), -1));
  188. alarm_log.old_value_string =
  189. sqlite3_column_type(res, 24) == SQLITE_NULL ?
  190. strdupz((char *)"-") :
  191. strdupz((char *)format_value_and_unit(
  192. old_value_string, 100, sqlite3_column_double(res, 24), (char *)sqlite3_column_text(res, 17), -1));
  193. alarm_log.value = (calculated_number) sqlite3_column_double(res, 23);
  194. alarm_log.old_value = (calculated_number) sqlite3_column_double(res, 24);
  195. alarm_log.updated = (sqlite3_column_int64(res, 8) & HEALTH_ENTRY_FLAG_UPDATED) ? 1 : 0;
  196. alarm_log.rendered_info = strdupz((char *)sqlite3_column_text(res, 18));
  197. aclk_send_alarm_log_entry(&alarm_log);
  198. if (first_sequence_id == 0)
  199. first_sequence_id = (uint64_t) sqlite3_column_int64(res, 0);
  200. if (log_first_sequence_id == 0)
  201. log_first_sequence_id = (uint64_t) sqlite3_column_int64(res, 0);
  202. last_sequence_id = (uint64_t) sqlite3_column_int64(res, 0);
  203. log_last_sequence_id = (uint64_t) sqlite3_column_int64(res, 0);
  204. destroy_alarm_log_entry(&alarm_log);
  205. freez(edit_command);
  206. }
  207. if (first_sequence_id) {
  208. buffer_flush(sql);
  209. buffer_sprintf(sql, "UPDATE aclk_alert_%s SET date_submitted=strftime('%%s') "
  210. "WHERE date_submitted IS NULL AND sequence_id BETWEEN %" PRIu64 " AND %" PRIu64 ";",
  211. wc->uuid_str, first_sequence_id, last_sequence_id);
  212. db_execute(buffer_tostring(sql));
  213. } else {
  214. if (log_first_sequence_id)
  215. log_access("OG [%s (%s)]: Sent alert events, first sequence_id %"PRIu64", last sequence_id %"PRIu64, wc->node_id, wc->host ? wc->host->hostname : "N/A", log_first_sequence_id, log_last_sequence_id);
  216. log_first_sequence_id = 0;
  217. log_last_sequence_id = 0;
  218. }
  219. rc = sqlite3_finalize(res);
  220. if (unlikely(rc != SQLITE_OK))
  221. error_report("Failed to finalize statement to send alert entries from the database, rc = %d", rc);
  222. freez(claim_id);
  223. buffer_free(sql);
  224. #endif
  225. return;
  226. }
  227. void aclk_send_alarm_health_log(char *node_id)
  228. {
  229. if (unlikely(!node_id))
  230. return;
  231. log_access("IN [%s (N/A)]: Request to send alarm health log.", node_id);
  232. struct aclk_database_worker_config *wc = NULL;
  233. struct aclk_database_cmd cmd;
  234. memset(&cmd, 0, sizeof(cmd));
  235. cmd.opcode = ACLK_DATABASE_ALARM_HEALTH_LOG;
  236. rrd_wrlock();
  237. RRDHOST *host = find_host_by_node_id(node_id);
  238. if (likely(host))
  239. wc = (struct aclk_database_worker_config *)host->dbsync_worker;
  240. rrd_unlock();
  241. if (wc)
  242. aclk_database_enq_cmd(wc, &cmd);
  243. else {
  244. if (aclk_worker_enq_cmd(node_id, &cmd))
  245. error_report("ACLK synchronization thread is not active for node id %s", node_id);
  246. }
  247. return;
  248. }
  249. void aclk_push_alarm_health_log(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  250. {
  251. UNUSED(cmd);
  252. #ifndef ENABLE_NEW_CLOUD_PROTOCOL
  253. UNUSED(wc);
  254. #else
  255. int rc;
  256. char *claim_id = is_agent_claimed();
  257. if (unlikely(!claim_id))
  258. return;
  259. uint64_t first_sequence = 0;
  260. uint64_t last_sequence = 0;
  261. struct timeval first_timestamp;
  262. struct timeval last_timestamp;
  263. BUFFER *sql = buffer_create(1024);
  264. sqlite3_stmt *res = NULL;
  265. //TODO: make this better: include info from health log too
  266. buffer_sprintf(sql, "SELECT MIN(sequence_id), MIN(date_created), MAX(sequence_id), MAX(date_created) " \
  267. "FROM aclk_alert_%s;", wc->uuid_str);
  268. rc = sqlite3_prepare_v2(db_meta, buffer_tostring(sql), -1, &res, 0);
  269. if (rc != SQLITE_OK) {
  270. error_report("Failed to prepare statement to get health log statistics from the database");
  271. buffer_free(sql);
  272. freez(claim_id);
  273. return;
  274. }
  275. first_timestamp.tv_sec = 0;
  276. first_timestamp.tv_usec = 0;
  277. last_timestamp.tv_sec = 0;
  278. last_timestamp.tv_usec = 0;
  279. while (sqlite3_step(res) == SQLITE_ROW) {
  280. first_sequence = sqlite3_column_bytes(res, 0) > 0 ? (uint64_t) sqlite3_column_int64(res, 0) : 0;
  281. if (sqlite3_column_bytes(res, 1) > 0) {
  282. first_timestamp.tv_sec = sqlite3_column_int64(res, 1);
  283. }
  284. last_sequence = sqlite3_column_bytes(res, 2) > 0 ? (uint64_t) sqlite3_column_int64(res, 2) : 0;
  285. if (sqlite3_column_bytes(res, 3) > 0) {
  286. last_timestamp.tv_sec = sqlite3_column_int64(res, 3);
  287. }
  288. }
  289. struct alarm_log_entries log_entries;
  290. log_entries.first_seq_id = first_sequence;
  291. log_entries.first_when = first_timestamp;
  292. log_entries.last_seq_id = last_sequence;
  293. log_entries.last_when = last_timestamp;
  294. struct alarm_log_health alarm_log;
  295. alarm_log.claim_id = claim_id;
  296. alarm_log.node_id = wc->node_id;
  297. alarm_log.log_entries = log_entries;
  298. alarm_log.status = wc->alert_updates == 0 ? 2 : 1;
  299. wc->alert_sequence_id = last_sequence;
  300. aclk_send_alarm_log_health(&alarm_log);
  301. log_access("OG [%s (%s)]: Alarm health log sent, first sequence id %ld, last sequence id %ld.", wc->node_id, wc->host ? wc->host->hostname : "N/A", first_sequence, last_sequence);
  302. rc = sqlite3_finalize(res);
  303. if (unlikely(rc != SQLITE_OK))
  304. error_report("Failed to reset statement to get health log statistics from the database, rc = %d", rc);
  305. freez(claim_id);
  306. buffer_free(sql);
  307. #endif
  308. return;
  309. }
  310. void aclk_send_alarm_configuration(char *config_hash)
  311. {
  312. if (unlikely(!config_hash))
  313. return;
  314. struct aclk_database_worker_config *wc = (struct aclk_database_worker_config *) localhost->dbsync_worker;
  315. if (unlikely(!wc)) {
  316. return;
  317. }
  318. log_access("IN [%s (%s)]: Request to send alert config %s.", wc->node_id, wc->host ? wc->host->hostname : "N/A", config_hash);
  319. struct aclk_database_cmd cmd;
  320. memset(&cmd, 0, sizeof(cmd));
  321. cmd.opcode = ACLK_DATABASE_PUSH_ALERT_CONFIG;
  322. cmd.data_param = (void *) strdupz(config_hash);
  323. cmd.completion = NULL;
  324. aclk_database_enq_cmd(wc, &cmd);
  325. return;
  326. }
  327. #define SQL_SELECT_ALERT_CONFIG "SELECT alarm, template, on_key, class, type, component, os, hosts, plugin," \
  328. "module, charts, families, lookup, every, units, green, red, calc, warn, crit, to_key, exec, delay, repeat, info," \
  329. "options, host_labels, p_db_lookup_dimensions, p_db_lookup_method, p_db_lookup_options, p_db_lookup_after," \
  330. "p_db_lookup_before, p_update_every FROM alert_hash WHERE hash_id = @hash_id;"
  331. int aclk_push_alert_config_event(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  332. {
  333. UNUSED(wc);
  334. #ifndef ENABLE_NEW_CLOUD_PROTOCOL
  335. UNUSED(cmd);
  336. #else
  337. int rc = 0;
  338. CHECK_SQLITE_CONNECTION(db_meta);
  339. sqlite3_stmt *res = NULL;
  340. char *config_hash = (char *) cmd.data_param;
  341. rc = sqlite3_prepare_v2(db_meta, SQL_SELECT_ALERT_CONFIG, -1, &res, 0);
  342. if (rc != SQLITE_OK) {
  343. error_report("Failed to prepare statement when trying to fetch an alarm hash configuration");
  344. return 1;
  345. }
  346. uuid_t hash_uuid;
  347. if (uuid_parse(config_hash, hash_uuid))
  348. return 1;
  349. rc = sqlite3_bind_blob(res, 1, &hash_uuid , sizeof(hash_uuid), SQLITE_STATIC);
  350. if (unlikely(rc != SQLITE_OK))
  351. goto bind_fail;
  352. struct aclk_alarm_configuration alarm_config;
  353. struct provide_alarm_configuration p_alarm_config;
  354. p_alarm_config.cfg_hash = NULL;
  355. if (sqlite3_step(res) == SQLITE_ROW) {
  356. alarm_config.alarm = sqlite3_column_bytes(res, 0) > 0 ? strdupz((char *)sqlite3_column_text(res, 0)) : NULL;
  357. alarm_config.tmpl = sqlite3_column_bytes(res, 1) > 0 ? strdupz((char *)sqlite3_column_text(res, 1)) : NULL;
  358. alarm_config.on_chart = sqlite3_column_bytes(res, 2) > 0 ? strdupz((char *)sqlite3_column_text(res, 2)) : NULL;
  359. alarm_config.classification = sqlite3_column_bytes(res, 3) > 0 ? strdupz((char *)sqlite3_column_text(res, 3)) : NULL;
  360. alarm_config.type = sqlite3_column_bytes(res, 4) > 0 ? strdupz((char *)sqlite3_column_text(res, 4)) : NULL;
  361. alarm_config.component = sqlite3_column_bytes(res, 5) > 0 ? strdupz((char *)sqlite3_column_text(res, 5)) : NULL;
  362. alarm_config.os = sqlite3_column_bytes(res, 6) > 0 ? strdupz((char *)sqlite3_column_text(res, 6)) : NULL;
  363. alarm_config.hosts = sqlite3_column_bytes(res, 7) > 0 ? strdupz((char *)sqlite3_column_text(res, 7)) : NULL;
  364. alarm_config.plugin = sqlite3_column_bytes(res, 8) > 0 ? strdupz((char *)sqlite3_column_text(res, 8)) : NULL;
  365. alarm_config.module = sqlite3_column_bytes(res, 9) > 0 ? strdupz((char *)sqlite3_column_text(res, 9)) : NULL;
  366. alarm_config.charts = sqlite3_column_bytes(res, 10) > 0 ? strdupz((char *)sqlite3_column_text(res, 10)) : NULL;
  367. alarm_config.families = sqlite3_column_bytes(res, 11) > 0 ? strdupz((char *)sqlite3_column_text(res, 11)) : NULL;
  368. alarm_config.lookup = sqlite3_column_bytes(res, 12) > 0 ? strdupz((char *)sqlite3_column_text(res, 12)) : NULL;
  369. alarm_config.every = sqlite3_column_bytes(res, 13) > 0 ? strdupz((char *)sqlite3_column_text(res, 13)) : NULL;
  370. alarm_config.units = sqlite3_column_bytes(res, 14) > 0 ? strdupz((char *)sqlite3_column_text(res, 14)) : NULL;
  371. alarm_config.green = sqlite3_column_bytes(res, 15) > 0 ? strdupz((char *)sqlite3_column_text(res, 15)) : NULL;
  372. alarm_config.red = sqlite3_column_bytes(res, 16) > 0 ? strdupz((char *)sqlite3_column_text(res, 16)) : NULL;
  373. alarm_config.calculation_expr = sqlite3_column_bytes(res, 17) > 0 ? strdupz((char *)sqlite3_column_text(res, 17)) : NULL;
  374. alarm_config.warning_expr = sqlite3_column_bytes(res, 18) > 0 ? strdupz((char *)sqlite3_column_text(res, 18)) : NULL;
  375. alarm_config.critical_expr = sqlite3_column_bytes(res, 19) > 0 ? strdupz((char *)sqlite3_column_text(res, 19)) : NULL;
  376. alarm_config.recipient = sqlite3_column_bytes(res, 20) > 0 ? strdupz((char *)sqlite3_column_text(res, 20)) : NULL;
  377. alarm_config.exec = sqlite3_column_bytes(res, 21) > 0 ? strdupz((char *)sqlite3_column_text(res, 21)) : NULL;
  378. alarm_config.delay = sqlite3_column_bytes(res, 22) > 0 ? strdupz((char *)sqlite3_column_text(res, 22)) : NULL;
  379. alarm_config.repeat = sqlite3_column_bytes(res, 23) > 0 ? strdupz((char *)sqlite3_column_text(res, 23)) : NULL;
  380. alarm_config.info = sqlite3_column_bytes(res, 24) > 0 ? strdupz((char *)sqlite3_column_text(res, 24)) : NULL;
  381. alarm_config.options = sqlite3_column_bytes(res, 25) > 0 ? strdupz((char *)sqlite3_column_text(res, 25)) : NULL;
  382. alarm_config.host_labels = sqlite3_column_bytes(res, 26) > 0 ? strdupz((char *)sqlite3_column_text(res, 26)) : NULL;
  383. alarm_config.p_db_lookup_dimensions = NULL;
  384. alarm_config.p_db_lookup_method = NULL;
  385. alarm_config.p_db_lookup_options = NULL;
  386. alarm_config.p_db_lookup_after = 0;
  387. alarm_config.p_db_lookup_before = 0;
  388. if (sqlite3_column_bytes(res, 30) > 0) {
  389. alarm_config.p_db_lookup_dimensions = sqlite3_column_bytes(res, 27) > 0 ? strdupz((char *)sqlite3_column_text(res, 27)) : NULL;
  390. alarm_config.p_db_lookup_method = sqlite3_column_bytes(res, 28) > 0 ? strdupz((char *)sqlite3_column_text(res, 28)) : NULL;
  391. BUFFER *tmp_buf = buffer_create(1024);
  392. buffer_data_options2string(tmp_buf, sqlite3_column_int(res, 29));
  393. alarm_config.p_db_lookup_options = strdupz((char *)buffer_tostring(tmp_buf));
  394. buffer_free(tmp_buf);
  395. alarm_config.p_db_lookup_after = sqlite3_column_int(res, 30);
  396. alarm_config.p_db_lookup_before = sqlite3_column_int(res, 31);
  397. }
  398. alarm_config.p_update_every = sqlite3_column_int(res, 32);
  399. p_alarm_config.cfg_hash = strdupz((char *) config_hash);
  400. p_alarm_config.cfg = alarm_config;
  401. }
  402. if (likely(p_alarm_config.cfg_hash)) {
  403. log_access("OG [%s (%s)]: Sent alert config %s.", wc->node_id, wc->host ? wc->host->hostname : "N/A", config_hash);
  404. aclk_send_provide_alarm_cfg(&p_alarm_config);
  405. freez((char *) cmd.data_param);
  406. freez(p_alarm_config.cfg_hash);
  407. destroy_aclk_alarm_configuration(&alarm_config);
  408. }
  409. else
  410. info("Alert config for %s not found", config_hash);
  411. bind_fail:
  412. rc = sqlite3_finalize(res);
  413. if (unlikely(rc != SQLITE_OK))
  414. error_report("Failed to reset statement when pushing alarm config hash, rc = %d", rc);
  415. return rc;
  416. #endif
  417. return 0;
  418. }
  419. // Start streaming alerts
  420. void aclk_start_alert_streaming(char *node_id, uint64_t batch_id, uint64_t start_seq_id)
  421. {
  422. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  423. if (unlikely(!node_id))
  424. return;
  425. log_access("IN [%s (N/A)]: Start streaming alerts with batch_id %"PRIu64" and start_seq_id %"PRIu64".", node_id, batch_id, start_seq_id);
  426. uuid_t node_uuid;
  427. if (uuid_parse(node_id, node_uuid))
  428. return;
  429. struct aclk_database_worker_config *wc = NULL;
  430. rrd_wrlock();
  431. RRDHOST *host = find_host_by_node_id(node_id);
  432. if (likely(host))
  433. wc = (struct aclk_database_worker_config *)host->dbsync_worker;
  434. rrd_unlock();
  435. if (unlikely(!host->health_enabled)) {
  436. log_access("AC [%s (%s)]: Ignoring request to stream alert state changes, health is disabled.", node_id, wc->host ? wc->host->hostname : "N/A");
  437. return;
  438. }
  439. if (likely(wc)) {
  440. log_access("AC [%s (%s)]: Start streaming alerts enabled with batch_id %"PRIu64" and start_seq_id %"PRIu64".", node_id, wc->host ? wc->host->hostname : "N/A", batch_id, start_seq_id);
  441. __sync_synchronize();
  442. wc->alerts_batch_id = batch_id;
  443. wc->alerts_start_seq_id = start_seq_id;
  444. wc->alert_updates = 1;
  445. __sync_synchronize();
  446. }
  447. else
  448. log_access("AC [%s (%s)]: ACLK synchronization thread is not active.", node_id, wc->host ? wc->host->hostname : "N/A");
  449. #else
  450. UNUSED(node_id);
  451. UNUSED(start_seq_id);
  452. UNUSED(batch_id);
  453. #endif
  454. return;
  455. }
  456. void sql_process_queue_removed_alerts_to_aclk(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  457. {
  458. UNUSED(cmd);
  459. #ifndef ENABLE_NEW_CLOUD_PROTOCOL
  460. UNUSED(wc);
  461. #else
  462. BUFFER *sql = buffer_create(1024);
  463. buffer_sprintf(sql,"insert into aclk_alert_%s (alert_unique_id, date_created) " \
  464. "select unique_id alert_unique_id, strftime('%%s') date_created from health_log_%s " \
  465. "where new_status = -2 and updated_by_id = 0 and unique_id not in " \
  466. "(select alert_unique_id from aclk_alert_%s) order by unique_id asc " \
  467. "on conflict (alert_unique_id) do nothing;", wc->uuid_str, wc->uuid_str, wc->uuid_str);
  468. db_execute(buffer_tostring(sql));
  469. buffer_free(sql);
  470. #endif
  471. return;
  472. }
  473. void sql_queue_removed_alerts_to_aclk(RRDHOST *host)
  474. {
  475. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  476. if (unlikely(!host->dbsync_worker))
  477. return;
  478. struct aclk_database_cmd cmd;
  479. memset(&cmd, 0, sizeof(cmd));
  480. cmd.opcode = ACLK_DATABASE_QUEUE_REMOVED_ALERTS;
  481. cmd.data = NULL;
  482. cmd.data_param = NULL;
  483. cmd.completion = NULL;
  484. aclk_database_enq_cmd((struct aclk_database_worker_config *) host->dbsync_worker, &cmd);
  485. #else
  486. UNUSED(host);
  487. #endif
  488. }
  489. void aclk_process_send_alarm_snapshot(char *node_id, char *claim_id, uint64_t snapshot_id, uint64_t sequence_id)
  490. {
  491. UNUSED(claim_id);
  492. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  493. if (unlikely(!node_id))
  494. return;
  495. uuid_t node_uuid;
  496. if (uuid_parse(node_id, node_uuid))
  497. return;
  498. struct aclk_database_worker_config *wc = NULL;
  499. rrd_wrlock();
  500. RRDHOST *host = find_host_by_node_id(node_id);
  501. if (likely(host))
  502. wc = (struct aclk_database_worker_config *)host->dbsync_worker;
  503. rrd_unlock();
  504. if (likely(wc)) {
  505. log_access(
  506. "IN [%s (%s)]: Request to send alerts snapshot, snapshot_id %" PRIu64 " and ack_sequence_id %" PRIu64,
  507. wc->node_id,
  508. wc->host ? wc->host->hostname : "N/A",
  509. snapshot_id,
  510. sequence_id);
  511. __sync_synchronize();
  512. wc->alerts_snapshot_id = snapshot_id;
  513. wc->alerts_ack_sequence_id = sequence_id;
  514. __sync_synchronize();
  515. struct aclk_database_cmd cmd;
  516. memset(&cmd, 0, sizeof(cmd));
  517. cmd.opcode = ACLK_DATABASE_PUSH_ALERT_SNAPSHOT;
  518. cmd.data_param = NULL;
  519. cmd.completion = NULL;
  520. aclk_database_enq_cmd(wc, &cmd);
  521. } else
  522. error("ACLK synchronization thread is not active for host %s", host->hostname);
  523. #else
  524. UNUSED(node_id);
  525. UNUSED(snapshot_id);
  526. UNUSED(sequence_id);
  527. #endif
  528. return;
  529. }
  530. void aclk_mark_alert_cloud_ack(char *uuid_str, uint64_t alerts_ack_sequence_id)
  531. {
  532. BUFFER *sql = buffer_create(1024);
  533. if (alerts_ack_sequence_id != 0) {
  534. buffer_sprintf(
  535. sql,
  536. "UPDATE aclk_alert_%s SET date_cloud_ack = strftime('%%s','now') WHERE sequence_id <= %" PRIu64 "",
  537. uuid_str,
  538. alerts_ack_sequence_id);
  539. db_execute(buffer_tostring(sql));
  540. }
  541. buffer_free(sql);
  542. }
  543. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  544. void health_alarm_entry2proto_nolock(struct alarm_log_entry *alarm_log, ALARM_ENTRY *ae, RRDHOST *host)
  545. {
  546. char *edit_command = ae->source ? health_edit_command_from_source(ae->source) : strdupz("UNKNOWN=0");
  547. char config_hash_id[GUID_LEN + 1];
  548. uuid_unparse_lower(ae->config_hash_id, config_hash_id);
  549. alarm_log->chart = strdupz((char *)ae->chart);
  550. alarm_log->name = strdupz((char *)ae->name);
  551. alarm_log->family = strdupz((char *)ae->family);
  552. alarm_log->batch_id = 0;
  553. alarm_log->sequence_id = 0;
  554. alarm_log->when = (time_t)ae->when;
  555. alarm_log->config_hash = strdupz((char *)config_hash_id);
  556. alarm_log->utc_offset = host->utc_offset;
  557. alarm_log->timezone = strdupz((char *)host->abbrev_timezone);
  558. alarm_log->exec_path = ae->exec ? strdupz((char *)ae->exec) : strdupz((char *)host->health_default_exec);
  559. alarm_log->conf_source = ae->source ? strdupz((char *)ae->source) : "";
  560. alarm_log->command = strdupz((char *)edit_command);
  561. alarm_log->duration = (time_t)ae->duration;
  562. alarm_log->non_clear_duration = (time_t)ae->non_clear_duration;
  563. alarm_log->status = rrdcalc_status_to_proto_enum((RRDCALC_STATUS)ae->new_status);
  564. alarm_log->old_status = rrdcalc_status_to_proto_enum((RRDCALC_STATUS)ae->old_status);
  565. alarm_log->delay = (int)ae->delay;
  566. alarm_log->delay_up_to_timestamp = (time_t)ae->delay_up_to_timestamp;
  567. alarm_log->last_repeat = (time_t)ae->last_repeat;
  568. alarm_log->silenced =
  569. ((ae->flags & HEALTH_ENTRY_FLAG_SILENCED) || (ae->recipient && !strncmp((char *)ae->recipient, "silent", 6))) ?
  570. 1 :
  571. 0;
  572. alarm_log->value_string = strdupz(ae->new_value_string);
  573. alarm_log->old_value_string = strdupz(ae->old_value_string);
  574. alarm_log->value = (!isnan(ae->new_value)) ? (calculated_number)ae->new_value : 0;
  575. alarm_log->old_value = (!isnan(ae->old_value)) ? (calculated_number)ae->old_value : 0;
  576. alarm_log->updated = (ae->flags & HEALTH_ENTRY_FLAG_UPDATED) ? 1 : 0;
  577. alarm_log->rendered_info = strdupz(ae->info);
  578. freez(edit_command);
  579. }
  580. #endif
  581. static int have_recent_alarm(RRDHOST *host, uint32_t alarm_id, time_t mark)
  582. {
  583. ALARM_ENTRY *ae = host->health_log.alarms;
  584. while (ae) {
  585. if (ae->alarm_id == alarm_id && ae->unique_id > mark &&
  586. (ae->new_status != RRDCALC_STATUS_WARNING && ae->new_status != RRDCALC_STATUS_CRITICAL))
  587. return 1;
  588. ae = ae->next;
  589. }
  590. return 0;
  591. }
  592. #define ALARM_EVENTS_PER_CHUNK 10
  593. void aclk_push_alert_snapshot_event(struct aclk_database_worker_config *wc, struct aclk_database_cmd cmd)
  594. {
  595. #ifndef ENABLE_NEW_CLOUD_PROTOCOL
  596. UNUSED(wc);
  597. UNUSED(cmd);
  598. #else
  599. UNUSED(cmd);
  600. // we perhaps we don't need this for snapshots
  601. if (unlikely(!wc->alert_updates)) {
  602. log_access("AC [%s (%s)]: Ignoring alert snapshot event, updates have been turned off for this node.", wc->node_id, wc->host ? wc->host->hostname : "N/A");
  603. return;
  604. }
  605. if (unlikely(!wc->host)) {
  606. error_report("ACLK synchronization thread for %s is not linked to HOST", wc->host_guid);
  607. return;
  608. }
  609. char *claim_id = is_agent_claimed();
  610. if (unlikely(!claim_id))
  611. return;
  612. log_access("OG [%s (%s)]: Sending alerts snapshot, snapshot_id %" PRIu64, wc->node_id, wc->host ? wc->host->hostname : "N/A", wc->alerts_snapshot_id);
  613. aclk_mark_alert_cloud_ack(wc->uuid_str, wc->alerts_ack_sequence_id);
  614. RRDHOST *host = wc->host;
  615. uint32_t cnt = 0;
  616. netdata_rwlock_rdlock(&host->health_log.alarm_log_rwlock);
  617. ALARM_ENTRY *ae = host->health_log.alarms;
  618. for (; ae; ae = ae->next) {
  619. if (likely(ae->updated_by_id))
  620. continue;
  621. if (unlikely(ae->new_status == RRDCALC_STATUS_UNINITIALIZED))
  622. continue;
  623. if (have_recent_alarm(host, ae->alarm_id, ae->unique_id))
  624. continue;
  625. cnt++;
  626. }
  627. if (cnt) {
  628. uint32_t chunk = 1, chunks = 0;
  629. chunks = (cnt / ALARM_EVENTS_PER_CHUNK) + (cnt % ALARM_EVENTS_PER_CHUNK != 0);
  630. ae = host->health_log.alarms;
  631. cnt = 0;
  632. struct alarm_snapshot alarm_snap;
  633. alarm_snap.node_id = wc->node_id;
  634. alarm_snap.claim_id = claim_id;
  635. alarm_snap.snapshot_id = wc->alerts_snapshot_id;
  636. alarm_snap.chunks = chunks;
  637. alarm_snap.chunk = chunk;
  638. alarm_snapshot_proto_ptr_t snapshot_proto = NULL;
  639. for (; ae; ae = ae->next) {
  640. if (likely(ae->updated_by_id))
  641. continue;
  642. if (unlikely(ae->new_status == RRDCALC_STATUS_UNINITIALIZED))
  643. continue;
  644. if (have_recent_alarm(host, ae->alarm_id, ae->unique_id))
  645. continue;
  646. cnt++;
  647. struct alarm_log_entry alarm_log;
  648. alarm_log.node_id = wc->node_id;
  649. alarm_log.claim_id = claim_id;
  650. if (!snapshot_proto)
  651. snapshot_proto = generate_alarm_snapshot_proto(&alarm_snap);
  652. health_alarm_entry2proto_nolock(&alarm_log, ae, host);
  653. add_alarm_log_entry2snapshot(snapshot_proto, &alarm_log);
  654. if (cnt == ALARM_EVENTS_PER_CHUNK) {
  655. aclk_send_alarm_snapshot(snapshot_proto);
  656. cnt = 0;
  657. if (chunk < chunks) {
  658. chunk++;
  659. struct alarm_snapshot alarm_snap;
  660. alarm_snap.node_id = wc->node_id;
  661. alarm_snap.claim_id = claim_id;
  662. alarm_snap.snapshot_id = wc->alerts_snapshot_id;
  663. alarm_snap.chunks = chunks;
  664. alarm_snap.chunk = chunk;
  665. snapshot_proto = generate_alarm_snapshot_proto(&alarm_snap);
  666. }
  667. }
  668. destroy_alarm_log_entry(&alarm_log);
  669. }
  670. if (cnt)
  671. aclk_send_alarm_snapshot(snapshot_proto);
  672. }
  673. netdata_rwlock_unlock(&host->health_log.alarm_log_rwlock);
  674. wc->alerts_snapshot_id = 0;
  675. freez(claim_id);
  676. #endif
  677. return;
  678. }
  679. void sql_aclk_alert_clean_dead_entries(RRDHOST *host)
  680. {
  681. #ifdef ENABLE_NEW_CLOUD_PROTOCOL
  682. if (!claimed())
  683. return;
  684. if (unlikely(!host->dbsync_worker))
  685. return;
  686. char uuid_str[GUID_LEN + 1];
  687. uuid_unparse_lower_fix(&host->host_uuid, uuid_str);
  688. BUFFER *sql = buffer_create(1024);
  689. buffer_sprintf(sql,"delete from aclk_alert_%s where alert_unique_id not in "
  690. " (select unique_id from health_log_%s); ", uuid_str, uuid_str);
  691. char *err_msg = NULL;
  692. int rc = sqlite3_exec(db_meta, buffer_tostring(sql), NULL, NULL, &err_msg);
  693. if (rc != SQLITE_OK) {
  694. error_report("Failed when trying to clean stale ACLK alert entries from aclk_alert_%s, error message \"%s""",
  695. uuid_str, err_msg);
  696. sqlite3_free(err_msg);
  697. }
  698. buffer_free(sql);
  699. #else
  700. UNUSED(host);
  701. #endif
  702. }