1 files changed, 51 insertions, 0 deletions

diff --git a/src/main/java/org/gnunet/requests/MatchingRequestContainer.java b/src/main/java/org/gnunet/requests/MatchingRequestContainer.java
new file mode 100644
index 0000000..656d162
--- /dev/null
+++ b/src/main/java/org/gnunet/requests/MatchingRequestContainer.java
@@ -0,0 +1,51 @@
+package org.gnunet.requests;
+
+import com.google.common.collect.Maps;
+import org.gnunet.mq.MessageQueue;
+import org.gnunet.util.Cancelable;
+
+import java.util.Map;
+
+
+public class MatchingRequestContainer<K, T extends RequestContainer.Request> extends RequestContainer {
+    private Map<K,T> requests = Maps.newHashMap();
+    private final MessageQueue mq;
+
+    public MatchingRequestContainer(MessageQueue mq) {
+        this.mq = mq;
+    }
+
+    public Cancelable addRequest(K key, final T request) {
+        if (requests.containsKey(key))
+            throw new AssertionError("key already present in request container");
+        requests.put(key, request);
+        mq.send(request.assembleRequest());
+        return new Cancelable() {
+            @Override
+            public void cancel() {
+                setRequestCancelled(request);
+                if (isRequestTransmitting(request)) {
+                    cancelRequestTransmission(request);
+                } else {
+                    request.cancel();
+                }
+            }
+        };
+    }
+
+    @Override
+    public void restart() {
+        Map<K,T> requestsOld = requests;
+        requests = Maps.newHashMap();
+        for (Map.Entry<K,T> e : requestsOld.entrySet()) {
+            if (!isRequestCancelled(e.getValue())) {
+                setRequestTransmitting(e.getValue(), false);
+                addRequest(e.getKey(), e.getValue());
+            }
+        }
+    }
+
+    public T getRequest(K key) {
+        return requests.get(key);
+    }
+}