external/ton
1
0
Fork 0
mirror of https://github.com/ton-blockchain/ton synced 2025-03-09 15:40:10 +00:00
Code Issues Releases Wiki Activity

Improve selection neighbours in overlay

Browse source
This commit is contained in:
SpyCheese 2024-07-15 16:21:01 +03:00
parent 5f9efc7add
commit f518c10444
1 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
overlay/overlay-peers.cpp
View file

@ -216,7 +216,7 @@ void OverlayImpl::add_peer(OverlayNode node) {
CHECK(X); CHECK(X);
if (peer_list_.neighbours_.size() < max_neighbours() && if (peer_list_.neighbours_.size() < max_neighbours() &&
!(X->get_node()->flags() & OverlayMemberFlags::DoNotReceiveBroadcasts)) { !(X->get_node()->flags() & OverlayMemberFlags::DoNotReceiveBroadcasts) && X->get_id() != local_id_) {
peer_list_.neighbours_.push_back(X->get_id()); peer_list_.neighbours_.push_back(X->get_id());
X->set_neighbour(true); X->set_neighbour(true);
} }
@ -622,9 +622,17 @@ std::vector<adnl::AdnlNodeIdShort> OverlayImpl::get_neighbours(td::uint32 max_si
return peer_list_.neighbours_; return peer_list_.neighbours_;
} else { } else {
std::vector<adnl::AdnlNodeIdShort> vec; std::vector<adnl::AdnlNodeIdShort> vec;
std::vector<td::uint32> ul;
for (td::uint32 i = 0; i < max_size; i++) { for (td::uint32 i = 0; i < max_size; i++) {
vec.push_back( td::uint32 t = td::Random::fast(0, static_cast<td::int32>(peer_list_.neighbours_.size()) - 1 - i);
peer_list_.neighbours_[td::Random::fast(0, static_cast<td::int32>(peer_list_.neighbours_.size()) - 1)]); td::uint32 j;
for (j = 0; j < i; j++) {
if (ul[j] <= t) {
t++;
}
}
ul.emplace(ul.begin() + j, t);
vec.push_back(peer_list_.neighbours_[t]);
} }
return vec; return vec;
} }