X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsonitus%2Fdata%2FPipeline.java;h=ee67eec2f1fd89da67853fb1727b5bfb7ba64f7c;hb=fa5196e029c48636f0c318311244395cfae49953;hp=4409151fba9cfeeb05a0c4451fc865a689203141;hpb=57737deafd02c03fcea5c3a3465d910338558f29;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 4409151..ee67eec 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;
@@ -45,7 +46,7 @@ import com.google.common.util.concurrent.MoreExecutors;
*
* @author David âBombeâ Roden
*/
-public class Pipeline implements Iterable {
+public class Pipeline implements Iterable {
/** The logger. */
private static final Logger logger = Logger.getLogger(Pipeline.class.getName());
@@ -98,6 +99,27 @@ public class Pipeline implements Iterable {
return sinks.get(source);
}
+ /**
+ * Returns the traffic counters of the given controlled component.
+ *
+ * @param controlledComponent
+ * The controlled component to get the traffic counters for
+ * @return The traffic counters for the given controlled component
+ */
+ public TrafficCounter trafficCounter(ControlledComponent controlledComponent) {
+ long input = -1;
+ long output = -1;
+ for (Connection connection : connections) {
+ /* the connection where the source matches knows the output. */
+ if (connection.source.equals(controlledComponent)) {
+ output = connection.counter();
+ } else if (connection.sinks.contains(controlledComponent)) {
+ input = connection.counter();
+ }
+ }
+ return new TrafficCounter(input, output);
+ }
+
//
// ACTIONS
//
@@ -149,8 +171,8 @@ public class Pipeline implements Iterable {
//
@Override
- public Iterator iterator() {
- return ImmutableSet.builder().add(source).addAll(sinks.values()).build().iterator();
+ public Iterator iterator() {
+ return ImmutableSet.builder().add(source).addAll(sinks.values()).build().iterator();
}
//
@@ -369,4 +391,58 @@ public class Pipeline implements Iterable {
}
+ /**
+ * Container for input and output counters.
+ *
+ * @author David âBombeâ Roden
+ */
+ 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 input() {
+ return (input == -1) ? Optional.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 output() {
+ return (output == -1) ? Optional.absent() : Optional.of(output);
+ }
+
+ }
+
}