[Fix][Connector-V2] Fix starrocks Content-Length header already prese…

…nt error (#7034)
apache · Jun 26, 2024 · a485a74 · a485a74
1 parent c290aac
commit a485a74
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/.../src/main/java/org/apache/seatunnel/connectors/seatunnel/starrocks/client/HttpHelper.java b/.../src/main/java/org/apache/seatunnel/connectors/seatunnel/starrocks/client/HttpHelper.java
@@ -21,6 +21,7 @@
 import org.apache.seatunnel.connectors.seatunnel.starrocks.config.SinkConfig;
 
 import org.apache.http.HttpEntity;
+import org.apache.http.HttpRequestInterceptor;
 import org.apache.http.HttpStatus;
 import org.apache.http.client.config.RequestConfig;
 import org.apache.http.client.methods.CloseableHttpResponse;
@@ -32,6 +33,7 @@
 import org.apache.http.impl.client.DefaultRedirectStrategy;
 import org.apache.http.impl.client.HttpClientBuilder;
 import org.apache.http.impl.client.HttpClients;
+import org.apache.http.protocol.HTTP;
 import org.apache.http.util.EntityUtils;
 
 import lombok.extern.slf4j.Slf4j;
@@ -127,6 +129,14 @@ public Map<String, Object> doHttpPut(String url, byte[] data, Map<String, String
             throws IOException {
         final HttpClientBuilder httpClientBuilder =
                 HttpClients.custom()
+                        .addInterceptorFirst(
+                                (HttpRequestInterceptor)
+                                        (request, context) -> {
+                                            // fighting org.apache.http.protocol.RequestContent's
+                                            // ProtocolException("Content-Length header already
+                                            // present");
+                                            request.removeHeaders(HTTP.CONTENT_LEN);
+                                        })
                         .setRedirectStrategy(
                                 new DefaultRedirectStrategy() {
                                     @Override