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;
import net.pterodactylus.fcp.GetNode;
import net.pterodactylus.fcp.NodeData;
import com.google.common.util.concurrent.MoreExecutors;
/**
- * Default {@link GetNodeCommandImpl} implementation based on {@link FcpReplySequence}.
+ * Default {@link GetNodeCommandImpl} implementation based on {@link FcpDialog}.
*
* @author <a href="mailto:bombe@freenetproject.org">David ‘Bombe’ Roden</a>
*/
private final ListeningExecutorService threadPool;
private final ConnectionSupplier connectionSupplier;
+ private final Supplier<String> identifierGenerator;
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) {
+ public GetNodeCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier, Supplier<String> identifierGenerator) {
this.threadPool = MoreExecutors.listeningDecorator(threadPool);
this.connectionSupplier = connectionSupplier;
+ this.identifierGenerator = identifierGenerator;
}
@Override
}
@Override
- public ListenableFuture<NodeData> execute() {
- GetNode getNode = new GetNode(new RandomIdentifierGenerator().generate(), giveOpennetRef.get(), false, false);
- return threadPool.submit(() -> new GetNodeReplySequence().send(getNode).get());
+ public GetNodeCommand includePrivate() {
+ includePrivate.set(true);
+ return this;
}
- private class GetNodeReplySequence extends FcpReplySequence<NodeData> {
+ @Override
+ public GetNodeCommand includeVolatile() {
+ includeVolatile.set(true);
+ return this;
+ }
- private final AtomicReference<NodeData> nodeData = new AtomicReference<>();
+ @Override
+ public ListenableFuture<NodeData> execute() {
+ return threadPool.submit(this::executeDialog);
+ }
- public GetNodeReplySequence() throws IOException {
- super(threadPool, connectionSupplier.get());
+ private NodeData executeDialog() throws InterruptedException, ExecutionException, IOException {
+ GetNode getNode = new GetNode(identifierGenerator.get(), giveOpennetRef.get(),
+ includePrivate.get(), includeVolatile.get());
+ try (GetNodeDialog getNodeDialog = new GetNodeDialog()) {
+ return getNodeDialog.send(getNode).get();
}
+ }
- @Override
- protected boolean isFinished() {
- return nodeData.get() != null;
- }
+ private class GetNodeDialog extends FcpDialog<NodeData> {
- @Override
- protected NodeData getResult() {
- return nodeData.get();
+ public GetNodeDialog() throws IOException {
+ super(threadPool, connectionSupplier.get(), null);
}
@Override
protected void consumeNodeData(NodeData nodeData) {
- this.nodeData.set(nodeData);
+ setResult(nodeData);
}
}