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

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ClientGetCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/ClientGetCommandImpl.java
index 0c93fc3..d01b82e 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ClientGetCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ClientGetCommandImpl.java
@@ -6,6 +6,7 @@ import java.util.Optional;
 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.AllData;
 import net.pterodactylus.fcp.ClientGet;
@@ -26,6 +27,7 @@ class ClientGetCommandImpl implements ClientGetCommand {
 
 	private final ListeningExecutorService threadPool;
 	private final ConnectionSupplier connectionSupplier;
+	private final Supplier<String> identifierGenerator;
 
 	private boolean ignoreDataStore;
 	private boolean dataStoreOnly;
@@ -34,9 +36,10 @@ class ClientGetCommandImpl implements ClientGetCommand {
 	private boolean realTime;
 	private boolean global;
 
-	public ClientGetCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier) {
+	public ClientGetCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier, Supplier<String> identifierGenerator) {
 		this.threadPool = MoreExecutors.listeningDecorator(threadPool);
 		this.connectionSupplier = connectionSupplier;
+		this.identifierGenerator = identifierGenerator;
 	}
 
 	@Override
@@ -88,7 +91,7 @@ class ClientGetCommandImpl implements ClientGetCommand {
 	}
 
 	private ClientGet createClientGetCommand(String uri) {
-		String identifier = new RandomIdentifierGenerator().generate();
+		String identifier = identifierGenerator.get();
 		ClientGet clientGet = new ClientGet(uri, identifier, ReturnType.direct);
 		if (ignoreDataStore) {
 			clientGet.setIgnoreDataStore(true);