diff --git a/src/main/java/com/coveo/pushapiclient/StreamService.java b/src/main/java/com/coveo/pushapiclient/StreamService.java index a536ded4..51ec841b 100644 --- a/src/main/java/com/coveo/pushapiclient/StreamService.java +++ b/src/main/java/com/coveo/pushapiclient/StreamService.java @@ -69,7 +69,7 @@ public StreamService(StreamEnabledSource source) { * @throws IOException */ public void add(DocumentBuilder document) throws IOException, InterruptedException { - this.service.add(document); + this.streamId = this.service.add(document); } /** diff --git a/src/main/java/com/coveo/pushapiclient/StreamServiceInternal.java b/src/main/java/com/coveo/pushapiclient/StreamServiceInternal.java index b0145504..e5579752 100644 --- a/src/main/java/com/coveo/pushapiclient/StreamServiceInternal.java +++ b/src/main/java/com/coveo/pushapiclient/StreamServiceInternal.java @@ -19,11 +19,12 @@ public StreamServiceInternal( this.platformClient = platformClient; } - public void add(DocumentBuilder document) throws IOException, InterruptedException { + public String add(DocumentBuilder document) throws IOException, InterruptedException { if (this.streamId == null) { this.streamId = this.getStreamId(); } queue.add(document); + return this.streamId; } public HttpResponse close()