import net.pterodactylus.fcp.FcpConnection;
import net.pterodactylus.fcp.NodeHello;
+import com.google.common.util.concurrent.ListeningExecutorService;
+import com.google.common.util.concurrent.MoreExecutors;
+
/**
* Default {@link FcpClient} implementation.
*
*/
public class DefaultFcpClient implements FcpClient {
- private final ExecutorService threadPool;
+ private final ListeningExecutorService threadPool;
private final String hostname;
private final int port;
private final AtomicReference<FcpConnection> fcpConnection = new AtomicReference<>();
private final Supplier<String> clientName;
- private final Supplier<String> expectedVersion;
- public DefaultFcpClient(ExecutorService threadPool, String hostname, int port, Supplier<String> clientName,
- Supplier<String> expectedVersion) {
- this.threadPool = threadPool;
+ public DefaultFcpClient(ExecutorService threadPool, String hostname, int port, Supplier<String> clientName) {
+ this.threadPool = MoreExecutors.listeningDecorator(threadPool);
this.hostname = hostname;
this.port = port;
this.clientName = clientName;
- this.expectedVersion = expectedVersion;
}
private FcpConnection connect() throws IOException {
private FcpConnection createConnection() throws IOException {
FcpConnection connection = new FcpConnection(hostname, port);
connection.connect();
- FcpReplySequence<?> nodeHelloSequence = new FcpReplySequence<Void>(threadPool, connection) {
- private final AtomicReference<NodeHello> receivedNodeHello = new AtomicReference<>();
- private final AtomicBoolean receivedClosed = new AtomicBoolean();
- @Override
- protected boolean isFinished() {
- return receivedNodeHello.get() != null || receivedClosed.get();
- }
-
- @Override
- protected void consumeNodeHello(NodeHello nodeHello) {
- receivedNodeHello.set(nodeHello);
- }
-
- @Override
- protected void consumeCloseConnectionDuplicateClientName(
- CloseConnectionDuplicateClientName closeConnectionDuplicateClientName) {
- receivedClosed.set(true);
- }
- };
- ClientHello clientHello = new ClientHello(clientName.get(), expectedVersion.get());
+ FcpReplySequence<?> nodeHelloSequence = new ClientHelloReplySequence(connection);
+ ClientHello clientHello = new ClientHello(clientName.get(), "2.0");
try {
nodeHelloSequence.send(clientHello).get();
} catch (InterruptedException | ExecutionException e) {
return new ClientGetCommandImpl(threadPool, this::connect);
}
+ @Override
+ public ClientPutCommand clientPut() {
+ return new ClientPutCommandImpl(threadPool, this::connect);
+ }
+
+ @Override
+ public ListPeersCommand listPeers() {
+ return new ListPeersCommandImpl(threadPool, this::connect);
+ }
+
+ private class ClientHelloReplySequence extends FcpReplySequence<Void> {
+
+ private final AtomicReference<NodeHello> receivedNodeHello;
+ private final AtomicBoolean receivedClosed;
+
+ public ClientHelloReplySequence(FcpConnection connection) {
+ super(DefaultFcpClient.this.threadPool, connection);
+ receivedNodeHello = new AtomicReference<>();
+ receivedClosed = new AtomicBoolean();
+ }
+
+ @Override
+ protected boolean isFinished() {
+ return receivedNodeHello.get() != null || receivedClosed.get();
+ }
+
+ @Override
+ protected void consumeNodeHello(NodeHello nodeHello) {
+ receivedNodeHello.set(nodeHello);
+ }
+
+ @Override
+ protected void consumeCloseConnectionDuplicateClientName(
+ CloseConnectionDuplicateClientName closeConnectionDuplicateClientName) {
+ receivedClosed.set(true);
+ }
+
+ }
+
}