mirrors/bisq
1
0
Fork 0
mirror of https://github.com/bisq-network/bisq.git synced 2024-11-19 09:52:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Replace persistedPeers with peerList

Browse Source
This commit is contained in:
chimp1984 2020-10-07 15:22:50 -05:00
parent 25bfe2d6ad
commit 25526750a8
No known key found for this signature in database
GPG Key ID: 9801B4EC591F90E3
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
p2p/src/main/java/bisq/network/p2p/peers/PeerManager.java
View File

@ -43,6 +43,7 @@ import javax.inject.Named;
import com.google.common.annotations.VisibleForTesting;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Comparator;
import java.util.Date;
import java.util.HashSet;
@ -108,9 +109,6 @@ public class PeerManager implements ConnectionListener, PersistedDataHost {
// Persistable peerList
private final PeerList peerList = new PeerList();
// Peers we had persisted TODO use peerList instead
@Getter
private final Set<Peer> persistedPeers = new HashSet<>();
// Peers we got reported from other peers
@Getter
private final Set<Peer> reportedPeers = new HashSet<>();
@ -188,7 +186,7 @@ public class PeerManager implements ConnectionListener, PersistedDataHost {
public void readPersisted() {
PeerList persisted = persistenceManager.getPersisted();
if (persisted != null) {
this.persistedPeers.addAll(persisted.getList());
peerList.setAll(persisted.getList());
}
}
@ -251,11 +249,16 @@ public class PeerManager implements ConnectionListener, PersistedDataHost {
public Set<Peer> getAllPeers() {
Set<Peer> allPeers = new HashSet<>(getLivePeers());
allPeers.addAll(persistedPeers);
allPeers.addAll(peerList.getList());
allPeers.addAll(reportedPeers);
return allPeers;
}
public Collection<Peer> getPersistedPeers() {
return peerList.getList();
}
///////////////////////////////////////////////////////////////////////////////////////////
// ConnectionListener implementation
///////////////////////////////////////////////////////////////////////////////////////////
@ -507,7 +510,7 @@ public class PeerManager implements ConnectionListener, PersistedDataHost {
reportedPeers.addAll(reportedPeersToAdd);
purgeReportedPeersIfExceeds();
persistedPeers.addAll(reportedPeersToAdd);
peerList.getList().addAll(reportedPeersToAdd);
purgePersistedPeersIfExceeds();
requestPersistence();
@ -582,8 +585,8 @@ public class PeerManager implements ConnectionListener, PersistedDataHost {
///////////////////////////////////////////////////////////////////////////////////////////
private boolean removePersistedPeer(Peer persistedPeer) {
if (persistedPeers.contains(persistedPeer)) {
persistedPeers.remove(persistedPeer);
if (peerList.getList().contains(persistedPeer)) {
peerList.getList().remove(persistedPeer);
requestPersistence();
return true;
} else {
@ -592,7 +595,6 @@ public class PeerManager implements ConnectionListener, PersistedDataHost {
}
private void requestPersistence() {
peerList.setAll(persistedPeers);
persistenceManager.requestPersistence();
}
@ -603,27 +605,27 @@ public class PeerManager implements ConnectionListener, PersistedDataHost {
}
private Optional<Peer> getOptionalPersistedPeer(NodeAddress nodeAddress) {
return persistedPeers.stream()
return peerList.getList().stream()
.filter(e -> e.getNodeAddress().equals(nodeAddress))
.findAny();
}
private void removeTooOldPersistedPeers() {
Set<Peer> persistedPeersToRemove = persistedPeers.stream()
Set<Peer> persistedPeersToRemove = peerList.getList().stream()
.filter(reportedPeer -> new Date().getTime() - reportedPeer.getDate().getTime() > MAX_AGE)
.collect(Collectors.toSet());
persistedPeersToRemove.forEach(this::removePersistedPeer);
}
private void purgePersistedPeersIfExceeds() {
int size = persistedPeers.size();
int size = peerList.getList().size();
int limit = MAX_PERSISTED_PEERS;
if (size > limit) {
log.trace("We have already {} persisted peers which exceeds our limit of {}." +
"We remove random peers from the persisted peers list.", size, limit);
int diff = size - limit;
List<Peer> list = new ArrayList<>(persistedPeers);
// we dont use sorting by lastActivityDate to avoid attack vectors and keep it more random
List<Peer> list = new ArrayList<>(peerList.getList());
// we don't use sorting by lastActivityDate to avoid attack vectors and keep it more random
for (int i = 0; i < diff; i++) {
if (!list.isEmpty()) {
Peer toRemove = list.remove(new Random().nextInt(list.size()));