X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fquelaton%2FModifyPeerCommandImpl.java;h=d5f58400b7507cc788439ed5fa0ed077c9029a6d;hb=c7b30c245806a4a67d2e924d6d4e2a9e97916c5b;hp=9fff845bedf82fe1f10f0813d25451991b5e607c;hpb=7a3be5ca037785f4f1acb225381578db12d15eb1;p=jFCPlib.git

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ModifyPeerCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/ModifyPeerCommandImpl.java
index 9fff845..d5f5840 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ModifyPeerCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ModifyPeerCommandImpl.java
@@ -4,7 +4,6 @@ import java.io.IOException;
 import java.util.Optional;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ExecutorService;
-import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Supplier;
 
@@ -135,32 +134,18 @@ public class ModifyPeerCommandImpl implements ModifyPeerCommand {
 
 	private class ModifyPeerDialog extends FcpDialog<Optional<Peer>> {
 
-		private final AtomicBoolean finished = new AtomicBoolean();
-		private final AtomicReference<Peer> peer = new AtomicReference<>();
-
 		public ModifyPeerDialog() throws IOException {
-			super(threadPool, connectionSupplier.get());
-		}
-
-		@Override
-		protected boolean isFinished() {
-			return finished.get();
-		}
-
-		@Override
-		protected Optional<Peer> getResult() {
-			return Optional.ofNullable(peer.get());
+			super(threadPool, connectionSupplier.get(), Optional.<Peer>empty());
 		}
 
 		@Override
 		protected void consumePeer(Peer peer) {
-			this.peer.set(peer);
-			finished.set(true);
+			setResult(Optional.ofNullable(peer));
 		}
 
 		@Override
 		protected void consumeUnknownNodeIdentifier(UnknownNodeIdentifier unknownNodeIdentifier) {
-			finished.set(true);
+			finish();
 		}
 
 	}