X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fnet%2Fpterodactylus%2Ffcp%2Fhighlevel%2FHighLevelClient.java;h=e263454b6f0e38fd3f4a192a47f05133ad34d174;hb=dca50466338d343fe4acb888554d8b11a8de0385;hp=a09b289b51251602c2cf93fb74522552fc3d868f;hpb=c1710356b299b5d3eebd019e69e370ac6f7b81b5;p=jFCPlib.git diff --git a/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java b/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java index a09b289..e263454 100644 --- a/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java +++ b/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java @@ -105,14 +105,8 @@ public class HighLevelClient { /** The name of the client. */ private final String clientName; - /** The address of the node. */ - private InetAddress address; - - /** The port number of the node. */ - private int port; - /** The FCP connection to the node. */ - private FcpConnection fcpConnection; + private FcpConnection fcpConnection = null; /** Listeners for high-level client events. */ private List highLevelClientListeners = Collections.synchronizedList(new ArrayList()); @@ -123,7 +117,7 @@ public class HighLevelClient { /** The listeners for progress events. */ private List highLevelProgressListeners = Collections.synchronizedList(new ArrayList()); - /** The callback for {@link #connect()}. */ + /** The callback for {@link #connect(String)}. */ private HighLevelCallback connectCallback; /** Mapping from request identifiers to callbacks. */ @@ -150,60 +144,9 @@ public class HighLevelClient { * * @param clientName * The name of the client - * @throws UnknownHostException - * if the hostname of the node can not be resolved. - */ - public HighLevelClient(String clientName) throws UnknownHostException { - this(clientName, "localhost"); - } - - /** - * Creates a new high-level client that connects to a node on the given - * host. - * - * @param clientName - * The name of the client - * @param host - * The hostname of the node - * @throws UnknownHostException - * if the hostname of the node can not be resolved. - */ - public HighLevelClient(String clientName, String host) throws UnknownHostException { - this(clientName, host, FcpConnection.DEFAULT_PORT); - } - - /** - * Creates a new high-level client that connects to a node on the given - * host. - * - * @param clientName - * The name of the client - * @param host - * The hostname of the node - * @param port - * The port number of the node - * @throws UnknownHostException - * if the hostname of the node can not be resolved. - */ - public HighLevelClient(String clientName, String host, int port) throws UnknownHostException { - this(clientName, InetAddress.getByName(host), port); - } - - /** - * Creates a new high-level client that connects to a node at the given - * address. - * - * @param clientName - * The name of the client - * @param address - * The address of the node - * @param port - * The port number of the node */ - public HighLevelClient(String clientName, InetAddress address, int port) { + public HighLevelClient(String clientName) { this.clientName = clientName; - this.address = address; - this.port = port; } // @@ -234,7 +177,7 @@ public class HighLevelClient { * Notifies all listeners that a client has connected. */ private void fireClientConnected() { - for (HighLevelClientListener highLevelClientListener: highLevelClientListeners) { + for (HighLevelClientListener highLevelClientListener : highLevelClientListeners) { highLevelClientListener.clientConnected(this); } } @@ -247,7 +190,7 @@ public class HighLevelClient { * if there was no exception */ private void fireClientDisconnected(Throwable throwable) { - for (HighLevelClientListener highLevelClientListener: highLevelClientListeners) { + for (HighLevelClientListener highLevelClientListener : highLevelClientListeners) { highLevelClientListener.clientDisconnected(this, throwable); } } @@ -282,8 +225,8 @@ public class HighLevelClient { * The progress of the request */ private void fireProgressReceived(String identifier, HighLevelProgress highLevelProgress) { - for (HighLevelProgressListener highLevelProgressListener: highLevelProgressListeners) { - highLevelProgressListener.progressReceived(identifier, highLevelProgress); + for (HighLevelProgressListener highLevelProgressListener : highLevelProgressListeners) { + highLevelProgressListener.progressReceived(this, identifier, highLevelProgress); } } @@ -302,6 +245,16 @@ public class HighLevelClient { return fcpConnection; } + /** + * Returns whether the node is connected. + * + * @return true if the node is currently connected, + * false otherwise + */ + public boolean isConnected() { + return fcpConnection != null; + } + // // ACTIONS // @@ -309,18 +262,61 @@ public class HighLevelClient { /** * Connects the client. * + * @param hostname + * The hostname of the node * @return A callback with a connection result + * @throws UnknownHostException + * if the hostname can not be resolved + * @throws IOException + * if an I/O error occurs communicating with the node + */ + public HighLevelCallback connect(String hostname) throws UnknownHostException, IOException { + return connect(hostname, 9481); + } + + /** + * Connects the client. + * + * @param hostname + * The hostname of the node + * @param port + * The port number of the node + * @return A callback with a connection result + * @throws UnknownHostException + * if the hostname can not be resolved * @throws IOException * if an I/O error occurs communicating with the node */ - public HighLevelCallback connect() throws IOException { - fcpConnection = new FcpConnection(address, port); - fcpConnection.addFcpListener(highLevelClientFcpListener); - fcpConnection.connect(); - ClientHello clientHello = new ClientHello(clientName); - connectCallback = new HighLevelCallback(new ConnectResult()); - fcpConnection.sendMessage(clientHello); - return connectCallback; + public HighLevelCallback connect(String hostname, int port) throws UnknownHostException, IOException { + return connect(InetAddress.getByName(hostname), port); + } + + /** + * Connects the client. + * + * @param address + * The address of the node + * @param port + * The port number of the node + * @return A callback with a connection result + * @throws IOException + * if an I/O error occurs communicating with the node + */ + public HighLevelCallback connect(InetAddress address, int port) throws IOException { + try { + synchronized (this) { + fcpConnection = new FcpConnection(address, port); + } + fcpConnection.addFcpListener(highLevelClientFcpListener); + fcpConnection.connect(); + ClientHello clientHello = new ClientHello(clientName); + connectCallback = new HighLevelCallback(new ConnectResult()); + fcpConnection.sendMessage(clientHello); + return connectCallback; + } catch (IOException ioe1) { + fcpConnection = null; + throw ioe1; + } } /** @@ -336,8 +332,11 @@ public class HighLevelClient { * @return A callback with the keypair * @throws IOException * if an I/O error occurs communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelCallback generateKey() throws IOException { + public HighLevelCallback generateKey() throws IOException, HighLevelException { + checkConnection(); String identifier = generateIdentifier("generateSSK"); GenerateSSK generateSSK = new GenerateSSK(identifier); HighLevelCallback keyGenerationCallback = new HighLevelCallback(new KeyGenerationResult(identifier)); @@ -355,8 +354,11 @@ public class HighLevelClient { * client-local queue * @throws IOException * if an I/O error occurs communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public void setWatchGlobal(boolean enabled) throws IOException { + public void setWatchGlobal(boolean enabled) throws IOException, HighLevelException { + checkConnection(); WatchGlobal watchGlobal = new WatchGlobal(enabled); fcpConnection.sendMessage(watchGlobal); } @@ -367,8 +369,11 @@ public class HighLevelClient { * @return A callback with the peer list * @throws IOException * if an I/O error occurs with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelCallback getPeers() throws IOException { + public HighLevelCallback getPeers() throws IOException, HighLevelException { + checkConnection(); String identifier = generateIdentifier("listPeers"); ListPeers listPeers = new ListPeers(identifier, true, true); HighLevelCallback peerListCallback = new HighLevelCallback(new PeerListResult(identifier)); @@ -385,8 +390,11 @@ public class HighLevelClient { * @return A peer callback * @throws IOException * if an I/O error occurs communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelCallback addPeer(String nodeRefFile) throws IOException { + public HighLevelCallback addPeer(String nodeRefFile) throws IOException, HighLevelException { + checkConnection(); String identifier = generateIdentifier("addPeer"); AddPeer addPeer = new AddPeer(nodeRefFile); HighLevelCallback peerCallback = new HighLevelCallback(new PeerResult(identifier)); @@ -403,8 +411,11 @@ public class HighLevelClient { * @return A peer callback * @throws IOException * if an I/O error occurs communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelCallback addPeer(URL nodeRefURL) throws IOException { + public HighLevelCallback addPeer(URL nodeRefURL) throws IOException, HighLevelException { + checkConnection(); String identifier = generateIdentifier("addPeer"); AddPeer addPeer = new AddPeer(nodeRefURL); HighLevelCallback peerCallback = new HighLevelCallback(new PeerResult(identifier)); @@ -421,8 +432,11 @@ public class HighLevelClient { * @return A peer callback * @throws IOException * if an I/O error occurs communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelCallback addPeer(NodeRef nodeRef) throws IOException { + public HighLevelCallback addPeer(NodeRef nodeRef) throws IOException, HighLevelException { + checkConnection(); String identifier = generateIdentifier("addPeer"); AddPeer addPeer = new AddPeer(nodeRef); HighLevelCallback peerCallback = new HighLevelCallback(new PeerResult(identifier)); @@ -444,8 +458,12 @@ public class HighLevelClient { * Whether you want to write to the given directory * @return A direct disk access callback * @throws IOException + * if an I/O error occurs communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelCallback checkDirectDiskAccess(String directory, boolean wantRead, boolean wantWrite) throws IOException { + public HighLevelCallback checkDirectDiskAccess(String directory, boolean wantRead, boolean wantWrite) throws IOException, HighLevelException { + checkConnection(); TestDDARequest testDDARequest = new TestDDARequest(directory, wantRead, wantWrite); HighLevelCallback directDiskAccessCallback = new HighLevelCallback(new DirectDiskAccessResult(directory)); directDiskAccessCallbacks.put(directory, directDiskAccessCallback); @@ -470,8 +488,11 @@ public class HighLevelClient { * @return A download result * @throws IOException * if an I/O error occurs communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelProgressCallback download(String uri, String filename, boolean global) throws IOException { + public HighLevelProgressCallback download(String uri, String filename, boolean global) throws IOException, HighLevelException { + checkConnection(); String identifier = generateIdentifier("download"); ClientGet clientGet = new ClientGet(uri, identifier, (filename == null) ? ReturnType.direct : ReturnType.disk); clientGet.setGlobal(global); @@ -487,8 +508,11 @@ public class HighLevelClient { * @return The request list result * @throws IOException * if an I/O errors communicating with the node + * @throws HighLevelException + * if the client is not connected */ - public HighLevelCallback getRequests() throws IOException { + public HighLevelCallback getRequests() throws IOException, HighLevelException { + checkConnection(); String identifier = generateIdentifier("list-persistent-requests"); ListPersistentRequests listPersistentRequests = new ListPersistentRequests(); synchronized (syncObject) { @@ -506,6 +530,21 @@ public class HighLevelClient { // /** + * Checks whether the client is already connected and throws an exception if + * it is not. + * + * @throws HighLevelException + * if the client is not connected + */ + private void checkConnection() throws HighLevelException { + synchronized (this) { + if (fcpConnection == null) { + throw new HighLevelException("client is not connected"); + } + } + } + + /** * Generates an identifier for the given function. * * @param function @@ -525,8 +564,10 @@ public class HighLevelClient { * if there was no exception */ private void disconnect(Throwable throwable) { - fcpConnection.close(); - fireClientDisconnected(throwable); + if (fcpConnection != null) { + fcpConnection.close(); + } + fcpConnection = null; } /** @@ -575,31 +616,31 @@ public class HighLevelClient { } if (identifier == null) { /* key generation callbacks */ - for (Entry> keyGenerationEntry: keyGenerationCallbacks.entrySet()) { + for (Entry> keyGenerationEntry : keyGenerationCallbacks.entrySet()) { keyGenerationEntry.getValue().getIntermediaryResult().setFailed(true); keyGenerationEntry.getValue().setDone(); } keyGenerationCallbacks.clear(); /* peer list callbacks. */ - for (Entry> peerListEntry: peerListCallbacks.entrySet()) { + for (Entry> peerListEntry : peerListCallbacks.entrySet()) { peerListEntry.getValue().getIntermediaryResult().setFailed(true); peerListEntry.getValue().setDone(); } peerListCallbacks.clear(); /* peer callbacks. */ - for (Entry> peerEntry: peerCallbacks.entrySet()) { + for (Entry> peerEntry : peerCallbacks.entrySet()) { peerEntry.getValue().getIntermediaryResult().setFailed(true); peerEntry.getValue().setDone(); } peerCallbacks.clear(); /* direct disk access callbacks. */ - for (Entry> directDiskAccessEntry: directDiskAccessCallbacks.entrySet()) { + for (Entry> directDiskAccessEntry : directDiskAccessCallbacks.entrySet()) { directDiskAccessEntry.getValue().getIntermediaryResult().setFailed(true); directDiskAccessEntry.getValue().setDone(); } directDiskAccessCallbacks.clear(); /* download callbacks. */ - for (Entry> downloadEntry: downloadCallbacks.entrySet()) { + for (Entry> downloadEntry : downloadCallbacks.entrySet()) { downloadEntry.getValue().getIntermediaryResult().setFailed(true); downloadEntry.getValue().setDone(); } @@ -719,6 +760,7 @@ public class HighLevelClient { } cancelIdentifier(null); disconnect(throwable); + fireClientDisconnected(throwable); } /** @@ -1071,7 +1113,6 @@ public class HighLevelClient { downloadResult.setFatallyFailedBlocks(simpleProgress.getFatallyFailed()); downloadResult.setTotalFinalized(simpleProgress.isFinalizedTotal()); downloadCallback.progressUpdated(); - return; } HighLevelProgress highLevelProgress = new HighLevelProgress(identifier, simpleProgress.getTotal(), simpleProgress.getRequired(), simpleProgress.getSucceeded(), simpleProgress.getFailed(), simpleProgress.getFatallyFailed(), simpleProgress.isFinalizedTotal()); fireProgressReceived(identifier, highLevelProgress);