From 63ff79adde9a0f2d8d0e4e78c5b768fb8cc0a983 Mon Sep 17 00:00:00 2001 From: Alex Sorokoumov Date: Wed, 20 Sep 2023 18:00:09 -0700 Subject: [PATCH] WIP: Increase http client's max chunk size --- .../io/confluent/ksql/rest/client/KsqlRestClient.java | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/ksqldb-rest-client/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java b/ksqldb-rest-client/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java index efca8a36f4bb..858caec38a36 100644 --- a/ksqldb-rest-client/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java +++ b/ksqldb-rest-client/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java @@ -56,6 +56,8 @@ public final class KsqlRestClient implements Closeable { static final String CCLOUD_CONNECT_USERNAME_HEADER = "X-Confluent-API-Key"; static final String CCLOUD_CONNECT_PASSWORD_HEADER = "X-Confluent-API-Secret"; + static final int CHUNK_SIZE = 16 * 1024 * 1204; + private final KsqlClient client; private final LocalProperties localProperties; private final AtomicReference serializedConsistencyVector; @@ -113,8 +115,10 @@ public static KsqlRestClient create( creds, ccloudApiKey, (cprops, credz, lprops) -> new KsqlClient(cprops, credz, lprops, - new HttpClientOptions(), - Optional.of(new HttpClientOptions().setProtocolVersion(HttpVersion.HTTP_2))) + new HttpClientOptions().setMaxChunkSize(CHUNK_SIZE), + Optional.of(new HttpClientOptions() + .setProtocolVersion(HttpVersion.HTTP_2) + .setMaxChunkSize(CHUNK_SIZE))) ); }