Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-04 15:48:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

change plugin after feedback of seth

Browse source
This commit is contained in:
Bernhard Amann 2013-04-10 10:45:45 -04:00
parent 07d44f3aa0
commit f10ed9e29a
2 changed files with 19 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

25
scripts/base/frameworks/measurement/plugins/hll_unique.bro
View file

@ -10,14 +10,23 @@ export {
redef record ResultVal += {
## If cardinality is being tracked, the number of unique
## items is tracked here.
hllunique: opaque of cardinality &default=hll_cardinality_init(0.01);
hllunique: count &default=0;
};
}
redef record ResultVal += {
# Internal use only. This is not meant to be publically available
# because probabilistic data structures have to be examined using
# specialized bifs.
card: opaque of cardinality &default=hll_cardinality_init(0.01);
};
hook init_resultval_hook(r: Reducer, rv: ResultVal)
{
if ( HLLUNIQUE in r$apply && ! rv?$hllunique )
rv$hllunique = hll_cardinality_init(0.01);
if ( HLLUNIQUE in r$apply && ! rv?$card )
rv$card = hll_cardinality_init(0.01);
rv$hllunique = 0;
}
@ -25,15 +34,17 @@ hook add_to_reducer_hook(r: Reducer, val: double, data: DataPoint, rv: ResultVal
{
if ( HLLUNIQUE in r$apply )
{
hll_cardinality_add(rv$hllunique, data);
hll_cardinality_add(rv$card, data);
rv$hllunique = double_to_count(hll_cardinality_estimate(rv$card));
}
}
hook compose_resultvals_hook(result: ResultVal, rv1: ResultVal, rv2: ResultVal)
{
local rhll = hll_cardinality_init(0.01);
hll_cardinality_merge_into(rhll, rv1$hllunique);
hll_cardinality_merge_into(rhll, rv2$hllunique);
hll_cardinality_merge_into(rhll, rv1$card);
hll_cardinality_merge_into(rhll, rv2$card);
result$hllunique = rhll;
result$card = rhll;
result$hllunique = double_to_count(hll_cardinality_estimate(rhll));
}