acl/lib_acl_cpp/include/acl_cpp/redis/redis_stream.hpp

#pragma once
#include "../acl_cpp_define.hpp"
#include "../stdlib/string.hpp"
#include "redis_command.hpp"

namespace acl
{

class redis_client;
class redis_client_cluster;

struct redis_stream_field
{
	string name;
	string value;
};

struct redis_stream_message
{
	string id;
	std::vector<redis_stream_field> fields;
};

struct redis_stream_messages
{
	string key;
	std::vector<redis_stream_message> messages;

	bool empty(void) const
	{
		return messages.empty();
	}

	size_t size(void) const
	{
		return messages.size();
	}
};

struct redis_xinfo_consumer
{
	string name;
	size_t pending;
	size_t idle;

	redis_xinfo_consumer(void)
	{
		pending = 0;
		idle    = 0;
	}
};

struct redis_xinfo_group
{
	string name;
	string last_delivered_id;
	size_t consumers;
	size_t pending;

	redis_xinfo_group(void)
	{
		consumers = 0;
		pending   = 0;
	}
};

struct redis_stream_info
{
	size_t length;
	size_t radix_tree_keys;
	size_t radix_tree_nodes;
	size_t groups;
	string last_generated_id;
	redis_stream_message first_entry;
	redis_stream_message last_entry;

	redis_stream_info(void)
	{
		length           = 0;
		radix_tree_keys  = 0;
		radix_tree_nodes = 0;
		groups           = 0;
	}
};

struct redis_pending_consumer
{
	string name;
	size_t pending_number;

	redis_pending_consumer(void)
	{
		pending_number = 0;
	}
};

struct redis_pending_summary
{
	string smallest_id;
	string greatest_id;
	std::vector<redis_pending_consumer> consumers;

	bool empty(void) const
	{
		return consumers.empty();
	}

	size_t size(void) const
	{
		return consumers.size();
	}
};

struct redis_pending_message
{
	string id;
	string consumer;
	unsigned long long elapsed;
	size_t delivered;

	redis_pending_message(void)
	{
		elapsed   = 0;
		delivered = 0;
	}
};

struct redis_pending_detail
{
	std::map<string, redis_pending_message> messages;

	bool empty(void) const
	{
		return messages.empty();
	}

	size_t size(void) const
	{
		return messages.size();
	}
};

class ACL_CPP_API redis_stream : virtual public redis_command
{
public:
	redis_stream(void);
	redis_stream(redis_client* conn);
	redis_stream(redis_client_cluster* cluster, size_t max_conns = 0);

	virtual ~redis_stream(void);

	/////////////////////////////////////////////////////////////////////

	bool xadd(const char* key, const std::map<string, string>& fields,
		string& result, const char* id = "*");
	bool xadd(const char* key, const std::vector<string>& names,
		const std::vector<string>& values,
		string& result, const char* id = "*");
	bool xadd(const char* key, const std::vector<const char*>& names,
		const std::vector<const char*>& values,
		string& result, const char* id = "*");
	bool xadd(const char* key, const char* names[], const size_t names_len[],
		const char* values[], const size_t values_len[], size_t argc,
		string& result, const char* id = "*");

	int  xlen(const char* key);
	int  xdel(const char* key, const char* id);
	int  xdel(const char* key, const std::vector<string>& ids);
	int  xdel(const char* key, const std::vector<const char*>& ids);
	int  xtrim(const char* key, size_t maxlen, bool tilde = false);

	/////////////////////////////////////////////////////////////////////

	bool xread(redis_stream_messages& messages,
		const std::map<string, string>& streams,
		size_t count = 0, size_t block = 0);
	bool xreadgroup(redis_stream_messages& messsages, const char* group,
		const char* consumer, const std::map<string, string>& streams,
		size_t count = 1000, size_t block = 0, bool noack = false);
	bool xreadgroup_with_noack(redis_stream_messages& messsages,
		const char* group, const char* consumer,
		const std::map<string, string>& streams,
		size_t count = 1000, size_t block = 0);
	bool xrange(redis_stream_messages& messages, const char* key,
		const char* start = "-", const char* end = "+", size_t count = 0);
	bool xrevrange(redis_stream_messages& messages, const char* key,
		const char* start = "+", const char* end = "-", size_t count = 0);

