[fix](stream_load)fix bug when stream without content-length or chunk…

…ed Transfer Encoding
apache · Nov 29, 2023 · 53d2d2a · 53d2d2a
1 parent c30299c
commit 53d2d2a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/be/src/http/action/stream_load.cpp b/be/src/http/action/stream_load.cpp
@@ -301,6 +301,12 @@ Status StreamLoadAction::_on_header(HttpRequest* http_req, std::shared_ptr<Strea
         }
     }
 
+    if (ctx->body_bytes == 0 && !ctx->is_chunked_transfer) {
+        return Status::InternalError(
+                "content_length is empty and transfer-encoding!=chunked, please set content_length "
+                "or transfer-encoding=chunked");
+    }
+
     if (!http_req->header(HTTP_TIMEOUT).empty()) {
         try {
             ctx->timeout_second = std::stoi(http_req->header(HTTP_TIMEOUT));