fix StreamBufferingEncoder

grpc · Mar 26, 2021 · 95060b2 · 95060b2
1 parent f545a2a
commit 95060b2
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/netty/src/main/java/io/grpc/netty/StreamBufferingEncoder.java b/netty/src/main/java/io/grpc/netty/StreamBufferingEncoder.java
@@ -63,6 +63,8 @@
  * drop-in decorator of {@link io.netty.handler.codec.http2.DefaultHttp2ConnectionEncoder}.
  * </p>
  */
+// This is a temporary copy of {@link io.netty.handler.codec.http2.DecoratingHttp2ConnectionEncoder}
+// with a bug fix that is not available yet in the latest netty release.
 class StreamBufferingEncoder extends DecoratingHttp2ConnectionEncoder {
 
   /**
@@ -156,11 +158,7 @@ public ChannelFuture writeHeaders(ChannelHandlerContext ctx, int streamId, Http2
     if (closed) {
       return promise.setFailure(new Http2ChannelClosedException());
     }
-    if (isExistingStream(streamId) || connection().goAwayReceived()) {
-      return super.writeHeaders(ctx, streamId, headers, streamDependency, weight,
-          exclusive, padding, endOfStream, promise);
-    }
-    if (canCreateStream()) {
+    if (isExistingStream(streamId) || canCreateStream()) {
       return super.writeHeaders(ctx, streamId, headers, streamDependency, weight,
           exclusive, padding, endOfStream, promise);
     }