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

broker/Manager: Add MakeHub() and ReleaseHub()

Browse source 
These are used by WebSocket clients to create broker::hub instances
This commit is contained in:
Arne Welzel 2025-04-09 17:50:25 +02:00
parent c9d7418a23
commit 591e3400d4
2 changed files with 23 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/broker/Manager.cc
View file

@ -282,6 +282,7 @@ Manager::Manager(bool arg_use_real_time) : Backend("Broker", nullptr, nullptr, n
bound_port = 0;
use_real_time = arg_use_real_time;
peer_count = 0;
hub_count = 0;
log_batch_size = 0;
log_topic_func = nullptr;
log_id_type = nullptr;
@ -509,7 +510,7 @@ bool Manager::Active() {
if ( bound_port > 0 )
return true;
return peer_count > 0;
return peer_count > 0 || hub_count > 0;
}
void Manager::AdvanceTime(double seconds_since_unix_epoch) {
@ -652,7 +653,7 @@ bool Manager::PublishEvent(string topic, std::string name, broker::vector args,
if ( bstate->endpoint.is_shutdown() )
return true;
if ( peer_count == 0 )
if ( peer_count == 0 && hub_count == 0 )
return true;
broker::zeek::Event ev(name, args, broker::to_timestamp(ts));
@ -666,7 +667,7 @@ bool Manager::PublishEvent(string topic, RecordVal* args) {
if ( bstate->endpoint.is_shutdown() )
return true;
if ( peer_count == 0 )
if ( peer_count == 0 && hub_count == 0 )
return true;
if ( ! args->HasField(0) )
@ -1987,4 +1988,11 @@ void Manager::PrepareForwarding(const std::string& name) {
DBG_LOG(DBG_BROKER, "Resolved table forward for data store %s", name.c_str());
}
broker::hub Manager::MakeHub(broker::filter_type ft) {
++hub_count;
return bstate->endpoint.make_hub(std::move(ft));
}
void Manager::DestroyHub(broker::hub&& hub) { --hub_count; }
} // namespace zeek::Broker