SNOW-1357377 Add request Id in all streaming ingest APIs

snowflakedb · May 1, 2024 · 8c5dc97 · 8c5dc97
1 parent eb87281
commit 8c5dc97
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/main/java/net/snowflake/ingest/connection/RequestBuilder.java b/src/main/java/net/snowflake/ingest/connection/RequestBuilder.java
@@ -683,7 +683,13 @@ public HttpPost generateStreamingIngestPostRequest(
     URI uri = null;
     try {
       uri =
-          new URIBuilder().setScheme(scheme).setHost(host).setPort(port).setPath(endPoint).build();
+          new URIBuilder()
+                  .setScheme(scheme)
+                  .setHost(host)
+                  .setPort(port)
+                  .setPath(endPoint)
+                  .setParameter(REQUEST_ID, UUID.randomUUID().toString())
+                  .build();
     } catch (URISyntaxException e) {
       throw new SFException(e, ErrorCode.BUILD_REQUEST_FAILURE, message);
     }

diff --git a/.../java/net/snowflake/ingest/streaming/internal/SnowflakeStreamingIngestClientInternal.java b/.../java/net/snowflake/ingest/streaming/internal/SnowflakeStreamingIngestClientInternal.java
@@ -493,7 +493,6 @@ ChannelsStatusResponse getChannelsStatus(
               .collect(Collectors.toList());
       request.setChannels(requestDTOs);
       request.setRole(this.role);
-      request.setRequestId(this.flushService.getClientPrefix() + "_" + counter.getAndIncrement());
 
       String payload = objectMapper.writeValueAsString(request);