googleapis · diegomarquezp · Dec 19, 2025 · Dec 1, 2025 · Dec 1, 2025 · Dec 1, 2025
@@ -48,10 +48,23 @@ public void addHeader(String name, String value) {
     request.addHeader(name, value);
   }
 
+  /**
+   * Configures the connection, connection request and read timeout of the request.
+   *
+   * @param connectTimeout timeout in milliseconds to establish a connection or {@code 0} for an
+   *     infinite timeout. This parameter will also set the underlying request configuration's
+   *     connection request timeout. See {@link
+   *     org.apache.hc.client5.http.config.RequestConfig.Builder#setConnectionRequestTimeout(long,
+   *     TimeUnit)} for more details.
+   * @param readTimeout Timeout in milliseconds to read data from an established connection or
+   *     {@code 0} for an infinite timeout
+   * @throws IOException
+   */
   @Override
   public void setTimeout(int connectTimeout, int readTimeout) throws IOException {
     requestConfig
         .setConnectTimeout(Timeout.of(connectTimeout, TimeUnit.MILLISECONDS))
+        .setConnectionRequestTimeout(connectTimeout, TimeUnit.MILLISECONDS)
         // ResponseTimeout behaves the same as 4.x's SocketTimeout
         .setResponseTimeout(Timeout.of(readTimeout, TimeUnit.MILLISECONDS));
   }

@@ -26,16 +26,19 @@
 import java.io.InputStream;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
+import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
 import org.apache.hc.client5.http.classic.methods.HttpPost;
 import org.apache.hc.client5.http.classic.methods.HttpUriRequestBase;
+import org.apache.hc.client5.http.config.RequestConfig;
 import org.apache.hc.core5.http.ClassicHttpRequest;
 import org.apache.hc.core5.http.ClassicHttpResponse;
 import org.apache.hc.core5.http.ContentType;
 import org.apache.hc.core5.http.HttpEntity;
 import org.apache.hc.core5.http.HttpHost;
 import org.apache.hc.core5.http.io.entity.BasicHttpEntity;
 import org.apache.hc.core5.http.protocol.HttpContext;
+import org.apache.hc.core5.util.Timeout;
 import org.junit.Test;
 
 public class Apache5HttpRequestTest {
@@ -128,4 +131,26 @@ public ClassicHttpResponse executeOpen(
     response.getContent().close();
     assertEquals(1, closedResponseCounter.get());
   }
+
+  @Test
+  public void testSetTimeout() throws Exception {
+    HttpUriRequestBase base = new HttpPost("http://www.google.com");
+    Apache5HttpRequest request =
+        new Apache5HttpRequest(
+            new MockHttpClient() {
+              @Override
+              public ClassicHttpResponse executeOpen(
+                  HttpHost target, ClassicHttpRequest request, HttpContext context) {
+                return new MockClassicHttpResponse();
+              }
+            },
+            base);
+    request.setTimeout(100, 200);
+    request.execute();
+
+    RequestConfig config = base.getConfig();
+    assertEquals(Timeout.of(100, TimeUnit.MILLISECONDS), config.getConnectTimeout());
+    assertEquals(Timeout.of(200, TimeUnit.MILLISECONDS), config.getResponseTimeout());
+    assertEquals(Timeout.of(100, TimeUnit.MILLISECONDS), config.getConnectionRequestTimeout());
+  }
 }