X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsonitus%2Fdata%2FPipeline.java;h=975c21c45a365da709e80e94ae3b510055e2f4aa;hb=8a4389d8a0567473daf35aaac3e084f7ed8bcd12;hp=a38d6fe62d1e2f7ae3a73066f613740b8e89df38;hpb=6ff6dcb1223b2c948b2456b5d4e63c9a8ee0971d;p=sonitus.git

diff --git a/src/main/java/net/pterodactylus/sonitus/data/Pipeline.java b/src/main/java/net/pterodactylus/sonitus/data/Pipeline.java
index a38d6fe..975c21c 100644
--- a/src/main/java/net/pterodactylus/sonitus/data/Pipeline.java
+++ b/src/main/java/net/pterodactylus/sonitus/data/Pipeline.java
@@ -30,6 +30,7 @@ import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.logging.Logger;
 
 import com.google.common.base.Function;
+import com.google.common.base.Optional;
 import com.google.common.base.Preconditions;
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.FluentIterable;
@@ -98,6 +99,27 @@ public class Pipeline implements Iterable<Controlled> {
 		return sinks.get(source);
 	}
 
+	/**
+	 * Returns the traffic counters of the given controlled component.
+	 *
+	 * @param controlled
+	 * 		The controlled component to get the traffic counters for
+	 * @return The traffic counters for the given controlled component
+	 */
+	public TrafficCounter trafficCounter(Controlled controlled) {
+		long input = -1;
+		long output = -1;
+		for (Connection connection : connections) {
+			/* the connection where the source matches knows the output. */
+			if (connection.source.equals(controlled)) {
+				output = connection.counter();
+			} else if (connection.sinks.contains(controlled)) {
+				input = connection.counter();
+			}
+		}
+		return new TrafficCounter(input, output);
+	}
+
 	//
 	// ACTIONS
 	//
@@ -260,6 +282,9 @@ public class Pipeline implements Iterable<Controlled> {
 		/** The executor service. */
 		private final ExecutorService executorService;
 
+		/** The number of copied bytes. */
+		private long counter;
+
 		/**
 		 * Creates a new connection.
 		 *
@@ -279,6 +304,20 @@ public class Pipeline implements Iterable<Controlled> {
 		}
 
 		//
+		// ACCESSORS
+		//
+
+		/**
+		 * Returns the number of bytes that this connection has received from its
+		 * source during its lifetime.
+		 *
+		 * @return The number of processed input bytes
+		 */
+		public long counter() {
+			return counter;
+		}
+
+		//
 		// ACTIONS
 		//
 
@@ -333,6 +372,7 @@ public class Pipeline implements Iterable<Controlled> {
 					for (Future<Void> future : futures) {
 						future.get();
 					}
+					counter += buffer.length;
 				} catch (IOException e) {
 					/* TODO */
 					e.printStackTrace();
@@ -351,4 +391,58 @@ public class Pipeline implements Iterable<Controlled> {
 
 	}
 
+	/**
+	 * Container for input and output counters.
+	 *
+	 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+	 */
+	public static class TrafficCounter {
+
+		/** The number of input bytes. */
+		private final long input;
+
+		/** The number of output bytes. */
+		private final long output;
+
+		/**
+		 * Creates a new traffic counter.
+		 *
+		 * @param input
+		 * 		The number of input bytes (may be {@code -1} to signify non-available
+		 * 		input)
+		 * @param output
+		 * 		The number of output bytes (may be {@code -1} to signify non-available
+		 * 		output)
+		 */
+		public TrafficCounter(long input, long output) {
+			this.input = input;
+			this.output = output;
+		}
+
+		//
+		// ACCESSORS
+		//
+
+		/**
+		 * Returns the number of input bytes.
+		 *
+		 * @return The number of input bytes, or {@link Optional#absent()} if the
+		 *         component can not receive input
+		 */
+		public Optional<Long> input() {
+			return (input == -1) ? Optional.<Long>absent() : Optional.of(input);
+		}
+
+		/**
+		 * Returns the number of output bytes.
+		 *
+		 * @return The number of output bytes, or {@link Optional#absent()} if the
+		 *         component can not send output
+		 */
+		public Optional<Long> output() {
+			return (output == -1) ? Optional.<Long>absent() : Optional.of(output);
+		}
+
+	}
+
 }