1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// SPDX-License-Identifier: GPL-3.0-or-later
#include "internal.h"
ssize_t query_scope_foreach_host(SIMPLE_PATTERN *scope_hosts_sp, SIMPLE_PATTERN *hosts_sp,
foreach_host_cb_t cb, void *data,
struct query_versions *versions,
char *host_node_id_str) {
char uuid[UUID_STR_LEN];
if(!host_node_id_str) host_node_id_str = uuid;
host_node_id_str[0] = '\0';
RRDHOST *host;
ssize_t added = 0;
uint64_t v_hash = 0;
uint64_t h_hash = 0;
uint64_t a_hash = 0;
uint64_t t_hash = 0;
dfe_start_read(rrdhost_root_index, host) {
if(!UUIDiszero(host->node_id))
uuid_unparse_lower(host->node_id.uuid, host_node_id_str);
else
host_node_id_str[0] = '\0';
SIMPLE_PATTERN_RESULT match = SP_MATCHED_POSITIVE;
if(scope_hosts_sp) {
match = simple_pattern_matches_string_extract(scope_hosts_sp, host->hostname, NULL, 0);
if(match == SP_NOT_MATCHED) {
match = simple_pattern_matches_extract(scope_hosts_sp, host->machine_guid, NULL, 0);
if(match == SP_NOT_MATCHED && *host_node_id_str)
match = simple_pattern_matches_extract(scope_hosts_sp, host_node_id_str, NULL, 0);
}
}
if(match != SP_MATCHED_POSITIVE)
continue;
dfe_unlock(host);
if(hosts_sp) {
match = simple_pattern_matches_string_extract(hosts_sp, host->hostname, NULL, 0);
if(match == SP_NOT_MATCHED) {
match = simple_pattern_matches_extract(hosts_sp, host->machine_guid, NULL, 0);
if(match == SP_NOT_MATCHED && *host_node_id_str)
match = simple_pattern_matches_extract(hosts_sp, host_node_id_str, NULL, 0);
}
}
bool queryable_host = (match == SP_MATCHED_POSITIVE);
v_hash += dictionary_version(host->rrdctx.contexts);
h_hash += dictionary_version(host->rrdctx.hub_queue);
a_hash += dictionary_version(host->rrdcalc_root_index);
t_hash += __atomic_load_n(&host->health_transitions, __ATOMIC_RELAXED);
ssize_t ret = cb(data, host, queryable_host);
if(ret < 0) {
added = ret;
break;
}
added += ret;
}
dfe_done(host);
if(versions) {
versions->contexts_hard_hash = v_hash;
versions->contexts_soft_hash = h_hash;
versions->alerts_hard_hash = a_hash;
versions->alerts_soft_hash = t_hash;
}
return added;
}
ssize_t query_scope_foreach_context(RRDHOST *host, const char *scope_contexts, SIMPLE_PATTERN *scope_contexts_sp,
SIMPLE_PATTERN *contexts_sp, foreach_context_cb_t cb, bool queryable_host, void *data) {
if(unlikely(!host->rrdctx.contexts))
return 0;
ssize_t added = 0;
RRDCONTEXT_ACQUIRED *rca = NULL;
if(scope_contexts)
rca = (RRDCONTEXT_ACQUIRED *)dictionary_get_and_acquire_item(host->rrdctx.contexts, scope_contexts);
if(likely(rca)) {
// we found it!
bool queryable_context = queryable_host;
RRDCONTEXT *rc = rrdcontext_acquired_value(rca);
if(queryable_context && contexts_sp && !simple_pattern_matches_string(contexts_sp, rc->id))
queryable_context = false;
added = cb(data, rca, queryable_context);
rrdcontext_release(rca);
}
else {
// Probably it is a pattern, we need to search for it...
RRDCONTEXT *rc;
dfe_start_read(host->rrdctx.contexts, rc) {
if(scope_contexts_sp && !simple_pattern_matches_string(scope_contexts_sp, rc->id))
continue;
dfe_unlock(rc);
bool queryable_context = queryable_host;
if(queryable_context && contexts_sp && !simple_pattern_matches_string(contexts_sp, rc->id))
queryable_context = false;
ssize_t ret = cb(data, (RRDCONTEXT_ACQUIRED *)rc_dfe.item, queryable_context);
if(ret < 0) {
added = ret;
break;
}
added += ret;
}
dfe_done(rc);
}
return added;
}
|
