microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-12 17:26:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

using find instead of searching

Browse Source
This commit is contained in:
Saeed Maleki
2023-04-25 21:27:23 +00:00
parent 4a165860b5
commit b73b0132ba
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/bootstrap/bootstrap.cc
View File

@@ -4,6 +4,7 @@
#include "mscclpp.hpp"
#include "utils.h"
#include <algorithm>
#include <cstring>
#include <mutex>
#include <queue>
@@ -399,14 +400,14 @@ void Bootstrap::Impl::send(void* data, int size, int peer, int tag)
void Bootstrap::Impl::recv(void* data, int size, int peer, int tag)
{
// search over all unexpected messages
for (auto it = unexpectedMessages_.begin(); it != unexpectedMessages_.end(); ++it) {
if (it->peer == peer && it->tag == tag) {
// found a match
netRecv(it->sock.get(), data, size);
MSCCLPPTHROW(mscclppSocketClose(it->sock.get()));
unexpectedMessages_.erase(it);
return;
}
auto lambda = [peer, tag](const UnexpectedMsg& msg) { return msg.peer == peer && msg.tag == tag; };
auto it = std::find_if(unexpectedMessages_.begin(), unexpectedMessages_.end(), lambda);
if (it != unexpectedMessages_.end()) {
// found a match
netRecv(it->sock.get(), data, size);
MSCCLPPTHROW(mscclppSocketClose(it->sock.get()));
unexpectedMessages_.erase(it);
return;
}
// didn't find one
while (true) {