1 package net.pterodactylus.fcp.quelaton;
3 import java.io.IOException;
4 import java.util.concurrent.ExecutionException;
5 import java.util.concurrent.ExecutorService;
6 import java.util.concurrent.atomic.AtomicReference;
8 import net.pterodactylus.fcp.ClientHello;
9 import net.pterodactylus.fcp.FcpConnection;
10 import net.pterodactylus.fcp.NodeHello;
12 import com.google.common.util.concurrent.ListenableFuture;
13 import com.google.common.util.concurrent.ListeningExecutorService;
14 import com.google.common.util.concurrent.MoreExecutors;
17 * Internal <code>ClientHello</code> implementation based on {@link FcpDialog}.
19 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
21 public class ClientHelloImpl {
23 private final ListeningExecutorService threadPool;
24 private final String hostname;
25 private final int port;
26 private final AtomicReference<String> clientName = new AtomicReference<>();
28 public ClientHelloImpl(ExecutorService threadPool, String hostname, int port) {
29 this.threadPool = MoreExecutors.listeningDecorator(threadPool);
30 this.hostname = hostname;
34 public Executable<FcpConnection> withName(String name) {
39 private ListenableFuture<FcpConnection> execute() {
40 return threadPool.submit(this::establishConnection);
43 private FcpConnection establishConnection() throws IOException {
44 FcpConnection connection = new FcpConnection(hostname, port);
46 ClientHello clientHello = new ClientHello(clientName.get(), "2.0");
47 try (ClientHelloDialog clientHelloDialog = new ClientHelloDialog(connection)) {
48 if (clientHelloDialog.send(clientHello).get()) {
51 } catch (InterruptedException | ExecutionException e) {
53 throw new IOException(String.format("Could not connect to %s:%d.", hostname, port), e);
56 throw new IOException(String.format("Could not connect to %s:%d.", hostname, port));
59 private class ClientHelloDialog extends FcpDialog<Boolean> {
61 private final AtomicReference<NodeHello> receivedNodeHello = new AtomicReference<>();
63 public ClientHelloDialog(FcpConnection connection) {
64 super(ClientHelloImpl.this.threadPool, connection);
68 protected boolean isFinished() {
69 return receivedNodeHello.get() != null;
73 protected Boolean getResult() {
74 return receivedNodeHello.get() != null;
78 protected void consumeNodeHello(NodeHello nodeHello) {
79 receivedNodeHello.set(nodeHello);