ZeroTierOne/nonfree/controller/RedisListener.hpp

#ifdef ZT_CONTROLLER_USE_LIBPQ

#ifndef ZT_CONTROLLER_REDIS_LISTENER_HPP
#define ZT_CONTROLLER_REDIS_LISTENER_HPP

#include "DB.hpp"
#include "NotificationListener.hpp"
#include "Redis.hpp"

#include <memory>
#include <string>
#include <sw/redis++/redis++.h>
#include <thread>

namespace ZeroTier {

class RedisListener : public NotificationListener {
  public:
	RedisListener(std::string controller_id, std::shared_ptr<sw::redis::Redis> redis);
	RedisListener(std::string controller_id, std::shared_ptr<sw::redis::RedisCluster> cluster);

	virtual ~RedisListener();

	virtual void listen() = 0;
	virtual bool onNotification(const std::string& payload) override
	{
		return true;
	}

	void start()
	{
		_run = true;
		_listenThread = std::thread(&RedisListener::listen, this);
	}

  protected:
	std::string _controller_id;
	std::shared_ptr<sw::redis::Redis> _redis;
	std::shared_ptr<sw::redis::RedisCluster> _cluster;
	bool _is_cluster = false;
	bool _run = false;

  private:
	std::thread _listenThread;
};

class RedisNetworkListener : public RedisListener {
  public:
	RedisNetworkListener(std::string controller_id, std::shared_ptr<sw::redis::Redis> redis, DB* db);
	RedisNetworkListener(std::string controller_id, std::shared_ptr<sw::redis::RedisCluster> cluster, DB* db);
	virtual ~RedisNetworkListener();

	virtual void listen() override;
	virtual bool onNotification(const std::string& payload) override;

  private:
	DB* _db;
};

class RedisMemberListener : public RedisListener {
  public:
	RedisMemberListener(std::string controller_id, std::shared_ptr<sw::redis::Redis> redis, DB* db);
	RedisMemberListener(std::string controller_id, std::shared_ptr<sw::redis::RedisCluster> cluster, DB* db);
	virtual ~RedisMemberListener();

	virtual void listen() override;
	virtual bool onNotification(const std::string& payload) override;

  private:
	DB* _db;
};

}	// namespace ZeroTier

#endif	 // ZT_CONTROLLER_REDIS_LISTENER_HPP

#endif	 // ZT_CONTROLLER_USE_LIBPQ