mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2025-03-12 20:36:38 +01:00
GetRandomEstablishedSession
This commit is contained in:
parent
770452cf85
commit
aaf386a0ab
2 changed files with 28 additions and 0 deletions
24
SSU.cpp
24
SSU.cpp
|
@ -1058,6 +1058,30 @@ namespace ssu
|
|||
}
|
||||
m_Sessions.clear ();
|
||||
}
|
||||
|
||||
template<typename Filter>
|
||||
SSUSession * SSUServer::GetRandomSession (Filter filter)
|
||||
{
|
||||
std::vector<SSUSession *> filteredSessions;
|
||||
for (auto s :m_Sessions)
|
||||
if (filter (s.second)) filteredSessions.push_back (s.second);
|
||||
if (filteredSessions.size () > 0)
|
||||
{
|
||||
auto ind = i2p::context.GetRandomNumberGenerator ().GenerateWord32 (0, filteredSessions.size ()-1);
|
||||
return filteredSessions[ind];
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
SSUSession * SSUServer::GetRandomEstablishedSession ()
|
||||
{
|
||||
return GetRandomSession (
|
||||
[](SSUSession * session)->bool
|
||||
{
|
||||
return session->GetState () == eSessionStateEstablished;
|
||||
}
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
|
4
SSU.h
4
SSU.h
|
@ -147,6 +147,7 @@ namespace ssu
|
|||
SSUSession * GetSession (const i2p::data::RouterInfo * router, bool peerTest = false);
|
||||
SSUSession * FindSession (const i2p::data::RouterInfo * router);
|
||||
SSUSession * FindSession (const boost::asio::ip::udp::endpoint& e);
|
||||
SSUSession * GetRandomEstablishedSession ();
|
||||
void DeleteSession (SSUSession * session);
|
||||
void DeleteAllSessions ();
|
||||
|
||||
|
@ -162,6 +163,9 @@ namespace ssu
|
|||
void Receive ();
|
||||
void HandleReceivedFrom (const boost::system::error_code& ecode, std::size_t bytes_transferred);
|
||||
|
||||
template<typename Filter>
|
||||
SSUSession * GetRandomSession (Filter filter);
|
||||
|
||||
private:
|
||||
|
||||
bool m_IsRunning;
|
||||
|
|
Loading…
Add table
Reference in a new issue