facets.c 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853
  1. #include "facets.h"
  2. #define HISTOGRAM_COLUMNS 60
  3. static void facets_row_free(FACETS *facets __maybe_unused, FACET_ROW *row);
  4. // ----------------------------------------------------------------------------
  5. time_t calculate_bar_width(time_t before, time_t after) {
  6. // Array of valid durations in seconds
  7. static time_t valid_durations[] = {
  8. 1,
  9. 15,
  10. 30,
  11. 1 * 60, 2 * 60, 3 * 60, 5 * 60, 10 * 60, 15 * 60, 30 * 60, // minutes
  12. 1 * 3600, 2 * 3600, 6 * 3600, 8 * 3600, 12 * 3600, // hours
  13. 1 * 86400, 2 * 86400, 3 * 86400, 5 * 86400, 7 * 86400, 14 * 86400, // days
  14. 1 * (30*86400) // months
  15. };
  16. static int array_size = sizeof(valid_durations) / sizeof(valid_durations[0]);
  17. time_t duration = before - after;
  18. time_t bar_width = 1;
  19. for (int i = array_size - 1; i >= 0; --i) {
  20. if (duration / valid_durations[i] >= HISTOGRAM_COLUMNS) {
  21. bar_width = valid_durations[i];
  22. break;
  23. }
  24. }
  25. return bar_width;
  26. }
  27. // ----------------------------------------------------------------------------
  28. static inline void uint32_to_char(uint32_t num, char *out) {
  29. static char id_encoding_characters[64 + 1] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ.abcdefghijklmnopqrstuvwxyz_0123456789";
  30. int i;
  31. for(i = 5; i >= 0; --i) {
  32. out[i] = id_encoding_characters[num & 63];
  33. num >>= 6;
  34. }
  35. out[6] = '\0';
  36. }
  37. inline void facets_string_hash(const char *src, char *out) {
  38. uint32_t hash1 = fnv1a_hash32(src);
  39. uint32_t hash2 = djb2_hash32(src);
  40. uint32_t hash3 = larson_hash32(src);
  41. uint32_to_char(hash1, out);
  42. uint32_to_char(hash2, &out[6]);
  43. uint32_to_char(hash3, &out[12]);
  44. out[18] = '\0';
  45. }
  46. // ----------------------------------------------------------------------------
  47. typedef struct facet_value {
  48. const char *name;
  49. bool selected;
  50. uint32_t rows_matching_facet_value;
  51. uint32_t final_facet_value_counter;
  52. } FACET_VALUE;
  53. struct facet_key {
  54. const char *name;
  55. DICTIONARY *values;
  56. FACET_KEY_OPTIONS options;
  57. bool default_selected_for_values; // the default "selected" for all values in the dictionary
  58. // members about the current row
  59. uint32_t key_found_in_row;
  60. uint32_t key_values_selected_in_row;
  61. struct {
  62. char hash[FACET_STRING_HASH_SIZE];
  63. bool updated;
  64. BUFFER *b;
  65. } current_value;
  66. uint32_t order;
  67. struct {
  68. facet_dynamic_row_t cb;
  69. void *data;
  70. } dynamic;
  71. struct {
  72. facets_key_transformer_t cb;
  73. void *data;
  74. } transform;
  75. struct facet_key *prev, *next;
  76. };
  77. struct facets {
  78. SIMPLE_PATTERN *visible_keys;
  79. SIMPLE_PATTERN *excluded_keys;
  80. SIMPLE_PATTERN *included_keys;
  81. FACETS_OPTIONS options;
  82. usec_t anchor;
  83. SIMPLE_PATTERN *query; // the full text search pattern
  84. size_t keys_filtered_by_query; // the number of fields we do full text search (constant)
  85. size_t keys_matched_by_query; // the number of fields matched the full text search (per row)
  86. DICTIONARY *accepted_params;
  87. FACET_KEY *keys_ll;
  88. DICTIONARY *keys;
  89. FACET_ROW *base; // double linked list of the selected facets rows
  90. uint32_t items_to_return;
  91. uint32_t max_items_to_return;
  92. uint32_t order;
  93. struct {
  94. FACET_ROW *last_added;
  95. size_t evaluated;
  96. size_t matched;
  97. size_t first;
  98. size_t forwards;
  99. size_t backwards;
  100. size_t skips_before;
  101. size_t skips_after;
  102. size_t prepends;
  103. size_t appends;
  104. size_t shifts;
  105. } operations;
  106. };
  107. // ----------------------------------------------------------------------------
  108. static inline void facet_value_is_used(FACET_KEY *k, FACET_VALUE *v) {
  109. if(!k->key_found_in_row)
  110. v->rows_matching_facet_value++;
  111. k->key_found_in_row++;
  112. if(v->selected)
  113. k->key_values_selected_in_row++;
  114. }
  115. static inline bool facets_key_is_facet(FACETS *facets, FACET_KEY *k) {
  116. bool included = true, excluded = false;
  117. if(k->options & (FACET_KEY_OPTION_FACET | FACET_KEY_OPTION_NO_FACET)) {
  118. if(k->options & FACET_KEY_OPTION_FACET) {
  119. included = true;
  120. excluded = false;
  121. }
  122. else if(k->options & FACET_KEY_OPTION_NO_FACET) {
  123. included = false;
  124. excluded = true;
  125. }
  126. }
  127. else {
  128. if (facets->included_keys) {
  129. if (!simple_pattern_matches(facets->included_keys, k->name))
  130. included = false;
  131. }
  132. if (facets->excluded_keys) {
  133. if (simple_pattern_matches(facets->excluded_keys, k->name))
  134. excluded = true;
  135. }
  136. }
  137. if(included && !excluded) {
  138. k->options |= FACET_KEY_OPTION_FACET;
  139. k->options &= ~FACET_KEY_OPTION_NO_FACET;
  140. return true;
  141. }
  142. k->options |= FACET_KEY_OPTION_NO_FACET;
  143. k->options &= ~FACET_KEY_OPTION_FACET;
  144. return false;
  145. }
  146. // ----------------------------------------------------------------------------
  147. // FACET_VALUE dictionary hooks
  148. static void facet_value_insert_callback(const DICTIONARY_ITEM *item __maybe_unused, void *value, void *data) {
  149. FACET_VALUE *v = value;
  150. FACET_KEY *k = data;
  151. if(!v->selected)
  152. v->selected = k->default_selected_for_values;
  153. if(v->name) {
  154. // an actual value, not a filter
  155. v->name = strdupz(v->name);
  156. facet_value_is_used(k, v);
  157. }
  158. }
  159. static bool facet_value_conflict_callback(const DICTIONARY_ITEM *item __maybe_unused, void *old_value, void *new_value, void *data) {
  160. FACET_VALUE *v = old_value;
  161. FACET_VALUE *nv = new_value;
  162. FACET_KEY *k = data;
  163. if(!v->name && nv->name)
  164. // an actual value, not a filter
  165. v->name = strdupz(nv->name);
  166. if(v->name)
  167. facet_value_is_used(k, v);
  168. internal_fatal(v->name && strcmp(v->name, nv->name) != 0, "hash conflict: '%s' and '%s' have the same hash '%s'", v->name, nv->name,
  169. dictionary_acquired_item_name(item));
  170. return false;
  171. }
  172. static void facet_value_delete_callback(const DICTIONARY_ITEM *item __maybe_unused, void *value, void *data __maybe_unused) {
  173. FACET_VALUE *v = value;
  174. freez((char *)v->name);
  175. }
  176. // ----------------------------------------------------------------------------
  177. // FACET_KEY dictionary hooks
  178. static inline void facet_key_late_init(FACETS *facets, FACET_KEY *k) {
  179. if(k->values)
  180. return;
  181. if(facets_key_is_facet(facets, k)) {
  182. k->values = dictionary_create_advanced(
  183. DICT_OPTION_SINGLE_THREADED | DICT_OPTION_DONT_OVERWRITE_VALUE | DICT_OPTION_FIXED_SIZE,
  184. NULL, sizeof(FACET_VALUE));
  185. dictionary_register_insert_callback(k->values, facet_value_insert_callback, k);
  186. dictionary_register_conflict_callback(k->values, facet_value_conflict_callback, k);
  187. dictionary_register_delete_callback(k->values, facet_value_delete_callback, k);
  188. }
  189. }
  190. static void facet_key_insert_callback(const DICTIONARY_ITEM *item __maybe_unused, void *value, void *data) {
  191. FACET_KEY *k = value;
  192. FACETS *facets = data;
  193. if(!(k->options & FACET_KEY_OPTION_REORDER))
  194. k->order = facets->order++;
  195. if((k->options & FACET_KEY_OPTION_FTS) || (facets->options & FACETS_OPTION_ALL_KEYS_FTS))
  196. facets->keys_filtered_by_query++;
  197. if(k->name) {
  198. // an actual value, not a filter
  199. k->name = strdupz(k->name);
  200. facet_key_late_init(facets, k);
  201. }
  202. k->current_value.b = buffer_create(0, NULL);
  203. DOUBLE_LINKED_LIST_APPEND_ITEM_UNSAFE(facets->keys_ll, k, prev, next);
  204. }
  205. static bool facet_key_conflict_callback(const DICTIONARY_ITEM *item __maybe_unused, void *old_value, void *new_value, void *data) {
  206. FACET_KEY *k = old_value;
  207. FACET_KEY *nk = new_value;
  208. FACETS *facets = data;
  209. if(!k->name && nk->name) {
  210. // an actual value, not a filter
  211. k->name = strdupz(nk->name);
  212. facet_key_late_init(facets, k);
  213. }
  214. if(k->options & FACET_KEY_OPTION_REORDER) {
  215. k->order = facets->order++;
  216. k->options &= ~FACET_KEY_OPTION_REORDER;
  217. }
  218. return false;
  219. }
  220. static void facet_key_delete_callback(const DICTIONARY_ITEM *item __maybe_unused, void *value, void *data __maybe_unused) {
  221. FACET_KEY *k = value;
  222. FACETS *facets = data;
  223. DOUBLE_LINKED_LIST_REMOVE_ITEM_UNSAFE(facets->keys_ll, k, prev, next);
  224. dictionary_destroy(k->values);
  225. buffer_free(k->current_value.b);
  226. freez((char *)k->name);
  227. }
  228. // ----------------------------------------------------------------------------
  229. FACETS *facets_create(uint32_t items_to_return, usec_t anchor, FACETS_OPTIONS options, const char *visible_keys, const char *facet_keys, const char *non_facet_keys) {
  230. FACETS *facets = callocz(1, sizeof(FACETS));
  231. facets->options = options;
  232. facets->keys = dictionary_create_advanced(DICT_OPTION_SINGLE_THREADED|DICT_OPTION_DONT_OVERWRITE_VALUE|DICT_OPTION_FIXED_SIZE, NULL, sizeof(FACET_KEY));
  233. dictionary_register_insert_callback(facets->keys, facet_key_insert_callback, facets);
  234. dictionary_register_conflict_callback(facets->keys, facet_key_conflict_callback, facets);
  235. dictionary_register_delete_callback(facets->keys, facet_key_delete_callback, facets);
  236. if(facet_keys && *facet_keys)
  237. facets->included_keys = simple_pattern_create(facet_keys, "|", SIMPLE_PATTERN_EXACT, true);
  238. if(non_facet_keys && *non_facet_keys)
  239. facets->excluded_keys = simple_pattern_create(non_facet_keys, "|", SIMPLE_PATTERN_EXACT, true);
  240. if(visible_keys && *visible_keys)
  241. facets->visible_keys = simple_pattern_create(visible_keys, "|", SIMPLE_PATTERN_EXACT, true);
  242. facets->max_items_to_return = items_to_return;
  243. facets->anchor = anchor;
  244. facets->order = 1;
  245. return facets;
  246. }
  247. void facets_destroy(FACETS *facets) {
  248. dictionary_destroy(facets->accepted_params);
  249. dictionary_destroy(facets->keys);
  250. simple_pattern_free(facets->visible_keys);
  251. simple_pattern_free(facets->included_keys);
  252. simple_pattern_free(facets->excluded_keys);
  253. while(facets->base) {
  254. FACET_ROW *r = facets->base;
  255. DOUBLE_LINKED_LIST_REMOVE_ITEM_UNSAFE(facets->base, r, prev, next);
  256. facets_row_free(facets, r);
  257. }
  258. freez(facets);
  259. }
  260. void facets_accepted_param(FACETS *facets, const char *param) {
  261. if(!facets->accepted_params)
  262. facets->accepted_params = dictionary_create(DICT_OPTION_SINGLE_THREADED|DICT_OPTION_DONT_OVERWRITE_VALUE);
  263. dictionary_set(facets->accepted_params, param, NULL, 0);
  264. }
  265. inline FACET_KEY *facets_register_key(FACETS *facets, const char *key, FACET_KEY_OPTIONS options) {
  266. FACET_KEY tk = {
  267. .name = key,
  268. .options = options,
  269. .default_selected_for_values = true,
  270. };
  271. char hash[FACET_STRING_HASH_SIZE];
  272. facets_string_hash(tk.name, hash);
  273. return dictionary_set(facets->keys, hash, &tk, sizeof(tk));
  274. }
  275. inline FACET_KEY *facets_register_key_transformation(FACETS *facets, const char *key, FACET_KEY_OPTIONS options, facets_key_transformer_t cb, void *data) {
  276. FACET_KEY *k = facets_register_key(facets, key, options);
  277. k->transform.cb = cb;
  278. k->transform.data = data;
  279. return k;
  280. }
  281. inline FACET_KEY *facets_register_dynamic_key(FACETS *facets, const char *key, FACET_KEY_OPTIONS options, facet_dynamic_row_t cb, void *data) {
  282. FACET_KEY *k = facets_register_key(facets, key, options);
  283. k->dynamic.cb = cb;
  284. k->dynamic.data = data;
  285. return k;
  286. }
  287. void facets_set_query(FACETS *facets, const char *query) {
  288. if(!query)
  289. return;
  290. facets->query = simple_pattern_create(query, " \t", SIMPLE_PATTERN_SUBSTRING, false);
  291. }
  292. void facets_set_items(FACETS *facets, uint32_t items) {
  293. facets->max_items_to_return = items;
  294. }
  295. void facets_set_anchor(FACETS *facets, usec_t anchor) {
  296. facets->anchor = anchor;
  297. }
  298. void facets_register_facet_filter(FACETS *facets, const char *key_id, char *value_ids, FACET_KEY_OPTIONS options) {
  299. FACET_KEY tk = {
  300. .options = options,
  301. };
  302. FACET_KEY *k = dictionary_set(facets->keys, key_id, &tk, sizeof(tk));
  303. k->default_selected_for_values = false;
  304. k->options |= FACET_KEY_OPTION_FACET;
  305. k->options &= ~FACET_KEY_OPTION_NO_FACET;
  306. facet_key_late_init(facets, k);
  307. FACET_VALUE tv = {
  308. .selected = true,
  309. };
  310. dictionary_set(k->values, value_ids, &tv, sizeof(tv));
  311. }
  312. // ----------------------------------------------------------------------------
  313. static inline void facets_check_value(FACETS *facets __maybe_unused, FACET_KEY *k) {
  314. if(!k->current_value.updated)
  315. buffer_flush(k->current_value.b);
  316. if(k->transform.cb)
  317. k->transform.cb(facets, k->current_value.b, k->transform.data);
  318. if(!k->current_value.updated) {
  319. buffer_strcat(k->current_value.b, FACET_VALUE_UNSET);
  320. k->current_value.updated = true;
  321. }
  322. // bool found = false;
  323. // if(strstr(buffer_tostring(k->current_value), "fprintd") != NULL)
  324. // found = true;
  325. if(facets->query && ((k->options & FACET_KEY_OPTION_FTS) || facets->options & FACETS_OPTION_ALL_KEYS_FTS)) {
  326. if(simple_pattern_matches(facets->query, buffer_tostring(k->current_value.b)))
  327. facets->keys_matched_by_query++;
  328. }
  329. if(k->values) {
  330. FACET_VALUE tk = {
  331. .name = buffer_tostring(k->current_value.b),
  332. };
  333. facets_string_hash(tk.name, k->current_value.hash);
  334. dictionary_set(k->values, k->current_value.hash, &tk, sizeof(tk));
  335. }
  336. else {
  337. k->key_found_in_row++;
  338. k->key_values_selected_in_row++;
  339. }
  340. }
  341. void facets_add_key_value(FACETS *facets, const char *key, const char *value) {
  342. FACET_KEY *k = facets_register_key(facets, key, 0);
  343. buffer_flush(k->current_value.b);
  344. buffer_strcat(k->current_value.b, value);
  345. k->current_value.updated = true;
  346. facets_check_value(facets, k);
  347. }
  348. void facets_add_key_value_length(FACETS *facets, const char *key, const char *value, size_t value_len) {
  349. FACET_KEY *k = facets_register_key(facets, key, 0);
  350. buffer_flush(k->current_value.b);
  351. buffer_strncat(k->current_value.b, value, value_len);
  352. k->current_value.updated = true;
  353. facets_check_value(facets, k);
  354. }
  355. // ----------------------------------------------------------------------------
  356. // FACET_ROW dictionary hooks
  357. static void facet_row_key_value_insert_callback(const DICTIONARY_ITEM *item __maybe_unused, void *value, void *data) {
  358. FACET_ROW_KEY_VALUE *rkv = value;
  359. FACET_ROW *row = data; (void)row;
  360. rkv->wb = buffer_create(0, NULL);
  361. buffer_strcat(rkv->wb, rkv->tmp && *rkv->tmp ? rkv->tmp : FACET_VALUE_UNSET);
  362. }
  363. static bool facet_row_key_value_conflict_callback(const DICTIONARY_ITEM *item __maybe_unused, void *old_value, void *new_value, void *data) {
  364. FACET_ROW_KEY_VALUE *rkv = old_value;
  365. FACET_ROW_KEY_VALUE *n_rkv = new_value;
  366. FACET_ROW *row = data; (void)row;
  367. buffer_flush(rkv->wb);
  368. buffer_strcat(rkv->wb, n_rkv->tmp && *n_rkv->tmp ? n_rkv->tmp : FACET_VALUE_UNSET);
  369. return false;
  370. }
  371. static void facet_row_key_value_delete_callback(const DICTIONARY_ITEM *item __maybe_unused, void *value, void *data) {
  372. FACET_ROW_KEY_VALUE *rkv = value;
  373. FACET_ROW *row = data; (void)row;
  374. buffer_free(rkv->wb);
  375. }
  376. // ----------------------------------------------------------------------------
  377. // FACET_ROW management
  378. static void facets_row_free(FACETS *facets __maybe_unused, FACET_ROW *row) {
  379. dictionary_destroy(row->dict);
  380. freez(row);
  381. }
  382. static FACET_ROW *facets_row_create(FACETS *facets, usec_t usec, FACET_ROW *into) {
  383. FACET_ROW *row;
  384. if(into)
  385. row = into;
  386. else {
  387. row = callocz(1, sizeof(FACET_ROW));
  388. row->dict = dictionary_create_advanced(DICT_OPTION_SINGLE_THREADED|DICT_OPTION_DONT_OVERWRITE_VALUE|DICT_OPTION_FIXED_SIZE, NULL, sizeof(FACET_ROW_KEY_VALUE));
  389. dictionary_register_insert_callback(row->dict, facet_row_key_value_insert_callback, row);
  390. dictionary_register_conflict_callback(row->dict, facet_row_key_value_conflict_callback, row);
  391. dictionary_register_delete_callback(row->dict, facet_row_key_value_delete_callback, row);
  392. }
  393. row->usec = usec;
  394. FACET_KEY *k;
  395. dfe_start_read(facets->keys, k) {
  396. FACET_ROW_KEY_VALUE t = {
  397. .tmp = (k->current_value.updated && buffer_strlen(k->current_value.b)) ?
  398. buffer_tostring(k->current_value.b) : FACET_VALUE_UNSET,
  399. .wb = NULL,
  400. };
  401. dictionary_set(row->dict, k->name, &t, sizeof(t));
  402. }
  403. dfe_done(k);
  404. return row;
  405. }
  406. // ----------------------------------------------------------------------------
  407. static void facets_row_keep(FACETS *facets, usec_t usec) {
  408. facets->operations.matched++;
  409. if(usec < facets->anchor) {
  410. facets->operations.skips_before++;
  411. return;
  412. }
  413. if(unlikely(!facets->base)) {
  414. facets->operations.last_added = facets_row_create(facets, usec, NULL);
  415. DOUBLE_LINKED_LIST_APPEND_ITEM_UNSAFE(facets->base, facets->operations.last_added, prev, next);
  416. facets->items_to_return++;
  417. facets->operations.first++;
  418. return;
  419. }
  420. if(likely(usec > facets->base->prev->usec))
  421. facets->operations.last_added = facets->base->prev;
  422. FACET_ROW *last = facets->operations.last_added;
  423. while(last->prev != facets->base->prev && usec > last->prev->usec) {
  424. last = last->prev;
  425. facets->operations.backwards++;
  426. }
  427. while(last->next && usec < last->next->usec) {
  428. last = last->next;
  429. facets->operations.forwards++;
  430. }
  431. if(facets->items_to_return >= facets->max_items_to_return) {
  432. if(last == facets->base->prev && usec < last->usec) {
  433. facets->operations.skips_after++;
  434. return;
  435. }
  436. }
  437. facets->items_to_return++;
  438. if(usec > last->usec) {
  439. if(facets->items_to_return > facets->max_items_to_return) {
  440. facets->items_to_return--;
  441. facets->operations.shifts++;
  442. facets->operations.last_added = facets->base->prev;
  443. DOUBLE_LINKED_LIST_REMOVE_ITEM_UNSAFE(facets->base, facets->operations.last_added, prev, next);
  444. facets->operations.last_added = facets_row_create(facets, usec, facets->operations.last_added);
  445. }
  446. DOUBLE_LINKED_LIST_PREPEND_ITEM_UNSAFE(facets->base, facets->operations.last_added, prev, next);
  447. facets->operations.prepends++;
  448. }
  449. else {
  450. facets->operations.last_added = facets_row_create(facets, usec, NULL);
  451. DOUBLE_LINKED_LIST_APPEND_ITEM_UNSAFE(facets->base, facets->operations.last_added, prev, next);
  452. facets->operations.appends++;
  453. }
  454. while(facets->items_to_return > facets->max_items_to_return) {
  455. // we have to remove something
  456. FACET_ROW *tmp = facets->base->prev;
  457. DOUBLE_LINKED_LIST_REMOVE_ITEM_UNSAFE(facets->base, tmp, prev, next);
  458. facets->items_to_return--;
  459. if(unlikely(facets->operations.last_added == tmp))
  460. facets->operations.last_added = facets->base->prev;
  461. facets_row_free(facets, tmp);
  462. facets->operations.shifts++;
  463. }
  464. }
  465. void facets_rows_begin(FACETS *facets) {
  466. FACET_KEY *k;
  467. // dfe_start_read(facets->keys, k) {
  468. for(k = facets->keys_ll ; k ; k = k->next) {
  469. k->key_found_in_row = 0;
  470. k->key_values_selected_in_row = 0;
  471. k->current_value.updated = false;
  472. k->current_value.hash[0] = '\0';
  473. }
  474. // dfe_done(k);
  475. facets->keys_matched_by_query = 0;
  476. }
  477. void facets_row_finished(FACETS *facets, usec_t usec) {
  478. if(facets->query && facets->keys_filtered_by_query && !facets->keys_matched_by_query)
  479. goto cleanup;
  480. facets->operations.evaluated++;
  481. uint32_t total_keys = 0;
  482. uint32_t selected_by = 0;
  483. FACET_KEY *k;
  484. // dfe_start_read(facets->keys, k) {
  485. for(k = facets->keys_ll ; k ; k = k->next) {
  486. if(!k->key_found_in_row) {
  487. // put the FACET_VALUE_UNSET value into it
  488. facets_check_value(facets, k);
  489. }
  490. internal_fatal(!k->key_found_in_row, "all keys should be found in the row at this point");
  491. internal_fatal(k->key_found_in_row != 1, "all keys should be matched exactly once at this point");
  492. internal_fatal(k->key_values_selected_in_row > 1, "key values are selected in row more than once");
  493. k->key_found_in_row = 1;
  494. total_keys += k->key_found_in_row;
  495. selected_by += (k->key_values_selected_in_row) ? 1 : 0;
  496. }
  497. // dfe_done(k);
  498. if(selected_by >= total_keys - 1) {
  499. uint32_t found = 0;
  500. // dfe_start_read(facets->keys, k){
  501. for(k = facets->keys_ll ; k ; k = k->next) {
  502. uint32_t counted_by = selected_by;
  503. if (counted_by != total_keys && !k->key_values_selected_in_row)
  504. counted_by++;
  505. if(counted_by == total_keys) {
  506. if(k->values) {
  507. if(!k->current_value.hash[0])
  508. facets_string_hash(buffer_tostring(k->current_value.b), k->current_value.hash);
  509. FACET_VALUE *v = dictionary_get(k->values, k->current_value.hash);
  510. v->final_facet_value_counter++;
  511. }
  512. found++;
  513. }
  514. }
  515. // dfe_done(k);
  516. internal_fatal(!found, "We should find at least one facet to count this row");
  517. (void)found;
  518. }
  519. if(selected_by == total_keys)
  520. facets_row_keep(facets, usec);
  521. cleanup:
  522. facets_rows_begin(facets);
  523. }
  524. // ----------------------------------------------------------------------------
  525. // output
  526. void facets_report(FACETS *facets, BUFFER *wb) {
  527. buffer_json_member_add_boolean(wb, "show_ids", false);
  528. buffer_json_member_add_boolean(wb, "has_history", true);
  529. buffer_json_member_add_object(wb, "pagination");
  530. buffer_json_member_add_boolean(wb, "enabled", true);
  531. buffer_json_member_add_string(wb, "key", "anchor");
  532. buffer_json_member_add_string(wb, "column", "timestamp");
  533. buffer_json_object_close(wb);
  534. buffer_json_member_add_array(wb, "accepted_params");
  535. {
  536. if(facets->accepted_params) {
  537. void *t;
  538. dfe_start_read(facets->accepted_params, t) {
  539. buffer_json_add_array_item_string(wb, t_dfe.name);
  540. }
  541. dfe_done(t);
  542. }
  543. FACET_KEY *k;
  544. dfe_start_read(facets->keys, k) {
  545. if(!k->values)
  546. continue;
  547. buffer_json_add_array_item_string(wb, k_dfe.name);
  548. }
  549. dfe_done(k);
  550. }
  551. buffer_json_array_close(wb); // accepted_params
  552. buffer_json_member_add_array(wb, "facets");
  553. {
  554. FACET_KEY *k;
  555. dfe_start_read(facets->keys, k) {
  556. if(!k->values)
  557. continue;
  558. buffer_json_add_array_item_object(wb); // key
  559. {
  560. buffer_json_member_add_string(wb, "id", k_dfe.name);
  561. buffer_json_member_add_string(wb, "name", k->name);
  562. if(!k->order)
  563. k->order = facets->order++;
  564. buffer_json_member_add_uint64(wb, "order", k->order);
  565. buffer_json_member_add_array(wb, "options");
  566. {
  567. FACET_VALUE *v;
  568. dfe_start_read(k->values, v) {
  569. buffer_json_add_array_item_object(wb);
  570. {
  571. buffer_json_member_add_string(wb, "id", v_dfe.name);
  572. buffer_json_member_add_string(wb, "name", v->name);
  573. buffer_json_member_add_uint64(wb, "count", v->final_facet_value_counter);
  574. }
  575. buffer_json_object_close(wb);
  576. }
  577. dfe_done(v);
  578. }
  579. buffer_json_array_close(wb); // options
  580. }
  581. buffer_json_object_close(wb); // key
  582. }
  583. dfe_done(k);
  584. }
  585. buffer_json_array_close(wb); // facets
  586. buffer_json_member_add_object(wb, "columns");
  587. {
  588. size_t field_id = 0;
  589. buffer_rrdf_table_add_field(
  590. wb, field_id++,
  591. "timestamp", "Timestamp",
  592. RRDF_FIELD_TYPE_TIMESTAMP,
  593. RRDF_FIELD_VISUAL_VALUE,
  594. RRDF_FIELD_TRANSFORM_DATETIME_USEC, 0, NULL, NAN,
  595. RRDF_FIELD_SORT_DESCENDING,
  596. NULL,
  597. RRDF_FIELD_SUMMARY_COUNT,
  598. RRDF_FIELD_FILTER_RANGE,
  599. RRDF_FIELD_OPTS_VISIBLE | RRDF_FIELD_OPTS_UNIQUE_KEY,
  600. NULL);
  601. FACET_KEY *k;
  602. dfe_start_read(facets->keys, k) {
  603. RRDF_FIELD_OPTIONS options = RRDF_FIELD_OPTS_NONE;
  604. bool visible = k->options & (FACET_KEY_OPTION_VISIBLE|FACET_KEY_OPTION_STICKY);
  605. if((facets->options & FACETS_OPTION_ALL_FACETS_VISIBLE && k->values))
  606. visible = true;
  607. if(!visible)
  608. visible = simple_pattern_matches(facets->visible_keys, k->name);
  609. if(visible)
  610. options |= RRDF_FIELD_OPTS_VISIBLE;
  611. if(k->options & FACET_KEY_OPTION_MAIN_TEXT)
  612. options |= RRDF_FIELD_OPTS_FULL_WIDTH | RRDF_FIELD_OPTS_WRAP;
  613. buffer_rrdf_table_add_field(
  614. wb, field_id++,
  615. k_dfe.name, k->name ? k->name : k_dfe.name,
  616. RRDF_FIELD_TYPE_STRING,
  617. RRDF_FIELD_VISUAL_VALUE,
  618. RRDF_FIELD_TRANSFORM_NONE, 0, NULL, NAN,
  619. RRDF_FIELD_SORT_ASCENDING,
  620. NULL,
  621. RRDF_FIELD_SUMMARY_COUNT,
  622. k->values ? RRDF_FIELD_FILTER_FACET : RRDF_FIELD_FILTER_NONE,
  623. options,
  624. FACET_VALUE_UNSET);
  625. }
  626. dfe_done(k);
  627. }
  628. buffer_json_object_close(wb); // columns
  629. buffer_json_member_add_array(wb, "data");
  630. {
  631. for(FACET_ROW *row = facets->base ; row ;row = row->next) {
  632. buffer_json_add_array_item_array(wb); // each row
  633. buffer_json_add_array_item_uint64(wb, row->usec);
  634. FACET_KEY *k;
  635. dfe_start_read(facets->keys, k)
  636. {
  637. FACET_ROW_KEY_VALUE *rkv = dictionary_get(row->dict, k->name);
  638. if(unlikely(k->dynamic.cb)) {
  639. if(unlikely(!rkv))
  640. rkv = dictionary_set(row->dict, k->name, NULL, sizeof(*rkv));
  641. k->dynamic.cb(facets, wb, rkv, row, k->dynamic.data);
  642. }
  643. else
  644. buffer_json_add_array_item_string(wb, rkv ? buffer_tostring(rkv->wb) : FACET_VALUE_UNSET);
  645. }
  646. dfe_done(k);
  647. buffer_json_array_close(wb); // each row
  648. }
  649. }
  650. buffer_json_array_close(wb); // data
  651. buffer_json_member_add_string(wb, "default_sort_column", "timestamp");
  652. buffer_json_member_add_array(wb, "default_charts");
  653. buffer_json_array_close(wb);
  654. buffer_json_member_add_object(wb, "items");
  655. {
  656. buffer_json_member_add_uint64(wb, "evaluated", facets->operations.evaluated);
  657. buffer_json_member_add_uint64(wb, "matched", facets->operations.matched);
  658. buffer_json_member_add_uint64(wb, "returned", facets->items_to_return);
  659. buffer_json_member_add_uint64(wb, "max_to_return", facets->max_items_to_return);
  660. buffer_json_member_add_uint64(wb, "before", facets->operations.skips_before);
  661. buffer_json_member_add_uint64(wb, "after", facets->operations.skips_after + facets->operations.shifts);
  662. }
  663. buffer_json_object_close(wb); // items
  664. buffer_json_member_add_object(wb, "stats");
  665. {
  666. buffer_json_member_add_uint64(wb, "first", facets->operations.first);
  667. buffer_json_member_add_uint64(wb, "forwards", facets->operations.forwards);
  668. buffer_json_member_add_uint64(wb, "backwards", facets->operations.backwards);
  669. buffer_json_member_add_uint64(wb, "skips_before", facets->operations.skips_before);
  670. buffer_json_member_add_uint64(wb, "skips_after", facets->operations.skips_after);
  671. buffer_json_member_add_uint64(wb, "prepends", facets->operations.prepends);
  672. buffer_json_member_add_uint64(wb, "appends", facets->operations.appends);
  673. buffer_json_member_add_uint64(wb, "shifts", facets->operations.shifts);
  674. }
  675. buffer_json_object_close(wb); // items
  676. }