Merge pull request #3919 from jooby-project/3918

fix(undertow): resolve buffer already freed exception during shutdown

Author: jknack

modules/jooby-undertow/src/main/java/io/jooby/internal/undertow/UndertowGrpcExchange.java

@@ -11,6 +11,7 @@
 import java.util.Map;
 import java.util.function.Consumer;
 
+import org.xnio.IoUtils;
 import org.xnio.channels.StreamSinkChannel;
 
 import io.jooby.rpc.grpc.GrpcExchange;
@@ -144,19 +145,19 @@ public void close(int statusCode, String description) {
                     try {
                       if (ch.flush()) {
                         ch.suspendWrites();
-                        exchange.endExchange();
+                        endExchange();
                       }
                     } catch (IOException ignored) {
                       ch.suspendWrites();
-                      exchange.endExchange();
+                      endExchange();
                     }
                   });
           responseChannel.resumeWrites();
         } else {
-          exchange.endExchange();
+          endExchange();
         }
       } catch (IOException e) {
-        exchange.endExchange();
+        endExchange();
       }
 
     } else {
@@ -170,4 +171,9 @@ public void close(int statusCode, String description) {
       exchange.endExchange();
     }
   }
+
+  private void endExchange() {
+    IoUtils.safeClose(responseChannel);
+    IoUtils.safeClose(exchange.getRequestChannel());
+  }
 }

modules/jooby-undertow/src/main/java/io/jooby/internal/undertow/UndertowGrpcInputBridge.java

@@ -55,7 +55,6 @@ public void request(long n) {
   public void cancel() {
     demand.set(0);
     IoUtils.safeClose(channel);
-    exchange.endExchange();
   }
 
   @Override
@@ -90,7 +89,6 @@ public void handleEvent(StreamSourceChannel channel) {
     } catch (Throwable t) {
       subscriber.onError(t);
       IoUtils.safeClose(channel);
-      exchange.endExchange();
     }
   }
 }