query_target.c 44 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "internal.h"
  3. #define QUERY_TARGET_MAX_REALLOC_INCREASE 500
  4. #define query_target_realloc_size(size, start) \
  5. (size) ? ((size) < QUERY_TARGET_MAX_REALLOC_INCREASE ? (size) * 2 : (size) + QUERY_TARGET_MAX_REALLOC_INCREASE) : (start);
  6. static void query_metric_release(QUERY_TARGET *qt, QUERY_METRIC *qm);
  7. static void query_dimension_release(QUERY_DIMENSION *qd);
  8. static void query_instance_release(QUERY_INSTANCE *qi);
  9. static void query_context_release(QUERY_CONTEXT *qc);
  10. static void query_node_release(QUERY_NODE *qn);
  11. static __thread QUERY_TARGET *thread_qt = NULL;
  12. static struct {
  13. struct {
  14. SPINLOCK spinlock;
  15. size_t count;
  16. QUERY_TARGET *base;
  17. } available;
  18. struct {
  19. SPINLOCK spinlock;
  20. size_t count;
  21. QUERY_TARGET *base;
  22. } used;
  23. } query_target_base = {
  24. .available = {
  25. .spinlock = NETDATA_SPINLOCK_INITIALIZER,
  26. .base = NULL,
  27. .count = 0,
  28. },
  29. .used = {
  30. .spinlock = NETDATA_SPINLOCK_INITIALIZER,
  31. .base = NULL,
  32. .count = 0,
  33. },
  34. };
  35. static void query_target_destroy(QUERY_TARGET *qt) {
  36. __atomic_sub_fetch(&netdata_buffers_statistics.query_targets_size, qt->query.size * sizeof(*qt->query.array), __ATOMIC_RELAXED);
  37. freez(qt->query.array);
  38. __atomic_sub_fetch(&netdata_buffers_statistics.query_targets_size, qt->dimensions.size * sizeof(*qt->dimensions.array), __ATOMIC_RELAXED);
  39. freez(qt->dimensions.array);
  40. __atomic_sub_fetch(&netdata_buffers_statistics.query_targets_size, qt->instances.size * sizeof(*qt->instances.array), __ATOMIC_RELAXED);
  41. freez(qt->instances.array);
  42. __atomic_sub_fetch(&netdata_buffers_statistics.query_targets_size, qt->contexts.size * sizeof(*qt->contexts.array), __ATOMIC_RELAXED);
  43. freez(qt->contexts.array);
  44. __atomic_sub_fetch(&netdata_buffers_statistics.query_targets_size, qt->nodes.size * sizeof(*qt->nodes.array), __ATOMIC_RELAXED);
  45. freez(qt->nodes.array);
  46. freez(qt);
  47. }
  48. void query_target_release(QUERY_TARGET *qt) {
  49. if(unlikely(!qt)) return;
  50. internal_fatal(!qt->internal.used, "QUERY TARGET: qt to be released is not used");
  51. simple_pattern_free(qt->nodes.scope_pattern);
  52. qt->nodes.scope_pattern = NULL;
  53. simple_pattern_free(qt->nodes.pattern);
  54. qt->nodes.pattern = NULL;
  55. simple_pattern_free(qt->contexts.scope_pattern);
  56. qt->contexts.scope_pattern = NULL;
  57. simple_pattern_free(qt->contexts.pattern);
  58. qt->contexts.pattern = NULL;
  59. simple_pattern_free(qt->instances.pattern);
  60. qt->instances.pattern = NULL;
  61. simple_pattern_free(qt->instances.chart_label_key_pattern);
  62. qt->instances.chart_label_key_pattern = NULL;
  63. simple_pattern_free(qt->instances.labels_pattern);
  64. qt->instances.labels_pattern = NULL;
  65. simple_pattern_free(qt->query.pattern);
  66. qt->query.pattern = NULL;
  67. // release the query
  68. for(size_t i = 0, used = qt->query.used; i < used ;i++) {
  69. QUERY_METRIC *qm = query_metric(qt, i);
  70. query_metric_release(qt, qm);
  71. }
  72. qt->query.used = 0;
  73. // release the dimensions
  74. for(size_t i = 0, used = qt->dimensions.used; i < used ; i++) {
  75. QUERY_DIMENSION *qd = query_dimension(qt, i);
  76. query_dimension_release(qd);
  77. }
  78. qt->dimensions.used = 0;
  79. // release the instances
  80. for(size_t i = 0, used = qt->instances.used; i < used ;i++) {
  81. QUERY_INSTANCE *qi = query_instance(qt, i);
  82. query_instance_release(qi);
  83. }
  84. qt->instances.used = 0;
  85. // release the contexts
  86. for(size_t i = 0, used = qt->contexts.used; i < used ;i++) {
  87. QUERY_CONTEXT *qc = query_context(qt, i);
  88. rrdcontext_release(qc->rca);
  89. qc->rca = NULL;
  90. }
  91. qt->contexts.used = 0;
  92. // release the nodes
  93. for(size_t i = 0, used = qt->nodes.used; i < used ; i++) {
  94. QUERY_NODE *qn = query_node(qt, i);
  95. query_node_release(qn);
  96. }
  97. qt->nodes.used = 0;
  98. qt->db.minimum_latest_update_every_s = 0;
  99. qt->db.first_time_s = 0;
  100. qt->db.last_time_s = 0;
  101. for(size_t g = 0; g < MAX_QUERY_GROUP_BY_PASSES ;g++)
  102. qt->group_by[g].used = 0;
  103. qt->id[0] = '\0';
  104. netdata_spinlock_lock(&query_target_base.used.spinlock);
  105. DOUBLE_LINKED_LIST_REMOVE_ITEM_UNSAFE(query_target_base.used.base, qt, internal.prev, internal.next);
  106. query_target_base.used.count--;
  107. netdata_spinlock_unlock(&query_target_base.used.spinlock);
  108. qt->internal.used = false;
  109. thread_qt = NULL;
  110. if (qt->internal.queries > 1000) {
  111. query_target_destroy(qt);
  112. }
  113. else {
  114. netdata_spinlock_lock(&query_target_base.available.spinlock);
  115. DOUBLE_LINKED_LIST_APPEND_ITEM_UNSAFE(query_target_base.available.base, qt, internal.prev, internal.next);
  116. query_target_base.available.count++;
  117. netdata_spinlock_unlock(&query_target_base.available.spinlock);
  118. }
  119. }
  120. static QUERY_TARGET *query_target_get(void) {
  121. netdata_spinlock_lock(&query_target_base.available.spinlock);
  122. QUERY_TARGET *qt = query_target_base.available.base;
  123. if (qt) {
  124. DOUBLE_LINKED_LIST_REMOVE_ITEM_UNSAFE(query_target_base.available.base, qt, internal.prev, internal.next);
  125. query_target_base.available.count--;
  126. }
  127. netdata_spinlock_unlock(&query_target_base.available.spinlock);
  128. if(unlikely(!qt))
  129. qt = callocz(1, sizeof(*qt));
  130. netdata_spinlock_lock(&query_target_base.used.spinlock);
  131. DOUBLE_LINKED_LIST_APPEND_ITEM_UNSAFE(query_target_base.used.base, qt, internal.prev, internal.next);
  132. query_target_base.used.count++;
  133. netdata_spinlock_unlock(&query_target_base.used.spinlock);
  134. qt->internal.used = true;
  135. qt->internal.queries++;
  136. thread_qt = qt;
  137. return qt;
  138. }
  139. // this is used to release a query target from a cancelled thread
  140. void query_target_free(void) {
  141. query_target_release(thread_qt);
  142. }
  143. // ----------------------------------------------------------------------------
  144. // query API
  145. typedef struct query_target_locals {
  146. time_t start_s;
  147. QUERY_TARGET *qt;
  148. RRDSET *st;
  149. const char *scope_nodes;
  150. const char *scope_contexts;
  151. const char *nodes;
  152. const char *contexts;
  153. const char *instances;
  154. const char *dimensions;
  155. const char *chart_label_key;
  156. const char *labels;
  157. const char *alerts;
  158. long long after;
  159. long long before;
  160. bool match_ids;
  161. bool match_names;
  162. size_t metrics_skipped_due_to_not_matching_timeframe;
  163. char host_uuid_buffer[UUID_STR_LEN];
  164. QUERY_NODE *qn; // temp to pass on callbacks, ignore otherwise - no need to free
  165. } QUERY_TARGET_LOCALS;
  166. struct storage_engine *query_metric_storage_engine(QUERY_TARGET *qt, QUERY_METRIC *qm, size_t tier) {
  167. QUERY_NODE *qn = query_node(qt, qm->link.query_node_id);
  168. return qn->rrdhost->db[tier].eng;
  169. }
  170. static inline void query_metric_release(QUERY_TARGET *qt, QUERY_METRIC *qm) {
  171. qm->plan.used = 0;
  172. // reset the tiers
  173. for(size_t tier = 0; tier < storage_tiers ;tier++) {
  174. if(qm->tiers[tier].db_metric_handle) {
  175. STORAGE_ENGINE *eng = query_metric_storage_engine(qt, qm, tier);
  176. eng->api.metric_release(qm->tiers[tier].db_metric_handle);
  177. qm->tiers[tier].db_metric_handle = NULL;
  178. }
  179. }
  180. }
  181. static bool query_metric_add(QUERY_TARGET_LOCALS *qtl, QUERY_NODE *qn, QUERY_CONTEXT *qc,
  182. QUERY_INSTANCE *qi, size_t qd_slot, RRDMETRIC *rm, RRDR_DIMENSION_FLAGS options) {
  183. QUERY_TARGET *qt = qtl->qt;
  184. RRDINSTANCE *ri = rm->ri;
  185. time_t common_first_time_s = 0;
  186. time_t common_last_time_s = 0;
  187. time_t common_update_every_s = 0;
  188. size_t tiers_added = 0;
  189. struct {
  190. STORAGE_ENGINE *eng;
  191. STORAGE_METRIC_HANDLE *db_metric_handle;
  192. time_t db_first_time_s;
  193. time_t db_last_time_s;
  194. time_t db_update_every_s;
  195. } tier_retention[storage_tiers];
  196. for (size_t tier = 0; tier < storage_tiers; tier++) {
  197. STORAGE_ENGINE *eng = qn->rrdhost->db[tier].eng;
  198. tier_retention[tier].eng = eng;
  199. tier_retention[tier].db_update_every_s = (time_t) (qn->rrdhost->db[tier].tier_grouping * ri->update_every_s);
  200. if(rm->rrddim && rm->rrddim->tiers[tier].db_metric_handle)
  201. tier_retention[tier].db_metric_handle = eng->api.metric_dup(rm->rrddim->tiers[tier].db_metric_handle);
  202. else
  203. tier_retention[tier].db_metric_handle = eng->api.metric_get(qn->rrdhost->db[tier].instance, &rm->uuid);
  204. if(tier_retention[tier].db_metric_handle) {
  205. tier_retention[tier].db_first_time_s = storage_engine_oldest_time_s(tier_retention[tier].eng->backend, tier_retention[tier].db_metric_handle);
  206. tier_retention[tier].db_last_time_s = storage_engine_latest_time_s(tier_retention[tier].eng->backend, tier_retention[tier].db_metric_handle);
  207. if(!common_first_time_s)
  208. common_first_time_s = tier_retention[tier].db_first_time_s;
  209. else if(tier_retention[tier].db_first_time_s)
  210. common_first_time_s = MIN(common_first_time_s, tier_retention[tier].db_first_time_s);
  211. if(!common_last_time_s)
  212. common_last_time_s = tier_retention[tier].db_last_time_s;
  213. else
  214. common_last_time_s = MAX(common_last_time_s, tier_retention[tier].db_last_time_s);
  215. if(!common_update_every_s)
  216. common_update_every_s = tier_retention[tier].db_update_every_s;
  217. else if(tier_retention[tier].db_update_every_s)
  218. common_update_every_s = MIN(common_update_every_s, tier_retention[tier].db_update_every_s);
  219. tiers_added++;
  220. }
  221. else {
  222. tier_retention[tier].db_first_time_s = 0;
  223. tier_retention[tier].db_last_time_s = 0;
  224. tier_retention[tier].db_update_every_s = 0;
  225. }
  226. }
  227. for (size_t tier = 0; tier < storage_tiers; tier++) {
  228. if(!qt->db.tiers[tier].update_every || (tier_retention[tier].db_update_every_s && tier_retention[tier].db_update_every_s < qt->db.tiers[tier].update_every))
  229. qt->db.tiers[tier].update_every = tier_retention[tier].db_update_every_s;
  230. if(!qt->db.tiers[tier].retention.first_time_s || (tier_retention[tier].db_first_time_s && tier_retention[tier].db_first_time_s < qt->db.tiers[tier].retention.first_time_s))
  231. qt->db.tiers[tier].retention.first_time_s = tier_retention[tier].db_first_time_s;
  232. if(!qt->db.tiers[tier].retention.last_time_s || (tier_retention[tier].db_last_time_s && tier_retention[tier].db_last_time_s > qt->db.tiers[tier].retention.last_time_s))
  233. qt->db.tiers[tier].retention.last_time_s = tier_retention[tier].db_last_time_s;
  234. }
  235. bool timeframe_matches =
  236. (tiers_added &&
  237. query_matches_retention(qt->window.after, qt->window.before, common_first_time_s, common_last_time_s, common_update_every_s))
  238. ? true : false;
  239. if(timeframe_matches) {
  240. if(ri->rrdset)
  241. ri->rrdset->last_accessed_time_s = qtl->start_s;
  242. if (qt->query.used == qt->query.size) {
  243. size_t old_mem = qt->query.size * sizeof(*qt->query.array);
  244. qt->query.size = query_target_realloc_size(qt->query.size, 4);
  245. size_t new_mem = qt->query.size * sizeof(*qt->query.array);
  246. qt->query.array = reallocz(qt->query.array, new_mem);
  247. __atomic_add_fetch(&netdata_buffers_statistics.query_targets_size, new_mem - old_mem, __ATOMIC_RELAXED);
  248. }
  249. QUERY_METRIC *qm = &qt->query.array[qt->query.used++];
  250. memset(qm, 0, sizeof(*qm));
  251. qm->status = options;
  252. qm->link.query_node_id = qn->slot;
  253. qm->link.query_context_id = qc->slot;
  254. qm->link.query_instance_id = qi->slot;
  255. qm->link.query_dimension_id = qd_slot;
  256. if (!qt->db.first_time_s || common_first_time_s < qt->db.first_time_s)
  257. qt->db.first_time_s = common_first_time_s;
  258. if (!qt->db.last_time_s || common_last_time_s > qt->db.last_time_s)
  259. qt->db.last_time_s = common_last_time_s;
  260. for (size_t tier = 0; tier < storage_tiers; tier++) {
  261. internal_fatal(tier_retention[tier].eng != query_metric_storage_engine(qt, qm, tier), "QUERY TARGET: storage engine mismatch");
  262. qm->tiers[tier].db_metric_handle = tier_retention[tier].db_metric_handle;
  263. qm->tiers[tier].db_first_time_s = tier_retention[tier].db_first_time_s;
  264. qm->tiers[tier].db_last_time_s = tier_retention[tier].db_last_time_s;
  265. qm->tiers[tier].db_update_every_s = tier_retention[tier].db_update_every_s;
  266. }
  267. return true;
  268. }
  269. // cleanup anything we allocated to the retention we will not use
  270. for(size_t tier = 0; tier < storage_tiers ;tier++) {
  271. if (tier_retention[tier].db_metric_handle)
  272. tier_retention[tier].eng->api.metric_release(tier_retention[tier].db_metric_handle);
  273. }
  274. return false;
  275. }
  276. static inline bool rrdmetric_retention_matches_query(QUERY_TARGET *qt, RRDMETRIC *rm, time_t now_s) {
  277. time_t first_time_s = rm->first_time_s;
  278. time_t last_time_s = rrd_flag_is_collected(rm) ? now_s : rm->last_time_s;
  279. time_t update_every_s = rm->ri->update_every_s;
  280. return query_matches_retention(qt->window.after, qt->window.before, first_time_s, last_time_s, update_every_s);
  281. }
  282. static inline void query_dimension_release(QUERY_DIMENSION *qd) {
  283. rrdmetric_release(qd->rma);
  284. qd->rma = NULL;
  285. }
  286. static QUERY_DIMENSION *query_dimension_allocate(QUERY_TARGET *qt, RRDMETRIC_ACQUIRED *rma, QUERY_STATUS status, size_t priority) {
  287. if(qt->dimensions.used == qt->dimensions.size) {
  288. size_t old_mem = qt->dimensions.size * sizeof(*qt->dimensions.array);
  289. qt->dimensions.size = query_target_realloc_size(qt->dimensions.size, 4);
  290. size_t new_mem = qt->dimensions.size * sizeof(*qt->dimensions.array);
  291. qt->dimensions.array = reallocz(qt->dimensions.array, new_mem);
  292. __atomic_add_fetch(&netdata_buffers_statistics.query_targets_size, new_mem - old_mem, __ATOMIC_RELAXED);
  293. }
  294. QUERY_DIMENSION *qd = &qt->dimensions.array[qt->dimensions.used];
  295. memset(qd, 0, sizeof(*qd));
  296. qd->slot = qt->dimensions.used++;
  297. qd->rma = rrdmetric_acquired_dup(rma);
  298. qd->status = status;
  299. qd->priority = priority;
  300. return qd;
  301. }
  302. static bool query_dimension_add(QUERY_TARGET_LOCALS *qtl, QUERY_NODE *qn, QUERY_CONTEXT *qc, QUERY_INSTANCE *qi,
  303. RRDMETRIC_ACQUIRED *rma, bool queryable_instance, size_t *metrics_added, size_t priority) {
  304. QUERY_TARGET *qt = qtl->qt;
  305. RRDMETRIC *rm = rrdmetric_acquired_value(rma);
  306. if(rrd_flag_is_deleted(rm))
  307. return false;
  308. QUERY_STATUS status = QUERY_STATUS_NONE;
  309. bool undo = false;
  310. if(!queryable_instance) {
  311. if(rrdmetric_retention_matches_query(qt, rm, qtl->start_s)) {
  312. qi->metrics.excluded++;
  313. qc->metrics.excluded++;
  314. qn->metrics.excluded++;
  315. status |= QUERY_STATUS_EXCLUDED;
  316. }
  317. else
  318. undo = true;
  319. }
  320. else {
  321. RRDR_DIMENSION_FLAGS options = RRDR_DIMENSION_DEFAULT;
  322. bool needed = false;
  323. if (qt->query.pattern) {
  324. // the user asked for specific dimensions
  325. SIMPLE_PATTERN_RESULT ret = SP_NOT_MATCHED;
  326. if(qtl->match_ids)
  327. ret = simple_pattern_matches_string_extract(qt->query.pattern, rm->id, NULL, 0);
  328. if(ret == SP_NOT_MATCHED && qtl->match_names && (rm->name != rm->id || !qtl->match_ids))
  329. ret = simple_pattern_matches_string_extract(qt->query.pattern, rm->name, NULL, 0);
  330. if(ret == SP_MATCHED_POSITIVE) {
  331. needed = true;
  332. options |= RRDR_DIMENSION_SELECTED | RRDR_DIMENSION_NONZERO;
  333. }
  334. else {
  335. // the user selection does not match this dimension
  336. // but, we may still need to query it
  337. if (query_target_needs_all_dimensions(qt)) {
  338. // this is percentage calculation
  339. // so, we need this dimension to calculate the percentage
  340. needed = true;
  341. options |= RRDR_DIMENSION_HIDDEN;
  342. }
  343. else {
  344. // the user did not select this dimension
  345. // and the calculation is not percentage
  346. // so, no need to query it
  347. ;
  348. }
  349. }
  350. }
  351. else {
  352. // we don't have a dimensions pattern
  353. // so this is a selected dimension
  354. // if it is not hidden
  355. if(rrd_flag_check(rm, RRD_FLAG_HIDDEN) || (rm->rrddim && rrddim_option_check(rm->rrddim, RRDDIM_OPTION_HIDDEN))) {
  356. // this is a hidden dimension
  357. // we don't need to query it
  358. status |= QUERY_STATUS_DIMENSION_HIDDEN;
  359. options |= RRDR_DIMENSION_HIDDEN;
  360. if (query_target_needs_all_dimensions(qt)) {
  361. // this is percentage calculation
  362. // so, we need this dimension to calculate the percentage
  363. needed = true;
  364. }
  365. }
  366. else {
  367. // this is a not hidden dimension
  368. // and the user did not provide any selection for dimensions
  369. // so, we need to query it
  370. needed = true;
  371. options |= RRDR_DIMENSION_SELECTED;
  372. }
  373. }
  374. if (needed) {
  375. if(query_metric_add(qtl, qn, qc, qi, qt->dimensions.used, rm, options)) {
  376. (*metrics_added)++;
  377. qi->metrics.selected++;
  378. qc->metrics.selected++;
  379. qn->metrics.selected++;
  380. }
  381. else {
  382. undo = true;
  383. qtl->metrics_skipped_due_to_not_matching_timeframe++;
  384. }
  385. }
  386. else if(rrdmetric_retention_matches_query(qt, rm, qtl->start_s)) {
  387. qi->metrics.excluded++;
  388. qc->metrics.excluded++;
  389. qn->metrics.excluded++;
  390. status |= QUERY_STATUS_EXCLUDED;
  391. }
  392. else
  393. undo = true;
  394. }
  395. if(undo)
  396. return false;
  397. query_dimension_allocate(qt, rma, status, priority);
  398. return true;
  399. }
  400. static inline STRING *rrdinstance_create_id_fqdn_v1(RRDINSTANCE_ACQUIRED *ria) {
  401. if(unlikely(!ria))
  402. return NULL;
  403. RRDINSTANCE *ri = rrdinstance_acquired_value(ria);
  404. return string_dup(ri->id);
  405. }
  406. static inline STRING *rrdinstance_create_name_fqdn_v1(RRDINSTANCE_ACQUIRED *ria) {
  407. if(unlikely(!ria))
  408. return NULL;
  409. RRDINSTANCE *ri = rrdinstance_acquired_value(ria);
  410. return string_dup(ri->name);
  411. }
  412. static inline STRING *rrdinstance_create_id_fqdn_v2(RRDINSTANCE_ACQUIRED *ria) {
  413. if(unlikely(!ria))
  414. return NULL;
  415. char buffer[RRD_ID_LENGTH_MAX + 1];
  416. RRDHOST *host = rrdinstance_acquired_rrdhost(ria);
  417. snprintfz(buffer, RRD_ID_LENGTH_MAX, "%s@%s", rrdinstance_acquired_id(ria), host->machine_guid);
  418. return string_strdupz(buffer);
  419. }
  420. static inline STRING *rrdinstance_create_name_fqdn_v2(RRDINSTANCE_ACQUIRED *ria) {
  421. if(unlikely(!ria))
  422. return NULL;
  423. char buffer[RRD_ID_LENGTH_MAX + 1];
  424. RRDHOST *host = rrdinstance_acquired_rrdhost(ria);
  425. snprintfz(buffer, RRD_ID_LENGTH_MAX, "%s@%s", rrdinstance_acquired_name(ria), rrdhost_hostname(host));
  426. return string_strdupz(buffer);
  427. }
  428. inline STRING *query_instance_id_fqdn(QUERY_INSTANCE *qi, size_t version) {
  429. if(!qi->id_fqdn) {
  430. if (version <= 1)
  431. qi->id_fqdn = rrdinstance_create_id_fqdn_v1(qi->ria);
  432. else
  433. qi->id_fqdn = rrdinstance_create_id_fqdn_v2(qi->ria);
  434. }
  435. return qi->id_fqdn;
  436. }
  437. inline STRING *query_instance_name_fqdn(QUERY_INSTANCE *qi, size_t version) {
  438. if(!qi->name_fqdn) {
  439. if (version <= 1)
  440. qi->name_fqdn = rrdinstance_create_name_fqdn_v1(qi->ria);
  441. else
  442. qi->name_fqdn = rrdinstance_create_name_fqdn_v2(qi->ria);
  443. }
  444. return qi->name_fqdn;
  445. }
  446. RRDSET *rrdinstance_acquired_rrdset(RRDINSTANCE_ACQUIRED *ria) {
  447. RRDINSTANCE *ri = rrdinstance_acquired_value(ria);
  448. return ri->rrdset;
  449. }
  450. const char *rrdcontext_acquired_units(RRDCONTEXT_ACQUIRED *rca) {
  451. RRDCONTEXT *rc = rrdcontext_acquired_value(rca);
  452. return string2str(rc->units);
  453. }
  454. RRDSET_TYPE rrdcontext_acquired_chart_type(RRDCONTEXT_ACQUIRED *rca) {
  455. RRDCONTEXT *rc = rrdcontext_acquired_value(rca);
  456. return rc->chart_type;
  457. }
  458. const char *rrdcontext_acquired_title(RRDCONTEXT_ACQUIRED *rca) {
  459. RRDCONTEXT *rc = rrdcontext_acquired_value(rca);
  460. return string2str(rc->title);
  461. }
  462. static void query_target_eval_instance_rrdcalc(QUERY_TARGET_LOCALS *qtl __maybe_unused,
  463. QUERY_NODE *qn, QUERY_CONTEXT *qc, QUERY_INSTANCE *qi) {
  464. RRDSET *st = rrdinstance_acquired_rrdset(qi->ria);
  465. if (st) {
  466. netdata_rwlock_rdlock(&st->alerts.rwlock);
  467. for (RRDCALC *rc = st->alerts.base; rc; rc = rc->next) {
  468. switch(rc->status) {
  469. case RRDCALC_STATUS_CLEAR:
  470. qi->alerts.clear++;
  471. qc->alerts.clear++;
  472. qn->alerts.clear++;
  473. break;
  474. case RRDCALC_STATUS_WARNING:
  475. qi->alerts.warning++;
  476. qc->alerts.warning++;
  477. qn->alerts.warning++;
  478. break;
  479. case RRDCALC_STATUS_CRITICAL:
  480. qi->alerts.critical++;
  481. qc->alerts.critical++;
  482. qn->alerts.critical++;
  483. break;
  484. default:
  485. case RRDCALC_STATUS_UNINITIALIZED:
  486. case RRDCALC_STATUS_UNDEFINED:
  487. case RRDCALC_STATUS_REMOVED:
  488. qi->alerts.other++;
  489. qc->alerts.other++;
  490. qn->alerts.other++;
  491. break;
  492. }
  493. }
  494. netdata_rwlock_unlock(&st->alerts.rwlock);
  495. }
  496. }
  497. static bool query_target_match_alert_pattern(RRDINSTANCE_ACQUIRED *ria, SIMPLE_PATTERN *pattern) {
  498. if(!pattern)
  499. return true;
  500. RRDSET *st = rrdinstance_acquired_rrdset(ria);
  501. if (!st)
  502. return false;
  503. BUFFER *wb = NULL;
  504. bool matched = false;
  505. netdata_rwlock_rdlock(&st->alerts.rwlock);
  506. if (st->alerts.base) {
  507. for (RRDCALC *rc = st->alerts.base; rc; rc = rc->next) {
  508. SIMPLE_PATTERN_RESULT ret = simple_pattern_matches_string_extract(pattern, rc->name, NULL, 0);
  509. if(ret == SP_MATCHED_POSITIVE) {
  510. matched = true;
  511. break;
  512. }
  513. else if(ret == SP_MATCHED_NEGATIVE)
  514. break;
  515. if (!wb)
  516. wb = buffer_create(0, NULL);
  517. else
  518. buffer_flush(wb);
  519. buffer_fast_strcat(wb, string2str(rc->name), string_strlen(rc->name));
  520. buffer_fast_strcat(wb, ":", 1);
  521. buffer_strcat(wb, rrdcalc_status2string(rc->status));
  522. ret = simple_pattern_matches_buffer_extract(pattern, wb, NULL, 0);
  523. if(ret == SP_MATCHED_POSITIVE) {
  524. matched = true;
  525. break;
  526. }
  527. else if(ret == SP_MATCHED_NEGATIVE)
  528. break;
  529. }
  530. }
  531. netdata_rwlock_unlock(&st->alerts.rwlock);
  532. buffer_free(wb);
  533. return matched;
  534. }
  535. static inline void query_instance_release(QUERY_INSTANCE *qi) {
  536. if(qi->ria) {
  537. rrdinstance_release(qi->ria);
  538. qi->ria = NULL;
  539. }
  540. string_freez(qi->id_fqdn);
  541. qi->id_fqdn = NULL;
  542. string_freez(qi->name_fqdn);
  543. qi->name_fqdn = NULL;
  544. }
  545. static inline QUERY_INSTANCE *query_instance_allocate(QUERY_TARGET *qt, RRDINSTANCE_ACQUIRED *ria, size_t qn_slot) {
  546. if(qt->instances.used == qt->instances.size) {
  547. size_t old_mem = qt->instances.size * sizeof(*qt->instances.array);
  548. qt->instances.size = query_target_realloc_size(qt->instances.size, 2);
  549. size_t new_mem = qt->instances.size * sizeof(*qt->instances.array);
  550. qt->instances.array = reallocz(qt->instances.array, new_mem);
  551. __atomic_add_fetch(&netdata_buffers_statistics.query_targets_size, new_mem - old_mem, __ATOMIC_RELAXED);
  552. }
  553. QUERY_INSTANCE *qi = &qt->instances.array[qt->instances.used];
  554. memset(qi, 0, sizeof(*qi));
  555. qi->slot = qt->instances.used;
  556. qt->instances.used++;
  557. qi->ria = rrdinstance_acquired_dup(ria);
  558. qi->query_host_id = qn_slot;
  559. return qi;
  560. }
  561. static inline SIMPLE_PATTERN_RESULT query_instance_matches(QUERY_INSTANCE *qi,
  562. RRDINSTANCE *ri,
  563. SIMPLE_PATTERN *instances_sp,
  564. bool match_ids,
  565. bool match_names,
  566. size_t version) {
  567. SIMPLE_PATTERN_RESULT ret = SP_MATCHED_POSITIVE;
  568. if(instances_sp) {
  569. ret = SP_NOT_MATCHED;
  570. if(match_ids)
  571. ret = simple_pattern_matches_string_extract(instances_sp, ri->id, NULL, 0);
  572. if (ret == SP_NOT_MATCHED && match_names && (ri->name != ri->id || !match_ids))
  573. ret = simple_pattern_matches_string_extract(instances_sp, ri->name, NULL, 0);
  574. if (ret == SP_NOT_MATCHED && match_ids)
  575. ret = simple_pattern_matches_string_extract(instances_sp, query_instance_id_fqdn(qi, version), NULL, 0);
  576. if (ret == SP_NOT_MATCHED && match_names)
  577. ret = simple_pattern_matches_string_extract(instances_sp, query_instance_name_fqdn(qi, version), NULL, 0);
  578. }
  579. return ret;
  580. }
  581. static inline bool query_instance_matches_labels(RRDINSTANCE *ri, SIMPLE_PATTERN *chart_label_key_sp, SIMPLE_PATTERN *labels_sp) {
  582. if ((chart_label_key_sp && !rrdlabels_match_simple_pattern_parsed(
  583. ri->rrdlabels, chart_label_key_sp, '\0', NULL)) ||
  584. (labels_sp && !rrdlabels_match_simple_pattern_parsed(
  585. ri->rrdlabels, labels_sp, ':', NULL)))
  586. return false;
  587. return true;
  588. }
  589. static bool query_instance_add(QUERY_TARGET_LOCALS *qtl, QUERY_NODE *qn, QUERY_CONTEXT *qc,
  590. RRDINSTANCE_ACQUIRED *ria, bool queryable_instance, bool filter_instances) {
  591. RRDINSTANCE *ri = rrdinstance_acquired_value(ria);
  592. if(rrd_flag_is_deleted(ri))
  593. return false;
  594. QUERY_TARGET *qt = qtl->qt;
  595. QUERY_INSTANCE *qi = query_instance_allocate(qt, ria, qn->slot);
  596. if(qt->db.minimum_latest_update_every_s == 0 || ri->update_every_s < qt->db.minimum_latest_update_every_s)
  597. qt->db.minimum_latest_update_every_s = ri->update_every_s;
  598. if(queryable_instance && filter_instances)
  599. queryable_instance = (SP_MATCHED_POSITIVE == query_instance_matches(
  600. qi, ri, qt->instances.pattern, qtl->match_ids, qtl->match_names, qt->request.version));
  601. if(queryable_instance)
  602. queryable_instance = query_instance_matches_labels(ri, qt->instances.chart_label_key_pattern, qt->instances.labels_pattern);
  603. if(queryable_instance) {
  604. if(qt->instances.alerts_pattern && !query_target_match_alert_pattern(ria, qt->instances.alerts_pattern))
  605. queryable_instance = false;
  606. }
  607. if(queryable_instance && qt->request.version >= 2)
  608. query_target_eval_instance_rrdcalc(qtl, qn, qc, qi);
  609. size_t dimensions_added = 0, metrics_added = 0, priority = 0;
  610. if(unlikely(qt->request.rma)) {
  611. if(query_dimension_add(qtl, qn, qc, qi, qt->request.rma, queryable_instance, &metrics_added, priority++))
  612. dimensions_added++;
  613. }
  614. else {
  615. RRDMETRIC *rm;
  616. dfe_start_read(ri->rrdmetrics, rm) {
  617. if(query_dimension_add(qtl, qn, qc, qi, (RRDMETRIC_ACQUIRED *) rm_dfe.item,
  618. queryable_instance, &metrics_added, priority++))
  619. dimensions_added++;
  620. }
  621. dfe_done(rm);
  622. }
  623. if(!dimensions_added) {
  624. qt->instances.used--;
  625. query_instance_release(qi);
  626. return false;
  627. }
  628. else {
  629. if(metrics_added) {
  630. qc->instances.selected++;
  631. qn->instances.selected++;
  632. }
  633. else {
  634. qc->instances.excluded++;
  635. qn->instances.excluded++;
  636. }
  637. }
  638. return true;
  639. }
  640. static inline void query_context_release(QUERY_CONTEXT *qc) {
  641. rrdcontext_release(qc->rca);
  642. qc->rca = NULL;
  643. }
  644. static inline QUERY_CONTEXT *query_context_allocate(QUERY_TARGET *qt, RRDCONTEXT_ACQUIRED *rca) {
  645. if(qt->contexts.used == qt->contexts.size) {
  646. size_t old_mem = qt->contexts.size * sizeof(*qt->contexts.array);
  647. qt->contexts.size = query_target_realloc_size(qt->contexts.size, 2);
  648. size_t new_mem = qt->contexts.size * sizeof(*qt->contexts.array);
  649. qt->contexts.array = reallocz(qt->contexts.array, new_mem);
  650. __atomic_add_fetch(&netdata_buffers_statistics.query_targets_size, new_mem - old_mem, __ATOMIC_RELAXED);
  651. }
  652. QUERY_CONTEXT *qc = &qt->contexts.array[qt->contexts.used];
  653. memset(qc, 0, sizeof(*qc));
  654. qc->slot = qt->contexts.used++;
  655. qc->rca = rrdcontext_acquired_dup(rca);
  656. return qc;
  657. }
  658. static ssize_t query_context_add(void *data, RRDCONTEXT_ACQUIRED *rca, bool queryable_context) {
  659. QUERY_TARGET_LOCALS *qtl = data;
  660. RRDCONTEXT *rc = rrdcontext_acquired_value(rca);
  661. if(rrd_flag_is_deleted(rc))
  662. return 0;
  663. QUERY_NODE *qn = qtl->qn;
  664. QUERY_TARGET *qt = qtl->qt;
  665. QUERY_CONTEXT *qc = query_context_allocate(qt, rca);
  666. ssize_t added = 0;
  667. if(unlikely(qt->request.ria)) {
  668. if(query_instance_add(qtl, qn, qc, qt->request.ria, queryable_context, false))
  669. added++;
  670. }
  671. else if(unlikely(qtl->st && qtl->st->rrdcontext == rca && qtl->st->rrdinstance)) {
  672. if(query_instance_add(qtl, qn, qc, qtl->st->rrdinstance, queryable_context, false))
  673. added++;
  674. }
  675. else {
  676. RRDINSTANCE *ri;
  677. dfe_start_read(rc->rrdinstances, ri) {
  678. if(query_instance_add(qtl, qn, qc, (RRDINSTANCE_ACQUIRED *) ri_dfe.item, queryable_context, true))
  679. added++;
  680. }
  681. dfe_done(ri);
  682. }
  683. if(!added) {
  684. query_context_release(qc);
  685. qt->contexts.used--;
  686. return 0;
  687. }
  688. return added;
  689. }
  690. static inline void query_node_release(QUERY_NODE *qn) {
  691. qn->rrdhost = NULL;
  692. }
  693. static inline QUERY_NODE *query_node_allocate(QUERY_TARGET *qt, RRDHOST *host) {
  694. if(qt->nodes.used == qt->nodes.size) {
  695. size_t old_mem = qt->nodes.size * sizeof(*qt->nodes.array);
  696. qt->nodes.size = query_target_realloc_size(qt->nodes.size, 2);
  697. size_t new_mem = qt->nodes.size * sizeof(*qt->nodes.array);
  698. qt->nodes.array = reallocz(qt->nodes.array, new_mem);
  699. __atomic_add_fetch(&netdata_buffers_statistics.query_targets_size, new_mem - old_mem, __ATOMIC_RELAXED);
  700. }
  701. QUERY_NODE *qn = &qt->nodes.array[qt->nodes.used];
  702. memset(qn, 0, sizeof(*qn));
  703. qn->slot = qt->nodes.used++;
  704. qn->rrdhost = host;
  705. return qn;
  706. }
  707. static ssize_t query_node_add(void *data, RRDHOST *host, bool queryable_host) {
  708. QUERY_TARGET_LOCALS *qtl = data;
  709. QUERY_TARGET *qt = qtl->qt;
  710. QUERY_NODE *qn = query_node_allocate(qt, host);
  711. if(host->node_id) {
  712. if(!qtl->host_uuid_buffer[0])
  713. uuid_unparse_lower(*host->node_id, qn->node_id);
  714. else
  715. memcpy(qn->node_id, qtl->host_uuid_buffer, sizeof(qn->node_id));
  716. }
  717. else
  718. qn->node_id[0] = '\0';
  719. // is the chart given valid?
  720. if(unlikely(qtl->st && (!qtl->st->rrdinstance || !qtl->st->rrdcontext))) {
  721. error("QUERY TARGET: RRDSET '%s' given, but it is not linked to rrdcontext structures. Linking it now.", rrdset_name(qtl->st));
  722. rrdinstance_from_rrdset(qtl->st);
  723. if(unlikely(qtl->st && (!qtl->st->rrdinstance || !qtl->st->rrdcontext))) {
  724. error("QUERY TARGET: RRDSET '%s' given, but failed to be linked to rrdcontext structures. Switching to context query.",
  725. rrdset_name(qtl->st));
  726. if (!is_valid_sp(qtl->instances))
  727. qtl->instances = rrdset_name(qtl->st);
  728. qtl->st = NULL;
  729. }
  730. }
  731. qtl->qn = qn;
  732. ssize_t added = 0;
  733. if(unlikely(qt->request.rca)) {
  734. if(query_context_add(qtl, qt->request.rca, true))
  735. added++;
  736. }
  737. else if(unlikely(qtl->st)) {
  738. // single chart data queries
  739. if(query_context_add(qtl, qtl->st->rrdcontext, true))
  740. added++;
  741. }
  742. else {
  743. // context pattern queries
  744. added = query_scope_foreach_context(
  745. host, qtl->scope_contexts,
  746. qt->contexts.scope_pattern, qt->contexts.pattern,
  747. query_context_add, queryable_host, qtl);
  748. if(added < 0)
  749. added = 0;
  750. }
  751. qtl->qn = NULL;
  752. if(!added) {
  753. query_node_release(qn);
  754. qt->nodes.used--;
  755. return false;
  756. }
  757. return true;
  758. }
  759. void query_target_generate_name(QUERY_TARGET *qt) {
  760. char options_buffer[100 + 1];
  761. web_client_api_request_v1_data_options_to_string(options_buffer, 100, qt->request.options);
  762. char resampling_buffer[20 + 1] = "";
  763. if(qt->request.resampling_time > 1)
  764. snprintfz(resampling_buffer, 20, "/resampling:%lld", (long long)qt->request.resampling_time);
  765. char tier_buffer[20 + 1] = "";
  766. if(qt->request.options & RRDR_OPTION_SELECTED_TIER)
  767. snprintfz(tier_buffer, 20, "/tier:%zu", qt->request.tier);
  768. if(qt->request.st)
  769. snprintfz(qt->id, MAX_QUERY_TARGET_ID_LENGTH, "chart://hosts:%s/instance:%s/dimensions:%s/after:%lld/before:%lld/points:%zu/group:%s%s/options:%s%s%s"
  770. , rrdhost_hostname(qt->request.st->rrdhost)
  771. , rrdset_name(qt->request.st)
  772. , (qt->request.dimensions) ? qt->request.dimensions : "*"
  773. , (long long)qt->request.after
  774. , (long long)qt->request.before
  775. , qt->request.points
  776. , time_grouping_tostring(qt->request.time_group_method)
  777. , qt->request.time_group_options ? qt->request.time_group_options : ""
  778. , options_buffer
  779. , resampling_buffer
  780. , tier_buffer
  781. );
  782. else if(qt->request.host && qt->request.rca && qt->request.ria && qt->request.rma)
  783. snprintfz(qt->id, MAX_QUERY_TARGET_ID_LENGTH, "metric://hosts:%s/context:%s/instance:%s/dimension:%s/after:%lld/before:%lld/points:%zu/group:%s%s/options:%s%s%s"
  784. , rrdhost_hostname(qt->request.host)
  785. , rrdcontext_acquired_id(qt->request.rca)
  786. , rrdinstance_acquired_id(qt->request.ria)
  787. , rrdmetric_acquired_id(qt->request.rma)
  788. , (long long)qt->request.after
  789. , (long long)qt->request.before
  790. , qt->request.points
  791. , time_grouping_tostring(qt->request.time_group_method)
  792. , qt->request.time_group_options ? qt->request.time_group_options : ""
  793. , options_buffer
  794. , resampling_buffer
  795. , tier_buffer
  796. );
  797. else if(qt->request.version >= 2)
  798. snprintfz(qt->id, MAX_QUERY_TARGET_ID_LENGTH, "data_v2://scope_nodes:%s/scope_contexts:%s/nodes:%s/contexts:%s/instances:%s/labels:%s/dimensions:%s/after:%lld/before:%lld/points:%zu/time_group:%s%s/options:%s%s%s"
  799. , qt->request.scope_nodes ? qt->request.scope_nodes : "*"
  800. , qt->request.scope_contexts ? qt->request.scope_contexts : "*"
  801. , qt->request.nodes ? qt->request.nodes : "*"
  802. , (qt->request.contexts) ? qt->request.contexts : "*"
  803. , (qt->request.instances) ? qt->request.instances : "*"
  804. , (qt->request.labels) ? qt->request.labels : "*"
  805. , (qt->request.dimensions) ? qt->request.dimensions : "*"
  806. , (long long)qt->request.after
  807. , (long long)qt->request.before
  808. , qt->request.points
  809. , time_grouping_tostring(qt->request.time_group_method)
  810. , qt->request.time_group_options ? qt->request.time_group_options : ""
  811. , options_buffer
  812. , resampling_buffer
  813. , tier_buffer
  814. );
  815. else
  816. snprintfz(qt->id, MAX_QUERY_TARGET_ID_LENGTH, "context://hosts:%s/contexts:%s/instances:%s/dimensions:%s/after:%lld/before:%lld/points:%zu/group:%s%s/options:%s%s%s"
  817. , (qt->request.host) ? rrdhost_hostname(qt->request.host) : ((qt->request.nodes) ? qt->request.nodes : "*")
  818. , (qt->request.contexts) ? qt->request.contexts : "*"
  819. , (qt->request.instances) ? qt->request.instances : "*"
  820. , (qt->request.dimensions) ? qt->request.dimensions : "*"
  821. , (long long)qt->request.after
  822. , (long long)qt->request.before
  823. , qt->request.points
  824. , time_grouping_tostring(qt->request.time_group_method)
  825. , qt->request.time_group_options ? qt->request.time_group_options : ""
  826. , options_buffer
  827. , resampling_buffer
  828. , tier_buffer
  829. );
  830. json_fix_string(qt->id);
  831. }
  832. QUERY_TARGET *query_target_create(QUERY_TARGET_REQUEST *qtr) {
  833. if(!service_running(ABILITY_DATA_QUERIES))
  834. return NULL;
  835. QUERY_TARGET *qt = query_target_get();
  836. if(!qtr->received_ut)
  837. qtr->received_ut = now_monotonic_usec();
  838. qt->timings.received_ut = qtr->received_ut;
  839. if(qtr->nodes && !qtr->scope_nodes)
  840. qtr->scope_nodes = qtr->nodes;
  841. if(qtr->contexts && !qtr->scope_contexts)
  842. qtr->scope_contexts = qtr->contexts;
  843. memset(&qt->db, 0, sizeof(qt->db));
  844. qt->query_points = STORAGE_POINT_UNSET;
  845. // copy the request into query_thread_target
  846. qt->request = *qtr;
  847. query_target_generate_name(qt);
  848. qt->window.after = qt->request.after;
  849. qt->window.before = qt->request.before;
  850. qt->window.options = qt->request.options;
  851. if(query_target_has_percentage_of_instance(qt))
  852. qt->window.options &= ~RRDR_OPTION_PERCENTAGE;
  853. rrdr_relative_window_to_absolute(&qt->window.after, &qt->window.before, &qt->window.now);
  854. // prepare our local variables - we need these across all these functions
  855. QUERY_TARGET_LOCALS qtl = {
  856. .qt = qt,
  857. .start_s = now_realtime_sec(),
  858. .st = qt->request.st,
  859. .scope_nodes = qt->request.scope_nodes,
  860. .scope_contexts = qt->request.scope_contexts,
  861. .nodes = qt->request.nodes,
  862. .contexts = qt->request.contexts,
  863. .instances = qt->request.instances,
  864. .dimensions = qt->request.dimensions,
  865. .chart_label_key = qt->request.chart_label_key,
  866. .labels = qt->request.labels,
  867. .alerts = qt->request.alerts,
  868. };
  869. RRDHOST *host = qt->request.host;
  870. // prepare all the patterns
  871. qt->nodes.scope_pattern = string_to_simple_pattern(qtl.scope_nodes);
  872. qt->nodes.pattern = string_to_simple_pattern(qtl.nodes);
  873. qt->contexts.pattern = string_to_simple_pattern(qtl.contexts);
  874. qt->contexts.scope_pattern = string_to_simple_pattern(qtl.scope_contexts);
  875. qt->instances.pattern = string_to_simple_pattern(qtl.instances);
  876. qt->query.pattern = string_to_simple_pattern(qtl.dimensions);
  877. qt->instances.chart_label_key_pattern = string_to_simple_pattern(qtl.chart_label_key);
  878. qt->instances.labels_pattern = string_to_simple_pattern(qtl.labels);
  879. qt->instances.alerts_pattern = string_to_simple_pattern(qtl.alerts);
  880. qtl.match_ids = qt->request.options & RRDR_OPTION_MATCH_IDS;
  881. qtl.match_names = qt->request.options & RRDR_OPTION_MATCH_NAMES;
  882. if(likely(!qtl.match_ids && !qtl.match_names))
  883. qtl.match_ids = qtl.match_names = true;
  884. // verify that the chart belongs to the host we are interested
  885. if(qtl.st) {
  886. if (!host) {
  887. // It is NULL, set it ourselves.
  888. host = qtl.st->rrdhost;
  889. }
  890. else if (unlikely(host != qtl.st->rrdhost)) {
  891. // Oops! A different host!
  892. error("QUERY TARGET: RRDSET '%s' given does not belong to host '%s'. Switching query host to '%s'",
  893. rrdset_name(qtl.st), rrdhost_hostname(host), rrdhost_hostname(qtl.st->rrdhost));
  894. host = qtl.st->rrdhost;
  895. }
  896. }
  897. if(host) {
  898. // single host query
  899. qt->versions.contexts_hard_hash = dictionary_version(host->rrdctx.contexts);
  900. qt->versions.contexts_soft_hash = dictionary_version(host->rrdctx.hub_queue);
  901. qt->versions.alerts_hard_hash = dictionary_version(host->rrdcalc_root_index);
  902. qt->versions.alerts_soft_hash = __atomic_load_n(&host->health_transitions, __ATOMIC_RELAXED);
  903. query_node_add(&qtl, host, true);
  904. qtl.nodes = rrdhost_hostname(host);
  905. }
  906. else
  907. query_scope_foreach_host(qt->nodes.scope_pattern, qt->nodes.pattern,
  908. query_node_add, &qtl,
  909. &qt->versions,
  910. qtl.host_uuid_buffer);
  911. // we need the available db retention for this call
  912. // so it has to be done last
  913. query_target_calculate_window(qt);
  914. qt->timings.preprocessed_ut = now_monotonic_usec();
  915. return qt;
  916. }
  917. ssize_t weights_foreach_rrdmetric_in_context(RRDCONTEXT_ACQUIRED *rca,
  918. SIMPLE_PATTERN *instances_sp,
  919. SIMPLE_PATTERN *chart_label_key_sp,
  920. SIMPLE_PATTERN *labels_sp,
  921. SIMPLE_PATTERN *alerts_sp,
  922. SIMPLE_PATTERN *dimensions_sp,
  923. bool match_ids, bool match_names,
  924. size_t version,
  925. weights_add_metric_t cb,
  926. void *data
  927. ) {
  928. RRDCONTEXT *rc = rrdcontext_acquired_value(rca);
  929. if(!rc || rrd_flag_is_deleted(rc))
  930. return 0;
  931. bool proceed = true;
  932. ssize_t count = 0;
  933. RRDINSTANCE *ri;
  934. dfe_start_read(rc->rrdinstances, ri) {
  935. if(rrd_flag_is_deleted(ri))
  936. continue;
  937. RRDINSTANCE_ACQUIRED *ria = (RRDINSTANCE_ACQUIRED *) ri_dfe.item;
  938. if(instances_sp) {
  939. QUERY_INSTANCE qi = { .ria = ria, };
  940. SIMPLE_PATTERN_RESULT ret = query_instance_matches(&qi, ri, instances_sp, match_ids, match_names, version);
  941. qi.ria = NULL;
  942. query_instance_release(&qi);
  943. if (ret != SP_MATCHED_POSITIVE)
  944. continue;
  945. }
  946. if(!query_instance_matches_labels(ri, chart_label_key_sp, labels_sp))
  947. continue;
  948. if(alerts_sp && !query_target_match_alert_pattern(ria, alerts_sp))
  949. continue;
  950. dfe_unlock(ri);
  951. RRDMETRIC *rm;
  952. dfe_start_read(ri->rrdmetrics, rm) {
  953. if(rrd_flag_is_deleted(rm))
  954. continue;
  955. if(dimensions_sp) {
  956. SIMPLE_PATTERN_RESULT ret = SP_NOT_MATCHED;
  957. if (match_ids)
  958. ret = simple_pattern_matches_string_extract(dimensions_sp, rm->id, NULL, 0);
  959. if (ret == SP_NOT_MATCHED && match_names && (rm->name != rm->id || !match_ids))
  960. ret = simple_pattern_matches_string_extract(dimensions_sp, rm->name, NULL, 0);
  961. if(ret != SP_MATCHED_POSITIVE)
  962. continue;
  963. }
  964. dfe_unlock(rm);
  965. RRDMETRIC_ACQUIRED *rma = (RRDMETRIC_ACQUIRED *)rm_dfe.item;
  966. ssize_t ret = cb(data, rc->rrdhost, rca, ria, rma);
  967. if(ret < 0) {
  968. proceed = false;
  969. break;
  970. }
  971. count += ret;
  972. }
  973. dfe_done(rm);
  974. if(unlikely(!proceed))
  975. break;
  976. }
  977. dfe_done(ri);
  978. return count;
  979. }