mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-17 15:25:35 +00:00
49 lines
1.1 KiB
C++
49 lines
1.1 KiB
C++
#include "lns_tracker.hpp"
|
|
|
|
namespace llarp::service
|
|
{
|
|
std::function<void(std::optional<LNSLookupTracker::Addr_t>)>
|
|
LNSLookupTracker::MakeResultHandler(
|
|
std::string name,
|
|
std::size_t numPeers,
|
|
std::function<void(std::optional<Addr_t>)> resultHandler)
|
|
{
|
|
m_PendingLookups.emplace(name, LookupInfo{numPeers, resultHandler});
|
|
return [name, this](std::optional<Addr_t> found) {
|
|
auto itr = m_PendingLookups.find(name);
|
|
if (itr == m_PendingLookups.end())
|
|
return;
|
|
itr->second.HandleOneResult(found);
|
|
if (itr->second.IsDone())
|
|
m_PendingLookups.erase(itr);
|
|
};
|
|
}
|
|
|
|
bool
|
|
LNSLookupTracker::LookupInfo::IsDone() const
|
|
{
|
|
return m_ResultsGotten == m_ResultsNeeded;
|
|
}
|
|
|
|
void
|
|
LNSLookupTracker::LookupInfo::HandleOneResult(std::optional<Addr_t> result)
|
|
{
|
|
if (result)
|
|
{
|
|
m_CurrentValues.insert(*result);
|
|
}
|
|
m_ResultsGotten++;
|
|
if (IsDone())
|
|
{
|
|
if (m_CurrentValues.size() == 1)
|
|
{
|
|
m_HandleResult(*m_CurrentValues.begin());
|
|
}
|
|
else
|
|
{
|
|
m_HandleResult(std::nullopt);
|
|
}
|
|
}
|
|
}
|
|
} // namespace llarp::service
|