Skip to content
Redis_Provider.cpp 1.86 KiB
Newer Older
Valerio Pastore's avatar
Valerio Pastore committed
#include <Redis_Provider.h>

Valerio Pastore's avatar
Valerio Pastore committed
using namespace inaf::oasbo::Providers;
Valerio Pastore's avatar
Valerio Pastore committed
RedisProvider::RedisProvider() {
Valerio Pastore's avatar
Valerio Pastore committed
	setIp("127.0.0.1");
	setPort(6379);
Valerio Pastore's avatar
Valerio Pastore committed
	setKey("DAQ_key");
}

Valerio Pastore's avatar
Valerio Pastore committed
RedisProvider::RedisProvider(std::string ip, int port, std::string key) {
Valerio Pastore's avatar
Valerio Pastore committed
	setIp(ip);
	setPort(port);
Valerio Pastore's avatar
Valerio Pastore committed
	setKey(key);
}

Valerio Pastore's avatar
Valerio Pastore committed
int RedisProvider::write(PacketLib::BasePacket &packet) {
Valerio Pastore's avatar
Valerio Pastore committed
	uint size = packet.getHeaderSize() + packet.getPayloadSize() + packet.getTailSize();
Valerio Pastore's avatar
Valerio Pastore committed
	redisReply *r = (redisReply *) redisCommand(context, "LPUSH %s %b", this->key.c_str(), (char*) packet.getBinaryPointer(), size);
	if (r == NULL) {
	    // Error executing Redis command
	    return -1;
	}
	// Check the type of the Redis reply
	if (r->type == REDIS_REPLY_ERROR) {
	    // Error executing Redis command
		std::cout << r->str << std::endl;
	    freeReplyObject(r);
	    return -1;
	}
    freeReplyObject(r);
Valerio Pastore's avatar
Valerio Pastore committed
	return size;
Valerio Pastore's avatar
Valerio Pastore committed
}

Valerio Pastore's avatar
Valerio Pastore committed
int RedisProvider::write(PacketLib::BasePacket &packet, std::string key) {
Valerio Pastore's avatar
Valerio Pastore committed
	uint size = packet.getHeaderSize() + packet.getPayloadSize() + packet.getTailSize();
Valerio Pastore's avatar
Valerio Pastore committed
	redisReply *r = (redisReply *) redisCommand(context, "LPUSH %s %b", key.c_str(), (char*) packet.getBinaryPointer(), size);
	if (r == NULL) {
	    // Error executing Redis command
	    return -1;
	}
	// Check the type of the Redis reply
	if (r->type == REDIS_REPLY_ERROR) {
	    // Error executing Redis command
		std::cout << r->str << std::endl;
	    freeReplyObject(r);
	    return -1;
	}
    freeReplyObject(r);
Valerio Pastore's avatar
Valerio Pastore committed
	return size;
Valerio Pastore's avatar
Valerio Pastore committed
}

Valerio Pastore's avatar
Valerio Pastore committed
int RedisProvider::open() {
Valerio Pastore's avatar
Valerio Pastore committed
	context = redisConnect(getIp().c_str(), getPort());
	if (context == NULL || context->err) {
	    if (context) {
astri's avatar
astri committed
	        printf("Redis Error: %s\n", context->errstr);
Valerio Pastore's avatar
Valerio Pastore committed
	    } else {
	        printf("Can't allocate Redis context\n");
Valerio Pastore's avatar
Valerio Pastore committed
	}
Valerio Pastore's avatar
Valerio Pastore committed
	return 1;
Valerio Pastore's avatar
Valerio Pastore committed
}

Valerio Pastore's avatar
Valerio Pastore committed
int RedisProvider::close() {
astri's avatar
astri committed
	if (!(context == NULL || context->err))
		redisFree(this->context);
Valerio Pastore's avatar
Valerio Pastore committed
	return 1;
Valerio Pastore's avatar
Valerio Pastore committed
}
Valerio Pastore's avatar
Valerio Pastore committed

RedisProvider::~RedisProvider(){
astri's avatar
astri committed
	close();
Valerio Pastore's avatar
Valerio Pastore committed
}