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

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/RemovePeerCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/RemovePeerCommandImpl.java
index e5083d7..b0113b8 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/RemovePeerCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/RemovePeerCommandImpl.java
@@ -3,7 +3,6 @@ package net.pterodactylus.fcp.quelaton;
 import java.io.IOException;
 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;
 
@@ -64,31 +63,18 @@ public class RemovePeerCommandImpl implements RemovePeerCommand {
 
 	private class RemovePeerDialog extends FcpDialog<Boolean> {
 
-		private final AtomicBoolean finished = new AtomicBoolean();
-		private final AtomicBoolean removed = new AtomicBoolean();
-
 		public RemovePeerDialog() throws IOException {
-			super(threadPool, connectionSupplier.get());
-		}
-
-		@Override
-		protected boolean isFinished() {
-			return finished.get() || removed.get();
-		}
-
-		@Override
-		protected Boolean getResult() {
-			return removed.get();
+			super(threadPool, connectionSupplier.get(), false);
 		}
 
 		@Override
 		protected void consumePeerRemoved(PeerRemoved peerRemoved) {
-			removed.set(true);
+			setResult(true);
 		}
 
 		@Override
 		protected void consumeUnknownNodeIdentifier(UnknownNodeIdentifier unknownNodeIdentifier) {
-			finished.set(true);
+			finish();
 		}
 
 	}