Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-13 12:08:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fix how cluster framework tracks worker count

Browse source
This commit is contained in:
Jon Siwek 2018-05-24 14:32:45 -05:00
parent 186d47c762
commit 95ea84e60e
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
scripts/base/frameworks/cluster/main.bro
View file

@ -238,6 +238,8 @@ export {
global node_topic: function(name: string): string; global node_topic: function(name: string): string;
} }
global active_worker_ids: set[string] = set();
type NamedNode: record { type NamedNode: record {
name: string; name: string;
node: Node; node: Node;
@ -305,7 +307,10 @@ event Cluster::hello(name: string, id: string) &priority=10
Cluster::log(fmt("got hello from %s (%s)", name, id)); Cluster::log(fmt("got hello from %s (%s)", name, id));
if ( n$node_type == WORKER ) if ( n$node_type == WORKER )
++worker_count; {
add active_worker_ids[id];
worker_count = |active_worker_ids|;
}
} }
event Broker::peer_added(endpoint: Broker::EndpointInfo, msg: string) &priority=10 event Broker::peer_added(endpoint: Broker::EndpointInfo, msg: string) &priority=10
@ -329,7 +334,10 @@ event Broker::peer_lost(endpoint: Broker::EndpointInfo, msg: string) &priority=1
delete n$id; delete n$id;
if ( n$node_type == WORKER ) if ( n$node_type == WORKER )
--worker_count; {
delete active_worker_ids[endpoint$id];
worker_count = |active_worker_ids|;
}
event Cluster::node_down(node_name, endpoint$id); event Cluster::node_down(node_name, endpoint$id);
break; break;