TarsCpp/unit-test/server/framework/RegisterQueryManager.cpp

//
// Created by jarod on 2022/9/13.
//

#include "RegisterQueryManager.h"
#include "servant/QueryPushF.h"

void RegisterQueryManager::registerChange(const vector<string> &ids, const string &name, CurrentPtr current)
{
	std::lock_guard<std::mutex> lock(_mutex);

	for(auto &id : ids)
	{
		_uidToChangeIds[current->getUId()].insert(id);

		_changes[id][current->getUId()] = current;
	}
}

void RegisterQueryManager::registerQuery(const string &id, const string &name, CurrentPtr current)
{
	std::lock_guard<std::mutex> lock(_mutex);

	_uidToQueryIds[current->getUId()].insert(id);
	_queries[id][current->getUId()] = current;
}

void RegisterQueryManager::closeQuery(CurrentPtr current)
{
	std::lock_guard<std::mutex> lock(_mutex);

	{
		auto it = _uidToChangeIds.find(current->getUId());

		if (it != _uidToChangeIds.end())
		{
			for (auto e: it->second)
			{
				auto idIt = _changes.find(e);

				if (idIt != _changes.end())
				{
					idIt->second.erase(current->getUId());

					if (idIt->second.empty())
					{
						_changes.erase(idIt);
					}
				}
			}
		}
	}

	{

		auto it = _uidToQueryIds.find(current->getUId());

		if (it != _uidToQueryIds.end())
		{
			for (auto e: it->second)
			{
				auto idIt = _queries.find(e);

				if (idIt != _queries.end())
				{
					idIt->second.erase(current->getUId());

					if (idIt->second.empty())
					{
						_queries.erase(idIt);
					}
				}
			}
		}
	}
}

void RegisterQueryManager::pushAll()
{
	std::lock_guard<std::mutex> lock(_mutex);

	LOG_CONSOLE_DEBUG << _queries.size() << endl;

	for(auto e : _queries)
	{
		LOG_CONSOLE_DEBUG << e.first << ", " << e.second.size() << endl;
		for(auto it : e.second)
		{
			QueryPushF::async_response_push_onQuery(it.second, e.first);
		}
	}

}

unordered_map<int, CurrentPtr> RegisterQueryManager::getChanges(const string &id)
{
	std::lock_guard<std::mutex> lock(_mutex);

	auto it = _changes.find(id);

	if(it != _changes.end())
	{
		return it->second;
	}

	return unordered_map<int, CurrentPtr>();
}

unordered_map<int, CurrentPtr> RegisterQueryManager::getQueries(const string &id)
{
	std::lock_guard<std::mutex> lock(_mutex);

	auto it = _queries.find(id);

	if(it != _queries.end())
	{
		return it->second;
	}

	return unordered_map<int, CurrentPtr>();
}