mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 14:48:21 +00:00
104 lines
2.6 KiB
Text
104 lines
2.6 KiB
Text
# @TEST-PORT: BROKER_MANAGER_PORT
|
|
# @TEST-PORT: BROKER_WORKER1_PORT
|
|
# @TEST-PORT: BROKER_WORKER2_PORT
|
|
#
|
|
# @TEST-EXEC: cp $FILES/broker/cluster-layout.zeek .
|
|
#
|
|
# @TEST-EXEC: zeek -b %INPUT>out
|
|
# @TEST-EXEC: btest-bg-run manager ZEEKPATH=$ZEEKPATH:.. CLUSTER_NODE=manager zeek -b %INPUT
|
|
# @TEST-EXEC: btest-bg-run worker-1 ZEEKPATH=$ZEEKPATH:.. CLUSTER_NODE=worker-1 zeek -b %INPUT runnumber=1
|
|
# @TEST-EXEC: btest-bg-run worker-2 ZEEKPATH=$ZEEKPATH:.. CLUSTER_NODE=worker-2 zeek -b %INPUT runnumber=2
|
|
# @TEST-EXEC: btest-bg-wait 30
|
|
#
|
|
# @TEST-EXEC: btest-diff manager/.stdout
|
|
# @TEST-EXEC: btest-diff worker-1/.stdout
|
|
# @TEST-EXEC: btest-diff worker-2/.stdout
|
|
|
|
@load base/frameworks/cluster
|
|
|
|
redef Log::default_rotation_interval = 0secs;
|
|
|
|
global hll_data: event(data: opaque of cardinality);
|
|
|
|
@if ( Cluster::local_node_type() == Cluster::WORKER )
|
|
|
|
global runnumber: count &redef; # differentiate runs
|
|
|
|
event Broker::peer_lost(endpoint: Broker::EndpointInfo, msg: string)
|
|
{
|
|
terminate();
|
|
}
|
|
|
|
event Broker::peer_added(endpoint: Broker::EndpointInfo, msg: string)
|
|
{
|
|
local c = hll_cardinality_init(0.01, 0.95);
|
|
|
|
local add1 = 2001;
|
|
local add2 = 2002;
|
|
local add3 = 2003;
|
|
|
|
if ( runnumber == 1 )
|
|
{
|
|
hll_cardinality_add(c, add1);
|
|
hll_cardinality_add(c, add2);
|
|
hll_cardinality_add(c, add3);
|
|
hll_cardinality_add(c, 1000);
|
|
hll_cardinality_add(c, 1001);
|
|
hll_cardinality_add(c, 101);
|
|
hll_cardinality_add(c, 1003);
|
|
hll_cardinality_add(c, 1004);
|
|
hll_cardinality_add(c, 1005);
|
|
hll_cardinality_add(c, 1006);
|
|
hll_cardinality_add(c, 1007);
|
|
hll_cardinality_add(c, 1008);
|
|
hll_cardinality_add(c, 1009);
|
|
print "This value should be around 13:";
|
|
print hll_cardinality_estimate(c);
|
|
}
|
|
else if ( runnumber == 2 )
|
|
{
|
|
hll_cardinality_add(c, add1);
|
|
hll_cardinality_add(c, add2);
|
|
hll_cardinality_add(c, add3);
|
|
hll_cardinality_add(c, 1);
|
|
hll_cardinality_add(c, 101);
|
|
hll_cardinality_add(c, 2);
|
|
hll_cardinality_add(c, 3);
|
|
hll_cardinality_add(c, 4);
|
|
hll_cardinality_add(c, 5);
|
|
hll_cardinality_add(c, 6);
|
|
hll_cardinality_add(c, 7);
|
|
hll_cardinality_add(c, 8);
|
|
print "This value should be about 12:";
|
|
print hll_cardinality_estimate(c);
|
|
}
|
|
|
|
Broker::publish(Cluster::manager_topic, hll_data, c);
|
|
}
|
|
|
|
@endif
|
|
|
|
@if ( Cluster::local_node_type() == Cluster::MANAGER )
|
|
|
|
global result_count = 0;
|
|
global hll: opaque of cardinality;
|
|
|
|
event zeek_init()
|
|
{
|
|
hll = hll_cardinality_init(0.01, 0.95);
|
|
}
|
|
|
|
event hll_data(data: opaque of cardinality)
|
|
{
|
|
hll_cardinality_merge_into(hll, data);
|
|
++result_count;
|
|
|
|
if ( result_count == 2 )
|
|
{
|
|
print "This value should be about 21:";
|
|
print hll_cardinality_estimate(hll);
|
|
terminate();
|
|
}
|
|
}
|
|
|
|
@endif
|