	/////////////////////////////////////////////////////////////////////

	bool xclaim(std::vector<redis_stream_message>& messages,
		const char* key, const char* group, const char* consumer,
		long min_idle_time, const std::vector<string>& ids,
		size_t idle = 0, long long time_ms = -1,
		int retry_count = -1, bool force = false);

	bool xclaim_with_justid(std::vector<string>& messages_ids,
		const char* key, const char* group, const char* consumer,
		long min_idle_time, const std::vector<string>& ids,
		size_t idle = 0, long long time_ms = -1,
		int retry_count = -1, bool force = false);

	/////////////////////////////////////////////////////////////////////

	int  xack(const char* key, const char* group, const char* id);
	int  xack(const char* key, const char* group,
		const std::vector<string>& ids);
	int  xack(const char* key, const char* group,
		const std::vector<const char*>& ids);
	int  xack(const char* key, const char* group,
		const std::list<string>& ids, size_t size);
	int  xack(const char* key, const char* group,
		const std::list<const char*>& ids, size_t size);

	/////////////////////////////////////////////////////////////////////

	bool xpending_summary(const char* key, const char* group,
		redis_pending_summary& result);
	bool xpending_detail(redis_pending_detail& result,
		const char* key, const char* group,
		const char* start_id = "-", const char* end_id = "+",
		size_t count = 1, const char* consumer = NULL);

	/////////////////////////////////////////////////////////////////////

	bool xgroup_help(std::vector<string>& result);
	bool xgroup_create(const char* key, const char* group,
		const char* id = "$", bool mkstream = true);
	int  xgroup_destroy(const char* key, const char* group);
	bool xgroup_setid(const char* key, const char* group,
		const char* id = "$");
	int  xgroup_delconsumer(const char* key, const char* group,
		const char* consumer);

	/////////////////////////////////////////////////////////////////////

	bool xinfo_help(std::vector<string>& result);
	bool xinfo_consumers(const char* key, const char* group,
		std::map<string, redis_xinfo_consumer>& result);
	bool xinfo_groups(const char* key,
		std::map<string, redis_xinfo_group>& result);
	bool xinfo_stream(const char* key, redis_stream_info& result);

	/////////////////////////////////////////////////////////////////////

private:
	void build(const char* cmd, const char* key, const char* id,
		const std::map<string, string>& fields);
	void build(const char* cmd, const char* key, const char* id,
		const std::vector<string>& names,
		const std::vector<string>& values);
	void build(const char* cmd, const char* key, const char* id,
		const std::vector<const char*>& names,
		const std::vector<const char*>& values);
	void build(const char* cmd, const char* key, const char* id,
		const char* names[], const size_t names_len[],
		const char* values[], const size_t values_len[], size_t argc);
	void build(const std::map<string, string>& streams, size_t i,
		size_t count, size_t block, bool noack = false);
	void xread_build(const std::map<string, string>& streams,
		size_t count, size_t block);
	void xreadgroup_build(const char* group, const char* consumer,
		const std::map<string, string>& streams,
		size_t count, size_t block, bool noack);
	bool get_results(redis_stream_messages& messages);
	bool get_messages(const redis_result& rr, redis_stream_messages& messages);
	bool get_one_message(const redis_result& rr, redis_stream_message& message);
	bool range(redis_stream_messages& messages, const char* cmd,
	     	const char* key, const char* start, const char* end, size_t count);

	bool get_one_consumer(const redis_result& rr, redis_xinfo_consumer& consumer);
	bool get_one_group(const redis_result& rr, redis_xinfo_group& group);
	bool get_pending_consumer(const redis_result& rr,
		redis_pending_consumer& consumer);
	bool get_pending_message(const redis_result& rr,
		redis_pending_message& message);

	void xclaim_build(const char* key, const char* group,
		const char* consumer, long min_idle_time,
		const std::vector<string>& ids, size_t idle, long long time_ms,
		int retry_count, bool force, bool justid);

};

}