Check stream message is confirmed

acogoluegnes · acogoluegnes · commit 2cf71a3d70df · 2024-07-09T11:03:48.000+02:00
diff --git a/go-stream/offset_tracking_receive.go b/go-stream/offset_tracking_receive.go
@@ -40,7 +40,7 @@ func main() {
 		if string(message.GetData()) == "marker" {
 			lastOffset.Store(consumerContext.Consumer.GetOffset())
 			_ = consumerContext.Consumer.StoreOffset()
-            _ = consumerContext.Consumer.Close()
+			_ = consumerContext.Consumer.Close()
 			ch <- true
 		}
 	}
diff --git a/go-stream/offset_tracking_send.go b/go-stream/offset_tracking_send.go
@@ -54,10 +54,12 @@ func handlePublishConfirm(confirms stream.ChannelPublishConfirm, messageCount in
 	go func() {
 		confirmedCount := 0
 		for confirmed := range confirms {
-			for _, _ = range confirmed {
-				confirmedCount++
-				if confirmedCount == messageCount {
-					ch <- true
+			for _, msg := range confirmed {
+				if msg.IsConfirmed() {
+					confirmedCount++
+					if confirmedCount == messageCount {
+						ch <- true
+					}
 				}
 			}
 		}
diff --git a/java-stream-mvn/src/main/java/OffsetTrackingSend.java b/java-stream-mvn/src/main/java/OffsetTrackingSend.java
@@ -23,7 +23,11 @@ public static void main(String[] args) throws InterruptedException {
       IntStream.range(0, messageCount).forEach(i -> {
         String body = i == messageCount - 1 ? "marker" : "hello";
         producer.send(producer.messageBuilder().addData(body.getBytes(UTF_8)).build(),
-            ctx -> confirmedLatch.countDown());
+            ctx -> {
+              if (ctx.isConfirmed()) {
+                confirmedLatch.countDown();
+              }
+            });
       });
 
       boolean completed = confirmedLatch.await(60, TimeUnit.SECONDS);

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@ func main() {`
`40`	`40`	`if string(message.GetData()) == "marker" {`
`41`	`41`	`lastOffset.Store(consumerContext.Consumer.GetOffset())`
`42`	`42`	`_ = consumerContext.Consumer.StoreOffset()`
`43`		`- _ = consumerContext.Consumer.Close()`
	`43`	`+ _ = consumerContext.Consumer.Close()`
`44`	`44`	`ch <- true`
`45`	`45`	`}`
`46`	`46`	`}`
Original file line number	Diff line number	Diff line change
`@@ -54,10 +54,12 @@ func handlePublishConfirm(confirms stream.ChannelPublishConfirm, messageCount in`
`54`	`54`	`go func() {`
`55`	`55`	`confirmedCount := 0`
`56`	`56`	`for confirmed := range confirms {`
`57`		`- for _, _ = range confirmed {`
`58`		`- confirmedCount++`
`59`		`- if confirmedCount == messageCount {`
`60`		`- ch <- true`
	`57`	`+ for _, msg := range confirmed {`
	`58`	`+ if msg.IsConfirmed() {`
	`59`	`+ confirmedCount++`
	`60`	`+ if confirmedCount == messageCount {`
	`61`	`+ ch <- true`
	`62`	`+ }`
`61`	`63`	`}`
`62`	`64`	`}`
`63`	`65`	`}`