2 * jSite2 - RequestManager.java -
3 * Copyright © 2008 David Roden
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
20 package net.pterodactylus.jsite.core;
22 import java.io.IOException;
23 import java.util.ArrayList;
24 import java.util.Collections;
25 import java.util.HashMap;
26 import java.util.List;
28 import java.util.logging.Level;
29 import java.util.logging.Logger;
31 import net.pterodactylus.fcp.highlevel.HighLevelCallback;
32 import net.pterodactylus.fcp.highlevel.HighLevelCallbackListener;
33 import net.pterodactylus.fcp.highlevel.HighLevelClient;
34 import net.pterodactylus.fcp.highlevel.HighLevelProgress;
35 import net.pterodactylus.fcp.highlevel.HighLevelProgressListener;
36 import net.pterodactylus.fcp.highlevel.RequestListResult;
37 import net.pterodactylus.fcp.highlevel.RequestResult;
38 import net.pterodactylus.util.logging.Logging;
41 * The request manager keeps track of all the request on all connected nodes.
42 * The request manager is added to the {@link NodeManager} as a
43 * {@link NodeListener} so that it can fire request-removed events in case a
44 * node is disconnected.
46 * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
49 public class RequestManager implements NodeListener, HighLevelProgressListener {
52 private static final Logger logger = Logging.getLogger(RequestManager.class.getName());
54 /** Request listeners. */
55 private List<RequestListener> requestListeners = Collections.synchronizedList(new ArrayList<RequestListener>());
57 /** The node manager. */
58 private NodeManager nodeManager;
60 /** Request lists for all nodes. */
61 @SuppressWarnings("unused")
62 private Map<Node, Map<String, Request>> nodeRequests = Collections.synchronizedMap(new HashMap<Node, Map<String, Request>>());
69 * Adds a request listener.
71 * @param requestListener
72 * The request listener to add
74 public void addRequestListener(RequestListener requestListener) {
75 requestListeners.add(requestListener);
79 * Removes a request listener.
81 * @param requestListener
82 * The request listener to remove
84 public void removeRequestListener(RequestListener requestListener) {
85 requestListeners.remove(requestListener);
89 * Notifies all listeners that a request was added.
92 * The request that was added
94 private void fireRequestAdded(Request request) {
95 for (RequestListener requestListener: requestListeners) {
96 requestListener.requestAdded(request);
101 * Notifies all listeners that a request progressed.
106 private void fireRequestProgressed(Request request) {
107 for (RequestListener requestListener: requestListeners) {
108 requestListener.requestProgressed(request);
113 * Notifies all listeners that a request was removed.
116 * The request that was removed
118 private void fireRequestRemoved(Request request) {
119 for (RequestListener requestListener: requestListeners) {
120 requestListener.requestRemoved(request);
129 * Sets the node manager to use.
134 public void setNodeManager(NodeManager nodeManager) {
135 this.nodeManager = nodeManager;
147 * Requests a list of all running requests from a node. This method will
148 * block until the request has been sent!
151 * The node to get all requests for
152 * @throws IOException
153 * if an I/O error occurs while communicating with the node
155 private void getRequests(final Node node) throws IOException {
156 HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
157 if (highLevelClient == null) {
158 logger.log(Level.WARNING, "no client for node: " + node);
161 final Map<String, Request> identifierRequests = new HashMap<String, Request>();
162 nodeRequests.put(node, identifierRequests);
163 HighLevelCallback<RequestListResult> requestListCallback = highLevelClient.getRequests();
164 requestListCallback.addHighLevelCallbackListener(new HighLevelCallbackListener<RequestListResult>() {
166 @SuppressWarnings("synthetic-access")
167 public void gotResult(HighLevelCallback<RequestListResult> highLevelCallback) {
168 RequestListResult requestListResult;
170 requestListResult = highLevelCallback.getResult();
171 } catch (InterruptedException e) {
172 logger.log(Level.SEVERE, "getResult() blocked and was interrupted");
175 for (RequestResult requestResult: requestListResult) {
176 Request request = new Request(node, requestResult.getIdentifier());
177 identifierRequests.put(requestResult.getIdentifier(), request);
178 /* TODO - fill request */
179 fireRequestAdded(request);
186 // INTERFACE NodeListener
192 public void nodeAdded(Node node) {
193 HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
194 if (highLevelClient == null) {
195 logger.warning("got nodeAdded but no high-level client: " + node);
198 highLevelClient.addHighLevelProgressListener(this);
204 public void nodeRemoved(Node node) {
205 HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
206 if (highLevelClient == null) {
207 logger.warning("got nodeRemoved but no high-level client: " + node);
210 highLevelClient.removeHighLevelProgressListener(this);
216 public void nodeConnected(Node node) {
217 HighLevelClient highLevelClient = nodeManager.getHighLevelClient(node);
218 if (highLevelClient == null) {
219 logger.log(Level.WARNING, "got no high-level client for node " + node);
223 highLevelClient.setWatchGlobal(true);
225 } catch (IOException e) {
226 /* ignore exception, disconnects are handled elsewhere. */
233 public void nodeConnectionFailed(Node node, Throwable cause) {
234 /* we don't care about this. */
240 public void nodeDisconnected(Node node, Throwable throwable) {
241 Map<String, Request> identifierRequests = nodeRequests.get(node);
242 if (identifierRequests == null) {
243 logger.warning("got node without request map: " + node);
246 for (Request request: identifierRequests.values()) {
247 fireRequestRemoved(request);
249 identifierRequests.clear();
253 // INTERFACE HighLevelProgressListener
257 * @see net.pterodactylus.fcp.highlevel.HighLevelProgressListener#progressReceived(HighLevelClient,
258 * String, HighLevelProgress)
260 public void progressReceived(HighLevelClient highLevelClient, String identifier, HighLevelProgress highLevelProgress) {
261 Node node = nodeManager.getNode(highLevelClient);
263 logger.warning("got high-level client without node: " + highLevelClient);
266 Map<String, Request> identifierRequests = nodeRequests.get(node);
267 if (identifierRequests == null) {
268 logger.warning("got node without request map: " + node);
271 Request request = identifierRequests.get(identifier);
272 if (request == null) {
273 logger.warning("got progress for unknown request: " + identifier);
276 request.setTotalBlocks(highLevelProgress.getTotalBlocks());
277 request.setRequiredBlocks(highLevelProgress.getRequiredBlocks());
278 request.setSuccessfulBlocks(highLevelProgress.getSuccessfulBlocks());
279 request.setFailedBlocks(highLevelProgress.getFailedBlocks());
280 request.setFatallyFailedBlocks(highLevelProgress.getFatallyFailedBlocks());
281 request.setTotalFinalized(highLevelProgress.isTotalFinalized());
282 fireRequestProgressed(request);