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

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ListPeerNotesCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/ListPeerNotesCommandImpl.java
index 8e1df8c..95e885f 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ListPeerNotesCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ListPeerNotesCommandImpl.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;
 
@@ -66,36 +65,23 @@ public class ListPeerNotesCommandImpl implements ListPeerNotesCommand {
 
 	private class ListPeerNotesDialog extends FcpDialog<Optional<PeerNote>> {
 
-		private final AtomicReference<PeerNote> peerNote = new AtomicReference<>();
-		private final AtomicBoolean finished = new AtomicBoolean();
-
 		public ListPeerNotesDialog() throws IOException {
-			super(threadPool, connectionSupplier.get());
-		}
-
-		@Override
-		protected boolean isFinished() {
-			return finished.get();
-		}
-
-		@Override
-		protected Optional<PeerNote> getResult() {
-			return Optional.ofNullable(peerNote.get());
+			super(threadPool, connectionSupplier.get(), Optional.<PeerNote>empty());
 		}
 
 		@Override
 		protected void consumePeerNote(PeerNote peerNote) {
-			this.peerNote.set(peerNote);
+			setResult(Optional.ofNullable(peerNote));
 		}
 
 		@Override
 		protected void consumeEndListPeerNotes(EndListPeerNotes endListPeerNotes) {
-			finished.set(true);
+			finish();
 		}
 
 		@Override
 		protected void consumeUnknownNodeIdentifier(UnknownNodeIdentifier unknownNodeIdentifier) {
-			finished.set(true);
+			finish();
 		}
 
 	}