mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-09-09 08:12:52 +02:00
99 lines
No EOL
2.4 KiB
C++
99 lines
No EOL
2.4 KiB
C++
#include "PostgresStatusWriter.hpp"
|
|
|
|
#include "../node/Metrics.hpp"
|
|
|
|
#include <nlohmann/json.hpp>
|
|
#include <pqxx/pqxx>
|
|
|
|
namespace ZeroTier {
|
|
|
|
PostgresStatusWriter::PostgresStatusWriter(std::shared_ptr<ConnectionPool<PostgresConnection> > pool) : _pool(pool)
|
|
{
|
|
}
|
|
|
|
PostgresStatusWriter::~PostgresStatusWriter()
|
|
{
|
|
writePending();
|
|
}
|
|
|
|
void PostgresStatusWriter::updateNodeStatus(
|
|
const std::string& network_id,
|
|
const std::string& node_id,
|
|
const std::string& os,
|
|
const std::string& arch,
|
|
const std::string& version,
|
|
const InetAddress& address,
|
|
int64_t last_seen)
|
|
{
|
|
std::lock_guard<std::mutex> l(_lock);
|
|
_pending.push_back({ network_id, node_id, os, arch, version, address, last_seen });
|
|
}
|
|
|
|
size_t PostgresStatusWriter::queueLength() const
|
|
{
|
|
std::lock_guard<std::mutex> l(_lock);
|
|
return _pending.size();
|
|
}
|
|
|
|
void PostgresStatusWriter::writePending()
|
|
{
|
|
std::vector<PendingStatusEntry> toWrite;
|
|
{
|
|
std::lock_guard<std::mutex> l(_lock);
|
|
toWrite.swap(_pending);
|
|
}
|
|
if (toWrite.empty()) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
auto conn = _pool->borrow();
|
|
pqxx::work w(*conn->c);
|
|
|
|
pqxx::pipeline pipe(w);
|
|
for (const auto& entry : toWrite) {
|
|
char iptmp[64] = { 0 };
|
|
nlohmann::json record = {
|
|
{ entry.address.toIpString(iptmp), entry.last_seen },
|
|
};
|
|
|
|
try {
|
|
// check if the member exists first.
|
|
//
|
|
// exec_params1 will throw pqxx::unexpected_rows if not exactly one row is returned. If that's the
|
|
// case, skip this record and move on.
|
|
w.exec_params1(
|
|
"SELECT device_id, network_id FROM network_memberships_ctl WHERE network_id = $1 AND device_id = "
|
|
"$2",
|
|
entry.network_id, entry.node_id);
|
|
}
|
|
catch (pqxx::unexpected_rows& e) {
|
|
continue;
|
|
}
|
|
|
|
std::string insert_statement =
|
|
"INSERT INTO network_memberships_ctl (device_id, network_id, last_seen, os, arch) "
|
|
"VALUES ('"
|
|
+ w.esc(entry.node_id) + "', '" + w.esc(entry.network_id) + "', '" + w.esc(record.dump())
|
|
+ "'::JSONB, "
|
|
"'"
|
|
+ w.esc(entry.os) + "', '" + w.esc(entry.arch)
|
|
+ "') "
|
|
"ON CONFLICT (device_id, network_id) DO UPDATE SET os = EXCLUDED.os, arch = EXCLUDED.arch, "
|
|
"last_seen = network_memberships_ctl.last_seen || EXCLUDED.last_seen";
|
|
|
|
pipe.insert(insert_statement);
|
|
Metrics::pgsql_node_checkin++;
|
|
}
|
|
|
|
pipe.complete();
|
|
w.commit();
|
|
_pool->unborrow(conn);
|
|
}
|
|
catch (const std::exception& e) {
|
|
// Log the error
|
|
fprintf(stderr, "Error writing to Postgres: %s\n", e.what());
|
|
}
|
|
}
|
|
|
|
} // namespace ZeroTier
|