X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fquelaton%2FListPeersCommandImpl.java;fp=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fquelaton%2FListPeersCommandImpl.java;h=6a352d57341190b75acdbeb858c308f3cd334aa5;hb=7a3be5ca037785f4f1acb225381578db12d15eb1;hp=f4907f03e1fb732b3b312227593850846a2426a8;hpb=91de0a638af50527fc314ed97a095d1beaf36866;p=jFCPlib.git

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ListPeersCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/ListPeersCommandImpl.java
index f4907f0..6a352d5 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ListPeersCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ListPeersCommandImpl.java
@@ -6,6 +6,7 @@ import java.util.HashSet;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.function.Supplier;
 
 import net.pterodactylus.fcp.EndListPeers;
 import net.pterodactylus.fcp.ListPeers;
@@ -24,12 +25,14 @@ public class ListPeersCommandImpl implements ListPeersCommand {
 
 	private final ListeningExecutorService threadPool;
 	private final ConnectionSupplier connectionSupplier;
+	private final Supplier<String> identifierGenerator;
 	private final AtomicBoolean includeMetadata = new AtomicBoolean(false);
 	private final AtomicBoolean includeVolatile = new AtomicBoolean(false);
 
-	public ListPeersCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier) {
+	public ListPeersCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier, Supplier<String> identifierGenerator) {
 		this.threadPool = MoreExecutors.listeningDecorator(threadPool);
 		this.connectionSupplier = connectionSupplier;
+		this.identifierGenerator = identifierGenerator;
 	}
 
 	@Override
@@ -50,8 +53,7 @@ public class ListPeersCommandImpl implements ListPeersCommand {
 	}
 
 	private Collection<Peer> executeDialog() throws InterruptedException, ExecutionException, IOException {
-		String identifier = new RandomIdentifierGenerator().generate();
-		ListPeers listPeers = new ListPeers(identifier, includeMetadata.get(), includeVolatile.get());
+		ListPeers listPeers = new ListPeers(identifierGenerator.get(), includeMetadata.get(), includeVolatile.get());
 		try (ListPeersDialog listPeersDialog = new ListPeersDialog()) {
 			return listPeersDialog.send(listPeers).get();
 		}