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

Add hook into cluster connection setup.

Browse source
This commit is contained in:
Jan Grashoefer 2023-04-21 17:38:19 +02:00
parent c7626d797f
commit 88c86cc7d4
2 changed files with 22 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
scripts/base/frameworks/cluster/main.zeek
View file

@ -191,6 +191,12 @@ export {
id: string &optional; id: string &optional;
}; };
## Record to represent a cluster node including its name.
type NamedNode: record {
name: string;
node: Node;
};
## This function can be called at any time to determine if the cluster ## This function can be called at any time to determine if the cluster
## framework is being enabled for this run. ## framework is being enabled for this run.
## ##
@ -281,11 +287,6 @@ export {
# Track active nodes per type. # Track active nodes per type.
global active_node_ids: table[NodeType] of set[string]; global active_node_ids: table[NodeType] of set[string];
type NamedNode: record {
name: string;
node: Node;
};
function nodes_with_type(node_type: NodeType): vector of NamedNode function nodes_with_type(node_type: NodeType): vector of NamedNode
{ {
local rval: vector of NamedNode = vector(); local rval: vector of NamedNode = vector();

16
scripts/base/frameworks/cluster/setup-connections.zeek
View file

@ -7,6 +7,15 @@
module Cluster; module Cluster;
export {
## This hook is called when the local node connects to other nodes based on
## the given cluster layout. Breaking from the hook will prevent connection
## establishment.
##
## connectee: The node to connect to.
global connect_node_hook: hook(connectee: NamedNode);
}
function connect_peer(node_type: NodeType, node_name: string) function connect_peer(node_type: NodeType, node_name: string)
{ {
local nn = nodes_with_type(node_type); local nn = nodes_with_type(node_type);
@ -17,12 +26,15 @@ function connect_peer(node_type: NodeType, node_name: string)
if ( n$name != node_name ) if ( n$name != node_name )
next; next;
if ( ! hook connect_node_hook(n) )
return;
local status = Broker::peer(cat(n$node$ip), n$node$p, local status = Broker::peer(cat(n$node$ip), n$node$p,
Cluster::retry_interval); Cluster::retry_interval);
Cluster::log(fmt("initiate peering with %s:%s, retry=%s, status=%s", Cluster::log(fmt("initiate peering with %s:%s, retry=%s, status=%s",
n$node$ip, n$node$p, Cluster::retry_interval, n$node$ip, n$node$p, Cluster::retry_interval,
status)); status));
return;
} }
} }
@ -33,6 +45,10 @@ function connect_peers_with_type(node_type: NodeType)
for ( i in nn ) for ( i in nn )
{ {
local n = nn[i]; local n = nn[i];
if ( ! hook connect_node_hook(n) )
next;
local status = Broker::peer(cat(n$node$ip), n$node$p, local status = Broker::peer(cat(n$node$ip), n$node$p,
Cluster::retry_interval); Cluster::retry_interval);
Cluster::log(fmt("initiate peering with %s:%s, retry=%s, status=%s", Cluster::log(fmt("initiate peering with %s:%s, retry=%s, status=%s",