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;
private final ListeningExecutorService threadPool;
private final ConnectionSupplier connectionSupplier;
private final AtomicBoolean giveOpennetRef = new AtomicBoolean(false);
+ private final AtomicBoolean includePrivate = new AtomicBoolean(false);
+ private final AtomicBoolean includeVolatile = new AtomicBoolean(false);
public GetNodeCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier) {
this.threadPool = MoreExecutors.listeningDecorator(threadPool);
}
@Override
+ public GetNodeCommand includePrivate() {
+ includePrivate.set(true);
+ return this;
+ }
+
+ @Override
+ public GetNodeCommand includeVolatile() {
+ includeVolatile.set(true);
+ return this;
+ }
+
+ @Override
public ListenableFuture<NodeData> execute() {
- GetNode getNode = new GetNode(new RandomIdentifierGenerator().generate(), giveOpennetRef.get(), false, false);
- return threadPool.submit(() -> new GetNodeReplySequence().send(getNode).get());
+ return threadPool.submit(this::executeSequence);
+ }
+
+ private NodeData executeSequence() throws InterruptedException, ExecutionException, IOException {
+ GetNode getNode = new GetNode(new RandomIdentifierGenerator().generate(), giveOpennetRef.get(),
+ includePrivate.get(), includeVolatile.get());
+ try (GetNodeReplySequence getNodeReplySequence = new GetNodeReplySequence()) {
+ return getNodeReplySequence.send(getNode).get();
+ }
}
private class GetNodeReplySequence extends FcpReplySequence<NodeData> {