zeek/scripts/base/frameworks/supervisor/main.zeek

##! Implements Zeek process supervision configuration options and default
##! behavior.
# TODO: add proper docs

@load ./api
@load base/frameworks/broker

module Supervisor;

export {
	const topic_prefix = "zeek/supervisor" &redef;
}

event zeek_init() &priority=10
	{
	Broker::subscribe(Supervisor::topic_prefix);
	}

event Supervisor::stop_request()
	{
	terminate();
	}

event Supervisor::status_request(id: count, nodes: string)
	{
	local res = Supervisor::status(nodes);
	local topic = Supervisor::topic_prefix + "/status_response";
	Broker::publish(topic, Supervisor::status_response, id, res);
	}

event Supervisor::create_request(id: count, config: NodeConfig)
	{
	local res = Supervisor::create(config);
	local topic = Supervisor::topic_prefix + "/create_response";
	Broker::publish(topic, Supervisor::create_response, id, res);
	}

event Supervisor::destroy_request(id: count, nodes: string)
	{
	local res = Supervisor::destroy(nodes);
	local topic = Supervisor::topic_prefix + "/destroy_response";
	Broker::publish(topic, Supervisor::destroy_response, id, res);
	}

event Supervisor::restart_request(id: count, nodes: string)
	{
	local res = Supervisor::restart(nodes);
	local topic = Supervisor::topic_prefix + "/restart_response";
	Broker::publish(topic, Supervisor::restart_response, id, res);
	}

function Supervisor::status(nodes: string): Status
	{
	return Supervisor::__status(nodes);
	}

function create(config: NodeConfig): string
	{
	return Supervisor::__create(config);
	}

function destroy(nodes: string): bool
	{
	return Supervisor::__destroy(nodes);
	}

function restart(nodes: string): bool
	{
	return Supervisor::__restart(nodes);
	}