import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
+import java.util.HashMap;
import java.util.List;
+import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import net.pterodactylus.fcp.highlevel.HighLevelCallback;
import net.pterodactylus.fcp.highlevel.HighLevelCallbackListener;
import net.pterodactylus.fcp.highlevel.HighLevelClient;
+import net.pterodactylus.fcp.highlevel.HighLevelProgress;
+import net.pterodactylus.fcp.highlevel.HighLevelProgressListener;
import net.pterodactylus.fcp.highlevel.RequestListResult;
import net.pterodactylus.fcp.highlevel.RequestResult;
import net.pterodactylus.util.logging.Logging;
* @author David ‘Bombe’ Roden <bombe@freenetproject.org>
* @version $Id$
*/
-public class RequestManager implements NodeListener {
+public class RequestManager implements NodeListener, HighLevelProgressListener {
/** Logger. */
private static final Logger logger = Logging.getLogger(RequestManager.class.getName());
/** The node manager. */
private NodeManager nodeManager;
+ /** Request lists for all nodes. */
+ @SuppressWarnings("unused")
+ private Map<Node, Map<String, Request>> nodeRequests = Collections.synchronizedMap(new HashMap<Node, Map<String, Request>>());
+
//
// EVENT MANAGEMENT
//
/**
* Notifies all listeners that a request was added.
*
- * @param node
- * The node that added the request
* @param request
* The request that was added
*/
- private void fireRequestAdded(Node node, Request request) {
+ private void fireRequestAdded(Request request) {
for (RequestListener requestListener: requestListeners) {
- requestListener.requestAdded(node, request);
+ requestListener.requestAdded(request);
}
}
+ /**
+ * Notifies all listeners that a request progressed.
+ *
+ * @param request
+ * The request
+ */
+ private void fireRequestProgressed(Request request) {
+ for (RequestListener requestListener: requestListeners) {
+ requestListener.requestProgressed(request);
+ }
+ }
+
+ // TODO - fireRequestRemoved
+
//
// ACCESSORS
//
// ACTIONS
//
+ //
+ // PRIVATE ACTIONS
+ //
+
/**
* Requests a list of all running requests from a node. This method will
* block until the request has been sent!
* @throws IOException
* if an I/O error occurs while communicating with the node
*/
- public void getRequests(final Node node) throws IOException {
- HighLevelClient highLevelClient = nodeManager.borrowHighLevelClient(node);
+ private void getRequests(final Node node) throws IOException {
+ HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
if (highLevelClient == null) {
logger.log(Level.WARNING, "no client for node: " + node);
return;
}
- try {
- HighLevelCallback<RequestListResult> requestListCallback = highLevelClient.getRequests();
- requestListCallback.addHighLevelCallbackListener(new HighLevelCallbackListener<RequestListResult>() {
-
- @SuppressWarnings("synthetic-access")
- public void gotResult(HighLevelCallback<RequestListResult> highLevelCallback) {
- RequestListResult requestListResult;
- try {
- requestListResult = highLevelCallback.getResult();
- } catch (InterruptedException e) {
- logger.log(Level.SEVERE, "getResult() blocked and was interrupted");
- return;
- }
- for (RequestResult requestResult: requestListResult) {
- Request request = new Request(requestResult.getIdentifier());
- /* TODO - fill request */
- fireRequestAdded(node, request);
- }
+ final Map<String, Request> identifierRequests = new HashMap<String, Request>();
+ nodeRequests.put(node, identifierRequests);
+ HighLevelCallback<RequestListResult> requestListCallback = highLevelClient.getRequests();
+ requestListCallback.addHighLevelCallbackListener(new HighLevelCallbackListener<RequestListResult>() {
+
+ @SuppressWarnings("synthetic-access")
+ public void gotResult(HighLevelCallback<RequestListResult> highLevelCallback) {
+ RequestListResult requestListResult;
+ try {
+ requestListResult = highLevelCallback.getResult();
+ } catch (InterruptedException e) {
+ logger.log(Level.SEVERE, "getResult() blocked and was interrupted");
+ return;
}
- });
- } finally {
- nodeManager.returnHighLevelClient(highLevelClient);
- }
+ for (RequestResult requestResult: requestListResult) {
+ Request request = new Request(node, requestResult.getIdentifier());
+ identifierRequests.put(requestResult.getIdentifier(), request);
+ /* TODO - fill request */
+ fireRequestAdded(request);
+ }
+ }
+ });
}
//
/**
* {@inheritDoc}
*/
+ public void nodeAdded(Node node) {
+ HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
+ if (highLevelClient == null) {
+ logger.warning("got nodeAdded but no high-level client: " + node);
+ return;
+ }
+ highLevelClient.addHighLevelProgressListener(this);
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ public void nodeRemoved(Node node) {
+ HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
+ if (highLevelClient == null) {
+ logger.warning("got nodeRemoved but no high-level client: " + node);
+ return;
+ }
+ highLevelClient.removeHighLevelProgressListener(this);
+ }
+
+ /**
+ * {@inheritDoc}
+ */
public void nodeConnected(Node node) {
- /* TODO - get all requests. */
+ HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
+ if (highLevelClient == null) {
+ logger.log(Level.WARNING, "got no high-level client for node " + node);
+ return;
+ }
+ try {
+ highLevelClient.setWatchGlobal(true);
+ getRequests(node);
+ } catch (IOException e) {
+ /* ignore exception, disconnects are handled elsewhere. */
+ }
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ public void nodeConnectionFailed(Node node, Throwable cause) {
+ /* we don't care about this. */
}
/**
* {@inheritDoc}
*/
public void nodeDisconnected(Node node, Throwable throwable) {
- /* TODO - remove all requests. */
+ HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
+ if (highLevelClient == null) {
+ logger.warning("got nodeDisconnected from node without high-level client: " + node);
+ return;
+ }
+ }
+
+ //
+ // INTERFACE HighLevelProgressListener
+ //
+
+ /**
+ * @see net.pterodactylus.fcp.highlevel.HighLevelProgressListener#progressReceived(HighLevelClient,
+ * String, HighLevelProgress)
+ */
+ public void progressReceived(HighLevelClient highLevelClient, String identifier, HighLevelProgress highLevelProgress) {
+ Node node = nodeManager.getNode(highLevelClient);
+ if (node == null) {
+ logger.warning("got high-level client without node: " + highLevelClient);
+ return;
+ }
+ Map<String, Request> identifierRequests = nodeRequests.get(node);
+ if (identifierRequests == null) {
+ logger.warning("got node without request map: " + node);
+ return;
+ }
+ Request request = identifierRequests.get(identifier);
+ if (request == null) {
+ logger.warning("got progress for unknown request: " + identifier);
+ return;
+ }
+ request.setTotalBlocks(highLevelProgress.getTotalBlocks());
+ request.setRequiredBlocks(highLevelProgress.getRequiredBlocks());
+ request.setSuccessfulBlocks(highLevelProgress.getSuccessfulBlocks());
+ request.setFailedBlocks(highLevelProgress.getFailedBlocks());
+ request.setFatallyFailedBlocks(highLevelProgress.getFatallyFailedBlocks());
+ request.setTotalFinalized(highLevelProgress.isTotalFinalized());
+ fireRequestProgressed(request);
}
}