(helix) branch master updated: Fix helix-rest memory leak (#2960)

[xyuanlu]

This is an automated email from the ASF dual-hosted git repository.

xyuanlu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/helix.git


The following commit(s) were added to refs/heads/master by this push:
     new 2b54fbb8c Fix helix-rest memory leak (#2960)
2b54fbb8c is described below

commit 2b54fbb8c01dd74cecfebc412d1ea88a2cd7c677
Author: Xiaxuan Gao <xiax...@linkedin.com>
AuthorDate: Fri Nov 8 09:41:43 2024 -0800

    Fix helix-rest memory leak (#2960)
    
    Fix helix-rest stoppable check memory leak
---
 .../helix/rest/client/CustomRestClientImpl.java    |  1 +
 .../helix/rest/client/TestCustomRestClient.java    | 26 ++++++++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git 
a/helix-rest/src/main/java/org/apache/helix/rest/client/CustomRestClientImpl.java
 
b/helix-rest/src/main/java/org/apache/helix/rest/client/CustomRestClientImpl.java
index 09783c8dd..db943b3dd 100644
--- 
a/helix-rest/src/main/java/org/apache/helix/rest/client/CustomRestClientImpl.java
+++ 
b/helix-rest/src/main/java/org/apache/helix/rest/client/CustomRestClientImpl.java
@@ -202,6 +202,7 @@ class CustomRestClientImpl implements CustomRestClient {
             LOG.warn("Received 302 but no Location header is present, stopping 
retries.");
             break;  // Break out if there is no valid redirect location
           }
+          EntityUtils.consumeQuietly(response.getEntity());
         } else {
           LOG.warn("Received non-200 and non-302 status code: {}, payloads: 
{}", status, payloads);
           return response;  // Return response without retry
diff --git 
a/helix-rest/src/test/java/org/apache/helix/rest/client/TestCustomRestClient.java
 
b/helix-rest/src/test/java/org/apache/helix/rest/client/TestCustomRestClient.java
index 5a5fec04b..3d81dc432 100644
--- 
a/helix-rest/src/test/java/org/apache/helix/rest/client/TestCustomRestClient.java
+++ 
b/helix-rest/src/test/java/org/apache/helix/rest/client/TestCustomRestClient.java
@@ -20,10 +20,12 @@ package org.apache.helix.rest.client;
  */
 
 import java.io.IOException;
+import java.time.Duration;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.TimeUnit;
 
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
@@ -36,6 +38,7 @@ import org.apache.http.client.ClientProtocolException;
 import org.apache.http.client.HttpClient;
 import org.apache.http.client.methods.HttpPost;
 import org.apache.http.conn.ConnectTimeoutException;
+import org.apache.http.impl.client.HttpClientBuilder;
 import org.apache.http.impl.client.HttpClients;
 import org.junit.Assert;
 import org.mockito.Mock;
@@ -169,6 +172,29 @@ public class TestCustomRestClient {
     }
   }
 
+  @Test (description = "Validate if the post request has memory leak or no")
+  public void testMultiplePost() throws IOException {
+    // a popular echo server that echos all the inputs
+    final String echoServer = 
"https://httpbin.org/redirect-to?url=http://httpbin.org/post";;
+    HttpClientBuilder httpClientBuilder = HttpClients.custom()
+        .evictExpiredConnections()
+        .setMaxConnPerRoute(1)
+        .evictIdleConnections(Duration.ofSeconds(30).toMillis(), 
TimeUnit.MILLISECONDS);
+    HttpClient httpClient = httpClientBuilder.build();
+    CustomRestClientImpl customRestClient = new 
CustomRestClientImpl(httpClient);
+    HttpResponse response;
+
+    for (int i = 0; i < 5; i++) {
+      response = customRestClient.post(echoServer, Collections.emptyMap());
+      if (response.getStatusLine().getStatusCode() == HttpStatus.SC_OK) {
+        JsonNode json = customRestClient.getJsonObject(response);
+
+        Assert.assertEquals(json.get("headers").get("Accept").asText(), 
"application/json");
+        Assert.assertEquals(json.get("data").asText(), "{}");
+      }
+    }
+  }
+
   @Test
   public void testGetPartitionStoppableCheckWhenTimeout() throws IOException {
     MockCustomRestClient customRestClient = new 
MockCustomRestClient(_httpClient);