mirror of
https://github.com/zeek/zeek.git
synced 2025-10-15 13:08:20 +00:00
737b1a2013
This eliminates one place in which we currently need to mirror changes to the script-land Cluster::Node record. Instead of keeping an exact in-core equivalent, the Supervisor now treats the data structure as opaque, and stores the whole cluster table as a JSON string. We may replace the script-layer Supervisor::ClusterEndpoint in the future, using Cluster::Node directly. But that's a more invasive change that will affect how people invoke Supervisor::create() and similars. Relying on JSON for serialization has the side-effect of removing the Supervisor's earlier quirk of using 0/tcp, not 0/unknown, to indicate unused ports in the Supervisor::ClusterEndpoint record.
103 lines
2.6 KiB
C++
103 lines
2.6 KiB
C++
##! The BIFs that define the Zeek supervisor control interface.
|
|
|
|
%%{
|
|
#include "zeek/supervisor/Supervisor.h"
|
|
%%}
|
|
|
|
module Supervisor;
|
|
|
|
enum ClusterRole %{
|
|
NONE,
|
|
LOGGER,
|
|
MANAGER,
|
|
PROXY,
|
|
WORKER,
|
|
%}
|
|
|
|
type Supervisor::Status: record;
|
|
type Supervisor::NodeConfig: record;
|
|
type Supervisor::NodeStatus: record;
|
|
|
|
function Supervisor::__status%(node: string%): Supervisor::Status
|
|
%{
|
|
if ( ! zeek::supervisor_mgr )
|
|
{
|
|
zeek::emit_builtin_error("supervisor mode not enabled");
|
|
return zeek::make_intrusive<zeek::RecordVal>(zeek::BifType::Record::Supervisor::Status);
|
|
}
|
|
|
|
return zeek::supervisor_mgr->Status(node->CheckString());
|
|
%}
|
|
|
|
function Supervisor::__create%(node: Supervisor::NodeConfig%): string
|
|
%{
|
|
if ( ! zeek::supervisor_mgr )
|
|
{
|
|
zeek::emit_builtin_error("supervisor mode not enabled");
|
|
return zeek::make_intrusive<zeek::StringVal>("supervisor mode not enabled");
|
|
}
|
|
|
|
auto rval = zeek::supervisor_mgr->Create(node->AsRecordVal());
|
|
return zeek::make_intrusive<zeek::StringVal>(rval);
|
|
%}
|
|
|
|
function Supervisor::__destroy%(node: string%): bool
|
|
%{
|
|
if ( ! zeek::supervisor_mgr )
|
|
{
|
|
zeek::emit_builtin_error("supervisor mode not enabled");
|
|
return zeek::val_mgr->Bool(false);
|
|
}
|
|
|
|
auto rval = zeek::supervisor_mgr->Destroy(node->CheckString());
|
|
return zeek::val_mgr->Bool(rval);
|
|
%}
|
|
|
|
function Supervisor::__restart%(node: string%): bool
|
|
%{
|
|
if ( ! zeek::supervisor_mgr )
|
|
{
|
|
zeek::emit_builtin_error("supervisor mode not enabled");
|
|
return zeek::val_mgr->Bool(false);
|
|
}
|
|
|
|
auto rval = zeek::supervisor_mgr->Restart(node->CheckString());
|
|
return zeek::val_mgr->Bool(rval);
|
|
%}
|
|
|
|
function Supervisor::__is_supervised%(%): bool
|
|
%{
|
|
return zeek::val_mgr->Bool(zeek::Supervisor::ThisNode().has_value());
|
|
%}
|
|
|
|
function Supervisor::__node%(%): Supervisor::NodeConfig
|
|
%{
|
|
if ( ! zeek::Supervisor::ThisNode() )
|
|
{
|
|
zeek::emit_builtin_error("not a supervised process");
|
|
const auto& rt = zeek::BifType::Record::Supervisor::NodeConfig;
|
|
auto rval = zeek::make_intrusive<zeek::RecordVal>(rt);
|
|
rval->AssignField("name", "<invalid>");
|
|
return std::move(rval);
|
|
}
|
|
|
|
auto rval = zeek::Supervisor::ThisNode()->config.ToRecord();
|
|
return std::move(rval);
|
|
%}
|
|
|
|
function Supervisor::__is_supervisor%(%): bool
|
|
%{
|
|
return zeek::val_mgr->Bool(zeek::supervisor_mgr != nullptr);
|
|
%}
|
|
|
|
function Supervisor::__stem_pid%(%): int
|
|
%{
|
|
if ( zeek::supervisor_mgr )
|
|
return zeek::val_mgr->Int(zeek::supervisor_mgr->StemPID());
|
|
|
|
if ( zeek::Supervisor::ThisNode() )
|
|
return zeek::val_mgr->Int(zeek::Supervisor::ThisNode()->parent_pid);
|
|
|
|
zeek::emit_builtin_error("supervisor mode not enabled and not a supervised node");
|
|
return zeek::val_mgr->Int(-1);
|
|
%}
|