1 files changed, 86 insertions, 0 deletions

diff --git a/src/main/java/org/gnunet/requests/SequentialRequestContainer.java b/src/main/java/org/gnunet/requests/SequentialRequestContainer.java
new file mode 100644
index 0000000..037055f
--- /dev/null
+++ b/src/main/java/org/gnunet/requests/SequentialRequestContainer.java
@@ -0,0 +1,86 @@
+package org.gnunet.requests;
+
+import org.gnunet.mq.Envelope;
+import org.gnunet.mq.MessageQueue;
+import org.gnunet.mq.NotifySentHandler;
+import org.gnunet.util.Cancelable;
+
+import java.util.LinkedList;
+
+/**
+ * Container for requests that are responded to in sequential order.
+ */
+public class SequentialRequestContainer<T extends RequestContainer.Request> extends RequestContainer {
+    private LinkedList<T> requests = new LinkedList<T>();
+    private MessageQueue mq;
+
+    int requestsActive = 0;
+
+    public SequentialRequestContainer(MessageQueue mq) {
+        this.mq = mq;
+    }
+
+    public T getRequest() {
+        return requests.getFirst();
+    }
+
+    public void next() {
+        if (requestsActive == 0 || requests.isEmpty())
+            throw new AssertionError();
+        requestsActive--;
+        requests.removeFirst();
+        if (requestsActive == 0 && !requests.isEmpty()) {
+            Request r = requests.getFirst();
+            setRequestTransmitting(r, true);
+            Envelope ev = r.assembleRequest();
+            setRequestTransmissionCancel(r, ev);
+            mq.send(r.assembleRequest());
+            requestsActive++;
+        }
+    }
+
+    public Cancelable addRequest(final T request) {
+        requests.addLast(request);
+        if (overlap || requestsActive == 0) {
+            requestsActive++;
+            setRequestTransmitting(request, true);
+            Envelope ev = request.assembleRequest();
+            ev.notifySent(new NotifySentHandler() {
+                @Override
+                public void onSent() {
+                    setRequestTransmitting(request, false);
+                }
+            });
+            setRequestTransmissionCancel(request, ev);
+            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() {
+        LinkedList<T> requestsOld = requests;
+        requests = new LinkedList<T>();
+        for (T r : requestsOld) {
+            if (!isRequestCancelled(r)) {
+                setRequestTransmitting(r, false);
+                addRequest(r);
+            }
+        }
+    }
+
+
+    public Iterable<T> iter() {
+        return requests;
+    }
+}