1
0
Fork 0
mirror of https://git.rwth-aachen.de/acs/public/villas/node/ synced 2025-03-09 00:00:00 +01:00
VILLASnode/include/villas/nodes/redis.hpp
Philipp Jungkamp 33cd6165df Fix fmt 10.0.0 related formatting errors.
Signed-off-by: Philipp Jungkamp <Philipp.Jungkamp@opal-rt.com>
2023-09-26 17:59:39 +02:00

133 lines
2.7 KiB
C++

/* Redis node-type.
*
* 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 <atomic>
#include <thread>
#include <unordered_map>
#include <sw/redis++/redis++.h>
#include <villas/format.hpp>
#include <villas/node.hpp>
#include <villas/node/config.hpp>
#include <villas/pool.hpp>
#include <villas/queue_signalled.h>
#include <villas/task.hpp>
#include <villas/timing.hpp>
namespace villas {
namespace node {
enum class RedisMode { KEY, HASH, CHANNEL };
inline std::ostream &operator<<(std::ostream &os, const enum villas::node::RedisMode &m) {
switch (m) {
case villas::node::RedisMode::KEY:
os << "key";
break;
case villas::node::RedisMode::HASH:
os << "hash";
break;
case villas::node::RedisMode::CHANNEL:
os << "channel";
break;
}
return os;
}
class RedisConnection {
public:
sw::redis::Redis context;
protected:
enum State { INITIALIZED, RUNNING, STOPPING };
std::thread thread;
std::atomic<enum State> state;
void onMessage(const std::string &channel, const std::string &msg);
void loop();
std::unordered_multimap<std::string, NodeCompat *> subscriberMap;
sw::redis::Subscriber subscriber;
villas::Logger logger;
public:
RedisConnection(const sw::redis::ConnectionOptions &opts);
static RedisConnection *get(const sw::redis::ConnectionOptions &opts);
void start();
void stop();
void subscribe(NodeCompat *n, const std::string &channel);
void unsubscribe(NodeCompat *n, const std::string &channel);
};
struct redis {
sw::redis::ConnectionOptions options;
RedisConnection *conn;
enum RedisMode mode;
std::string key;
bool notify; // Use Redis Keyspace notifications to listen for updates.
struct Task task; // Timer for periodic events.
double
rate; // Rate for polling key updates if keyspace notifications are disabled.
Format *formatter;
struct Pool pool;
struct CQueueSignalled queue;
};
int redis_init(NodeCompat *n);
int redis_destroy(NodeCompat *n);
int redis_parse(NodeCompat *n, json_t *json);
char *redis_print(NodeCompat *n);
int redis_check(NodeCompat *n);
int redis_prepare(NodeCompat *n);
int redis_start(NodeCompat *n);
int redis_stop(NodeCompat *n);
int redis_pause(NodeCompat *n);
int redis_resume(NodeCompat *n);
int redis_write(NodeCompat *n, struct Sample *const smps[], unsigned cnt);
int redis_read(NodeCompat *n, struct Sample *const smps[], unsigned cnt);
int redis_reverse(NodeCompat *n);
int redis_poll_fds(NodeCompat *n, int fds[]);
int redis_netem_fds(NodeCompat *n, int fds[]);
} // namespace node
} // namespace villas