Whitespace and formatting fixes.

[jSite.git] / src / de / todesbaum / util / freenet / fcp2 / Connection.java

diff --git a/src/de/todesbaum/util/freenet/fcp2/Connection.java b/src/de/todesbaum/util/freenet/fcp2/Connection.java
index 9932d25..3fd39f6 100644 (file)
--- a/src/de/todesbaum/util/freenet/fcp2/Connection.java
+++ b/src/de/todesbaum/util/freenet/fcp2/Connection.java
@@ -1,5 +1,5 @@
 /*
- * todesbaum-lib - 
+ * todesbaum-lib -
  * Copyright (C) 2006 David Roden
  *
  * This program is free software; you can redistribute it and/or modify
@@ -38,7 +38,7 @@ import de.todesbaum.util.io.TempFileInputStream;
 
 /**
  * A physical connection to a Freenet node.
- * 
+ *
  * @author David Roden <droden@gmail.com>
  * @version $Id$
  */
@@ -71,9 +71,12 @@ public class Connection {
        /** The NodeHello message sent by the node on connect. */
        protected Message nodeHello;
 
+       /** The temp directory to use. */
+       private String tempDirectory;
+
        /**
         * Creates a new connection to the specified node with the specified name.
-        * 
+        *
         * @param node
         *            The node to connect to
         * @param name
@@ -86,7 +89,7 @@ public class Connection {
 
        /**
         * Adds a listener that gets notified on connection events.
-        * 
+        *
         * @param connectionListener
         *            The listener to add
         */
@@ -97,7 +100,7 @@ public class Connection {
        /**
         * Removes a listener from the list of registered listeners. Only the first
         * matching listener is removed.
-        * 
+        *
         * @param connectionListener
         *            The listener to remove
         * @see List#remove(java.lang.Object)
@@ -108,12 +111,12 @@ public class Connection {
 
        /**
         * Notifies listeners about a received message.
-        * 
+        *
         * @param message
         *            The received message
         */
        protected void fireMessageReceived(Message message) {
-               for (ConnectionListener connectionListener: connectionListeners) {
+               for (ConnectionListener connectionListener : connectionListeners) {
                        connectionListener.messageReceived(this, message);
                }
        }
@@ -122,14 +125,14 @@ public class Connection {
         * Notifies listeners about the loss of the connection.
         */
        protected void fireConnectionTerminated() {
-               for (ConnectionListener connectionListener: connectionListeners) {
+               for (ConnectionListener connectionListener : connectionListeners) {
                        connectionListener.connectionTerminated(this);
                }
        }
 
        /**
         * Returns the name of the connection.
-        * 
+        *
         * @return The name of the connection
         */
        public String getName() {
@@ -137,8 +140,19 @@ public class Connection {
        }
 
        /**
+        * Sets the temp directory to use for creation of temporary files.
+        *
+        * @param tempDirectory
+        *            The temp directory to use, or {@code null} to use the default
+        *            temp directory
+        */
+       public void setTempDirectory(String tempDirectory) {
+               this.tempDirectory = tempDirectory;
+       }
+
+       /**
         * Connects to the node.
-        * 
+        *
         * @return <code>true</code> if the connection succeeded and the node
         *         returned a NodeHello message
         * @throws IOException
@@ -180,7 +194,7 @@ public class Connection {
 
        /**
         * Returns whether this connection is still connected to the node.
-        * 
+        *
         * @return <code>true</code> if this connection is still valid,
         *         <code>false</code> otherwise
         */
@@ -190,7 +204,7 @@ public class Connection {
 
        /**
         * Returns the NodeHello message the node sent on connection.
-        * 
+        *
         * @return The NodeHello message of the node
         */
        public Message getNodeHello() {
@@ -237,7 +251,7 @@ public class Connection {
 
        /**
         * Executes the specified command.
-        * 
+        *
         * @param command
         *            The command to execute
         * @throws IllegalStateException
@@ -269,7 +283,7 @@ public class Connection {
         * The reader thread for this connection. This is essentially a thread that
         * reads lines from the node, creates messages from them and notifies
         * listeners about the messages.
-        * 
+        *
         * @author David Roden &lt;droden@gmail.com&gt;
         * @version $Id$
         */
@@ -281,7 +295,7 @@ public class Connection {
 
                /**
                 * Creates a new reader that reads from the specified input stream.
-                * 
+                *
                 * @param nodeInputStream
                 *            The input stream to read from
                 */
@@ -313,7 +327,7 @@ public class Connection {
                                                /* need to read message from stream now */
                                                File tempFile = null;
                                                try {
-                                                       tempFile = File.createTempFile("fcpv2", "data");
+                                                       tempFile = File.createTempFile("fcpv2", "data", (tempDirectory != null) ? new File(tempDirectory) : null);
                                                        tempFile.deleteOnExit();
                                                        FileOutputStream tempFileOutputStream = new FileOutputStream(tempFile);
                                                        long dataLength = Long.parseLong(message.get("DataLength"));