From 80f92f90cbc9a8267cad3f338641982b53478e8a Mon Sep 17 00:00:00 2001
From: Shawn Silverman <ssilverman@twitter.com>
Date: Fri, 19 Jul 2013 13:03:18 -0700
Subject: [PATCH] netty-1616: Adding a 'last' check to SpdyHttpEncoder.

---
 .../handler/codec/spdy/SpdyHttpEncoder.java   | 27 +++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/codec-http/src/main/java/io/netty/handler/codec/spdy/SpdyHttpEncoder.java b/codec-http/src/main/java/io/netty/handler/codec/spdy/SpdyHttpEncoder.java
index 3862e2aedb..a8e3eefd26 100644
--- a/codec-http/src/main/java/io/netty/handler/codec/spdy/SpdyHttpEncoder.java
+++ b/codec-http/src/main/java/io/netty/handler/codec/spdy/SpdyHttpEncoder.java
@@ -18,6 +18,7 @@ package io.netty.handler.codec.spdy;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.handler.codec.MessageToMessageEncoder;
 import io.netty.handler.codec.UnsupportedMessageTypeException;
+import io.netty.handler.codec.http.FullHttpMessage;
 import io.netty.handler.codec.http.FullHttpRequest;
 import io.netty.handler.codec.http.HttpContent;
 import io.netty.handler.codec.http.HttpHeaders;
@@ -141,6 +142,7 @@ public class SpdyHttpEncoder extends MessageToMessageEncoder<HttpObject> {
     protected void encode(ChannelHandlerContext ctx, HttpObject msg, List<Object> out) throws Exception {
 
         boolean valid = false;
+        boolean last = false;
 
         if (msg instanceof HttpRequest) {
 
@@ -148,6 +150,7 @@ public class SpdyHttpEncoder extends MessageToMessageEncoder<HttpObject> {
             SpdySynStreamFrame spdySynStreamFrame = createSynStreamFrame(httpRequest);
             out.add(spdySynStreamFrame);
 
+            last = spdySynStreamFrame.isLast();
             valid = true;
         }
         if (msg instanceof HttpResponse) {
@@ -155,15 +158,17 @@ public class SpdyHttpEncoder extends MessageToMessageEncoder<HttpObject> {
             HttpResponse httpResponse = (HttpResponse) msg;
             if (httpResponse.headers().contains(SpdyHttpHeaders.Names.ASSOCIATED_TO_STREAM_ID)) {
                 SpdySynStreamFrame spdySynStreamFrame = createSynStreamFrame(httpResponse);
+                last = spdySynStreamFrame.isLast();
                 out.add(spdySynStreamFrame);
             } else {
                 SpdySynReplyFrame spdySynReplyFrame = createSynReplyFrame(httpResponse);
+                last = spdySynReplyFrame.isLast();
                 out.add(spdySynReplyFrame);
             }
 
             valid = true;
         }
-        if (msg instanceof HttpContent) {
+        if (msg instanceof HttpContent && !last) {
 
             HttpContent chunk = (HttpContent) msg;
 
@@ -255,6 +260,7 @@ public class SpdyHttpEncoder extends MessageToMessageEncoder<HttpObject> {
             spdySynStreamFrame.headers().add(entry.getKey(), entry.getValue());
         }
         currentStreamId = spdySynStreamFrame.getStreamId();
+        spdySynStreamFrame.setLast(isLast(httpMessage));
 
         return spdySynStreamFrame;
     }
@@ -284,8 +290,25 @@ public class SpdyHttpEncoder extends MessageToMessageEncoder<HttpObject> {
         }
 
         currentStreamId = streamID;
-        spdySynReplyFrame.setLast(false);
+        spdySynReplyFrame.setLast(isLast(httpResponse));
 
         return spdySynReplyFrame;
     }
+
+    /**
+     * Checks if the given HTTP message should be considered as a last SPDY frame.
+     *
+     * @param httpMessage check this HTTP message
+     * @return whether the given HTTP message should generate a <em>last</em> SPDY frame.
+     */
+    private static boolean isLast(HttpMessage httpMessage) {
+        if (httpMessage instanceof FullHttpMessage) {
+            FullHttpMessage fullMessage = (FullHttpMessage) httpMessage;
+            if (fullMessage.trailingHeaders().isEmpty() && !fullMessage.content().isReadable()) {
+                return true;
+            }
+        }
+
+        return false;
+    }
 }