mirror of
https://git.rwth-aachen.de/acs/public/villas/node/
synced 2025-03-09 00:00:00 +01:00
171 lines
4.5 KiB
C++
171 lines
4.5 KiB
C++
/* Redis node-type helpers.
|
|
*
|
|
* Author: Steffen Vogel <post@steffenvogel.de>
|
|
* SPDX-FileCopyrightText: 2014-2023 Institute for Automation of Complex Power Systems, RWTH Aachen University
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <chrono>
|
|
#include <fmt/ostream.h>
|
|
#include <functional>
|
|
#include <sw/redis++/connection.h>
|
|
#include <sw/redis++/redis++.h>
|
|
#include <villas/node/config.hpp>
|
|
|
|
namespace std {
|
|
|
|
template <typename _rep, typename ratio>
|
|
struct hash<std::chrono::duration<_rep, ratio>> {
|
|
std::size_t operator()(std::chrono::duration<_rep, ratio> const &s) const {
|
|
return std::hash<_rep>{}(s.count());
|
|
}
|
|
};
|
|
|
|
template <> struct hash<sw::redis::tls::TlsOptions> {
|
|
std::size_t operator()(const sw::redis::tls::TlsOptions &t) const {
|
|
#ifdef REDISPP_WITH_TLS
|
|
return hash<bool>()(t.enabled) ^ hash<std::string>()(t.cacert) ^
|
|
hash<std::string>()(t.cacertdir) ^ hash<std::string>()(t.cert) ^
|
|
hash<std::string>()(t.key) ^ hash<std::string>()(t.sni);
|
|
#else
|
|
return 0;
|
|
#endif // REDISPP_WITH_TLS
|
|
}
|
|
};
|
|
|
|
template <> struct hash<sw::redis::ConnectionOptions> {
|
|
std::size_t operator()(const sw::redis::ConnectionOptions &o) const {
|
|
return hash<int>()(static_cast<int>(o.type)) ^ hash<std::string>()(o.host) ^
|
|
hash<int>()(o.port) ^ hash<std::string>()(o.path) ^
|
|
hash<std::string>()(o.user) ^ hash<std::string>()(o.password) ^
|
|
hash<int>()(o.db) ^ hash<bool>()(o.keep_alive) ^
|
|
hash<std::chrono::milliseconds>()(o.connect_timeout) ^
|
|
hash<std::chrono::milliseconds>()(o.socket_timeout) ^
|
|
hash<sw::redis::tls::TlsOptions>()(o.tls);
|
|
}
|
|
};
|
|
|
|
} // namespace std
|
|
|
|
namespace sw {
|
|
namespace redis {
|
|
|
|
inline bool operator==(const tls::TlsOptions &o1, const tls::TlsOptions &o2) {
|
|
#ifdef REDISPP_WITH_TLS
|
|
return o1.enabled == o2.enabled && o1.cacert == o2.cacert &&
|
|
o1.cacertdir == o2.cacertdir && o1.cert == o2.cert &&
|
|
o1.key == o2.key && o1.sni == o2.sni;
|
|
#else
|
|
return true;
|
|
#endif // REDISPP_WITH_TLS
|
|
}
|
|
|
|
inline bool operator==(const ConnectionOptions &o1,
|
|
const ConnectionOptions &o2) {
|
|
return o1.type == o2.type && o1.host == o2.host && o1.port == o2.port &&
|
|
o1.path == o2.path && o1.user == o2.user &&
|
|
o1.password == o2.password && o1.db == o2.db &&
|
|
o1.keep_alive == o2.keep_alive &&
|
|
o1.connect_timeout == o2.connect_timeout &&
|
|
o1.socket_timeout == o2.socket_timeout && o1.tls == o2.tls;
|
|
}
|
|
|
|
#ifdef REDISPP_WITH_TLS
|
|
namespace tls {
|
|
std::ostream &operator<<(std::ostream &os, const TlsOptions &t) {
|
|
os << "tls.enabled=" << (t.enabled ? "yes" : "no");
|
|
|
|
if (t.enabled) {
|
|
if (!t.cacert.empty())
|
|
os << ", tls.cacert=" << t.cacert;
|
|
|
|
if (!t.cacertdir.empty())
|
|
os << ", tls.cacertdir=" << t.cacertdir;
|
|
|
|
if (!t.cert.empty())
|
|
os << ", tls.cert=" << t.cert;
|
|
|
|
if (!t.key.empty())
|
|
os << ", tls.key=" << t.key;
|
|
|
|
if (!t.sni.empty())
|
|
os << ", tls.sni=" << t.sni;
|
|
}
|
|
|
|
return os;
|
|
}
|
|
} // namespace tls
|
|
#endif // REDISPP_WITH_TLS
|
|
|
|
inline std::ostream &operator<<(std::ostream &os, const ConnectionType &t) {
|
|
switch (t) {
|
|
case ConnectionType::TCP:
|
|
os << "tcp";
|
|
break;
|
|
|
|
case ConnectionType::UNIX:
|
|
os << "unix";
|
|
break;
|
|
}
|
|
|
|
return os;
|
|
}
|
|
|
|
inline std::ostream &operator<<(std::ostream &os, const ConnectionOptions &o) {
|
|
os << "type=" << o.type;
|
|
|
|
switch (o.type) {
|
|
case ConnectionType::TCP:
|
|
os << ", host=" << o.host << ", port=" << o.port;
|
|
break;
|
|
|
|
case ConnectionType::UNIX:
|
|
os << ", path=" << o.path;
|
|
break;
|
|
}
|
|
|
|
os << ", db=" << o.db << ", user=" << o.user
|
|
<< ", keepalive=" << (o.keep_alive ? "yes" : "no");
|
|
|
|
#ifdef REDISPP_WITH_TLS
|
|
os << ", " << o.tls;
|
|
#endif
|
|
|
|
return os;
|
|
}
|
|
|
|
} // namespace redis
|
|
} // namespace sw
|
|
|
|
namespace villas {
|
|
namespace node {
|
|
#ifdef REDISPP_WITH_URI
|
|
inline sw::redis::ConnectionOptions
|
|
make_redis_connection_options(char const *uri) {
|
|
auto u = sw::redis::Uri{uri};
|
|
return u.connection_options();
|
|
}
|
|
#else
|
|
inline sw::redis::ConnectionOptions
|
|
make_redis_connection_options(char const *uri) {
|
|
return sw::redis::ConnectionOptions{uri};
|
|
}
|
|
#endif
|
|
} // namespace node
|
|
} // namespace villas
|
|
|
|
#ifndef FMT_LEGACY_OSTREAM_FORMATTER
|
|
template <>
|
|
class fmt::formatter<sw::redis::ConnectionType>
|
|
: public fmt::ostream_formatter {};
|
|
template <>
|
|
class fmt::formatter<sw::redis::ConnectionOptions>
|
|
: public fmt::ostream_formatter {};
|
|
#ifdef REDISPP_WITH_TLS
|
|
template <>
|
|
class fmt::formatter<sw::redis::tls::TlsOptions>
|
|
: public fmt::ostream_formatter {};
|
|
#endif
|
|
#endif
|