ZeroTierOne/nonfree/controller/DBMirrorSet.hpp

/* (c) ZeroTier, Inc.
 * See LICENSE.txt in nonfree/
 */

#ifndef ZT_DBMIRRORSET_HPP
#define ZT_DBMIRRORSET_HPP

#include "DB.hpp"

#include <memory>
#include <set>
#include <shared_mutex>
#include <thread>
#include <vector>

namespace ZeroTier {

class DBMirrorSet : public DB::ChangeListener {
  public:
	DBMirrorSet(DB::ChangeListener* listener);
	virtual ~DBMirrorSet();

	bool hasNetwork(const uint64_t networkId) const;

	bool get(const uint64_t networkId, nlohmann::json& network);
	bool get(const uint64_t networkId, nlohmann::json& network, const uint64_t memberId, nlohmann::json& member);
	bool get(const uint64_t networkId, nlohmann::json& network, const uint64_t memberId, nlohmann::json& member, DB::NetworkSummaryInfo& info);
	bool get(const uint64_t networkId, nlohmann::json& network, std::vector<nlohmann::json>& members);

	void networks(std::set<uint64_t>& networks);

	bool waitForReady();
	bool isReady();
	bool save(nlohmann::json& record, bool notifyListeners);
	void eraseNetwork(const uint64_t networkId);
	void eraseMember(const uint64_t networkId, const uint64_t memberId);
	virtual void nodeIsOnline(const uint64_t networkId, const uint64_t memberId, const InetAddress& physicalAddress);
	virtual void nodeIsOnline(const uint64_t networkId, const uint64_t memberId, const InetAddress& physicalAddress, const char* osArch);

	// These are called by various DB instances when changes occur.
	virtual void onNetworkUpdate(const void* db, uint64_t networkId, const nlohmann::json& network);
	virtual void onNetworkMemberUpdate(const void* db, uint64_t networkId, uint64_t memberId, const nlohmann::json& member);
	virtual void onNetworkMemberDeauthorize(const void* db, uint64_t networkId, uint64_t memberId);

	AuthInfo getSSOAuthInfo(const nlohmann::json& member, const std::string& redirectURL);

	inline void addDB(const std::shared_ptr<DB>& db)
	{
		db->addListener(this);
		std::unique_lock<std::shared_mutex> l(_dbs_l);
		_dbs.push_back(db);
	}

  private:
	DB::ChangeListener* const _listener;
	std::atomic_bool _running;
	std::thread _syncCheckerThread;
	std::vector<std::shared_ptr<DB> > _dbs;
	mutable std::shared_mutex _dbs_l;
};

}	// namespace ZeroTier

#endif