mirrors/ZeroTierOne
1
Fork 0
mirror of https://github.com/zerotier/ZeroTierOne.git synced 2026-01-10 11:11:39 -08:00
Code Issues Projects Releases Wiki Activity

Refactor Redis & Posgres notification listeners into listener subclass in new CentralDB class

Browse source 
This allows us to interchangeably use different listeners (pgsql, redis, pubsub) depending on configuration values passed into the constructor.
This commit is contained in:
Grant Limberg 2025-08-20 17:04:28 -07:00
parent ebe8fdb08e
commit 95224379aa
13 changed files with 2486 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

162
controller/PostgreSQL.cpp
View file

@ -2,10 +2,168 @@
#include "PostgreSQL.hpp"
#include "opentelemetry/trace/provider.h"
#include <nlohmann/json.hpp>
using namespace nlohmann;
namespace ZeroTier {
using namespace ZeroTier;
PostgresMemberListener::PostgresMemberListener(DB* db, std::shared_ptr<ConnectionPool<PostgresConnection> > pool, const std::string& channel, uint64_t timeout)
: NotificationListener()
, _db(db)
, _pool(pool)
, _notification_timeout(timeout)
, _listenerThread()
{
_conn = _pool->borrow();
_receiver = new _notificationReceiver<PostgresMemberListener>(this, *_conn->c, channel);
_run = true;
_listenerThread = std::thread(&PostgresMemberListener::listen, this);
}
PostgresMemberListener::~PostgresMemberListener()
{
_run = false;
if (_listenerThread.joinable()) {
_listenerThread.join();
}
delete _receiver;
if (_conn) {
_pool->unborrow(_conn);
_conn.reset();
}
}
void PostgresMemberListener::listen()
{
while (_run) {
_conn->c->await_notification(_notification_timeout, 0);
}
}
void PostgresMemberListener::onNotification(const std::string& payload)
{
auto provider = opentelemetry::trace::Provider::GetTracerProvider();
auto tracer = provider->GetTracer("PostgresMemberNotificationListener");
auto span = tracer->StartSpan("PostgresMemberNotificationListener::onNotification");
auto scope = tracer->WithActiveSpan(span);
span->SetAttribute("payload", payload);
fprintf(stderr, "Member Notification received: %s\n", payload.c_str());
Metrics::pgsql_mem_notification++;
nlohmann::json tmp(nlohmann::json::parse(payload));
nlohmann::json& ov = tmp["old_val"];
nlohmann::json& nv = tmp["new_val"];
nlohmann::json oldConfig, newConfig;
if (ov.is_object())
oldConfig = ov;
if (nv.is_object())
newConfig = nv;
if (oldConfig.is_object() && newConfig.is_object()) {
_db->save(newConfig, true);
fprintf(stderr, "payload sent\n");
}
else if (newConfig.is_object() && ! oldConfig.is_object()) {
// new member
Metrics::member_count++;
_db->save(newConfig, true);
fprintf(stderr, "new member payload sent\n");
}
else if (! newConfig.is_object() && oldConfig.is_object()) {
// member delete
uint64_t networkId = OSUtils::jsonIntHex(oldConfig["nwid"], 0ULL);
uint64_t memberId = OSUtils::jsonIntHex(oldConfig["id"], 0ULL);
if (memberId && networkId) {
_db->eraseMember(networkId, memberId);
fprintf(stderr, "member delete payload sent\n");
}
}
}
PostgresNetworkListener::PostgresNetworkListener(DB* db, std::shared_ptr<ConnectionPool<PostgresConnection> > pool, const std::string& channel, uint64_t timeout)
: NotificationListener()
, _db(db)
, _pool(pool)
, _notification_timeout(timeout)
, _listenerThread()
{
_conn = _pool->borrow();
_receiver = new _notificationReceiver<PostgresNetworkListener>(this, *_conn->c, channel);
_run = true;
_listenerThread = std::thread(&PostgresNetworkListener::listen, this);
}
PostgresNetworkListener::~PostgresNetworkListener()
{
_run = false;
if (_listenerThread.joinable()) {
_listenerThread.join();
}
delete _receiver;
if (_conn) {
_pool->unborrow(_conn);
_conn.reset();
}
}
void PostgresNetworkListener::listen()
{
while (_run) {
_conn->c->await_notification(_notification_timeout, 0);
}
}
void PostgresNetworkListener::onNotification(const std::string& payload)
{
auto provider = opentelemetry::trace::Provider::GetTracerProvider();
auto tracer = provider->GetTracer("db_network_notification");
auto span = tracer->StartSpan("db_network_notification::operator()");
auto scope = tracer->WithActiveSpan(span);
span->SetAttribute("payload", payload);
fprintf(stderr, "Network Notification received: %s\n", payload.c_str());
Metrics::pgsql_net_notification++;
nlohmann::json tmp(nlohmann::json::parse(payload));
nlohmann::json& ov = tmp["old_val"];
nlohmann::json& nv = tmp["new_val"];
nlohmann::json oldConfig, newConfig;
if (ov.is_object())
oldConfig = ov;
if (nv.is_object())
newConfig = nv;
if (oldConfig.is_object() && newConfig.is_object()) {
std::string nwid = oldConfig["id"];
span->SetAttribute("action", "network_change");
span->SetAttribute("network_id", nwid);
_db->save(newConfig, true);
fprintf(stderr, "payload sent\n");
}
else if (newConfig.is_object() && ! oldConfig.is_object()) {
std::string nwid = newConfig["id"];
span->SetAttribute("network_id", nwid);
span->SetAttribute("action", "new_network");
// new network
_db->save(newConfig, true);
fprintf(stderr, "new network payload sent\n");
}
else if (! newConfig.is_object() && oldConfig.is_object()) {
// network delete
span->SetAttribute("action", "delete_network");
std::string nwid = oldConfig["id"];
span->SetAttribute("network_id", nwid);
uint64_t networkId = Utils::hexStrToU64(nwid.c_str());
span->SetAttribute("network_id_int", networkId);
if (networkId) {
_db->eraseNetwork(networkId);
fprintf(stderr, "network delete payload sent\n");
}
}
}
} // namespace ZeroTier
#endif