ntopng/include/RedisStub.h
2026-08-17 21:45:35 +02:00

117 lines
4.6 KiB
C++

/*
*
* (C) 2013-26 - ntop.org
*
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*
*/
#ifndef _REDIS_H_
#define _REDIS_H_
#include <memory>
#include "ntop_includes.h"
class Redis {
private:
std::string redisVersion;
std::map<std::string, std::vector<std::string>> listStore;
std::map<std::string, std::set<std::string>> setStore;
std::map<std::string, std::string> store;
std::unique_ptr<StringFifoQueue> localToResolve;
std::unique_ptr<StringFifoQueue> remoteToResolve;
struct {
u_int32_t num_expire, num_get, num_ttl, num_del, num_hget, num_hset,
num_hdel, num_set, num_keys, num_hkeys, num_llen, num_other,
num_hgetall, num_trim, num_lpush_rpush, num_lpop_rpop, num_strlen,
num_saved_lookups, num_get_address, num_set_resolved_address;
u_int32_t num_reconnections;
} stats;
bool checkList(std::string key);
bool checkSet(std::string key);
public:
Redis(const char* redis_host = (char*)"127.0.0.1",
const char* redis_password = NULL, u_int16_t redis_port = 6379,
u_int8_t _redis_db_id = 0, bool giveup_on_failure = false,
const char* tls_ca_cert = NULL, const char* tls_cert = NULL,
const char* tls_key = NULL, bool tls_skip_verify = false);
char* getVersion();
u_int32_t getNumVersion();
bool hasRedisDump();
void setDefaults();
bool isOperational();
void setInitializationComplete();
int info(char* rsp, u_int rsp_len);
u_int dbsize();
int expire(char* key, u_int expire_sec);
char* getWithAlloc(char* key, bool cache_it = false);
int get(char* key, char* rsp, u_int rsp_len, bool cache_it = false);
char* getset(const char* key, const char* value, u_int expire_secs = 0);
int hashGet(const char* key, const char* member, char* const rsp,
u_int rsp_len);
int hashDel(const char* key, const char* field);
int hashSet(const char* key, const char* field, const char* value);
int delHash(char* key, char* member);
int set(const char* key, const char* value, u_int expire_secs = 0);
/* setnx = set if not existing */
int setnx(const char* key, const char* value, u_int expire_secs = 0);
int keys(const char* pattern, char*** keys_p);
int hashKeys(const char* pattern, char*** keys_p);
int hashGetAll(const char* key, char*** keys_p, char*** values_p);
int del(char* key);
int rename(char* key, char* new_key);
int pushHostToResolve(char* hostname, bool dont_check_for_existence,
bool localHost);
int popHostToResolve(char* hostname, u_int hostname_len);
int getAddress(char* numeric_ip, char* rsp, u_int rsp_len,
bool queue_if_not_found);
int setResolvedAddress(char* numeric_ip, char* symbolic_ip);
int sadd(const char* set_name, char* item);
int srem(const char* set_name, char* item);
int smembers(lua_State* vm, char* setName);
int smembers(const char* set_name, char*** members);
bool sismember(const char* set_name, const char* member);
int lpush(const char* queue_name, const char* msg, u_int queue_trim_size,
bool trace_errors = true);
int rpush(const char* queue_name, const char* msg, u_int queue_trim_size);
int lindex(const char* queue_name, int idx, char* buf, u_int buf_len);
int ltrim(const char* queue_name, int start_idx, int end_idx);
u_int hstrlen(const char* key, const char* value);
u_int len(const char* key);
u_int llen(const char* queue_name);
int lset(const char* queue_name, u_int32_t idx, const char* value);
int lrem(const char* queue_name, const char* value);
int lrange(const char* list_name, char*** elements, int start_offset,
int end_offset);
int lpop(const char* queue_name, char** buf);
int rpop(const char* queue_name, char** buf);
int incr(const char* key, int amount);
int flushDb();
void flushCache();
void lua(lua_State* vm);
char* dump(char* key);
int restore(char* key, char* buf);
char* findAllWithAlloc(char* key);
char* findBestWithAlloc(char* key);
};
#endif /* _REDIS_H_ */