fix http ERR_CONTENT_LENGTH_MISMATCH. #2916

alibaba · Sep 25, 2024 · 58237ce · 58237ce
1 parent 66d6712
commit 58237ce
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/core/src/main/java/com/taobao/arthas/core/shell/term/impl/http/DirectoryBrowser.java b/core/src/main/java/com/taobao/arthas/core/shell/term/impl/http/DirectoryBrowser.java
@@ -186,9 +186,15 @@ public static DefaultFullHttpResponse directView(File dir, String path, FullHttp
                     } finally {
                         IOUtils.close(fileInputStream);
                     }
-                    ctx.write(fullResp);
-                    ChannelFuture future = ctx.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT);
-                    future.addListener(ChannelFutureListener.CLOSE);
+                    ChannelFuture channelFuture = ctx.writeAndFlush(fullResp);
+                    channelFuture.addListener((ChannelFutureListener) future -> {
+                        if (future.isSuccess()) {
+                            ChannelFuture lastContentFuture = ctx.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT);
+                            lastContentFuture.addListener(ChannelFutureListener.CLOSE);
+                        } else {
+                            future.channel().close();
+                        }
+                    });
                     return fullResp;
                 }
                 logger.info("file {} size bigger than {}, send by future.",file.getName(), MIN_NETTY_DIRECT_SEND_SIZE);