iterating on fragmenter

Author: tmontgomery

src/main/java/io/reactivesocket/internal/PayloadFragmenter.java

@@ -27,11 +27,11 @@
  *
  * Not thread-safe
  */
-public class PayloadFragmenter implements Iterator<Frame>
+public class PayloadFragmenter implements Iterable<Frame>, Iterator<Frame>
 {
     private enum Type
     {
-        RESPONSE, REQUEST_CHANNEL
+        RESPONSE, RESPONSE_COMPLETE, REQUEST_CHANNEL
     }
 
     private final int metadataMtu;
@@ -56,13 +56,32 @@ public void resetForResponse(final int streamId, final Payload payload)
         type = Type.RESPONSE;
     }
 
+    public void resetForResponseComplete(final int streamId, final Payload payload)
+    {
+        reset(streamId, payload);
+        type = Type.RESPONSE_COMPLETE;
+    }
+
     public void resetForRequestChannel(final int streamId, final Payload payload, final int initialRequestN)
     {
         reset(streamId, payload);
         type = Type.REQUEST_CHANNEL;
         this.initialRequestN = initialRequestN;
     }
 
+    public static boolean requiresFragmenting(final int metadataMtu, final int dataMtu, final Payload payload)
+    {
+        final ByteBuffer metadata = payload.getMetadata();
+        final ByteBuffer data = payload.getData();
+
+        return (metadata.remaining() > metadataMtu || data.remaining() > dataMtu);
+    }
+
+    public Iterator<Frame> iterator()
+    {
+        return this;
+    }
+
     public boolean hasNext()
     {
         return (dataOffset < data.capacity() || metadataOffset < metadata.capacity());
@@ -96,6 +115,15 @@ public Frame next()
 
             result = Frame.Response.from(streamId, FrameType.NEXT, metadataBuffer, dataBuffer, flags);
         }
+        if (Type.RESPONSE_COMPLETE == type)
+        {
+            if (isMoreFollowing)
+            {
+                flags |= FrameHeaderFlyweight.FLAGS_RESPONSE_F;
+            }
+
+            result = Frame.Response.from(streamId, FrameType.NEXT_COMPLETE, metadataBuffer, dataBuffer, flags);
+        }
         else if (Type.REQUEST_CHANNEL == type)
         {
             if (isMoreFollowing)