snapshot of https://github.com/ThreeDotsLabs/watermill.git (master, 2026-09-15; blobs > 131072B omitted)

message/subscriber/read.go

@@ -0,0 +1,53 @@
+package subscriber
+
+import (
+	"time"
+
+	"github.com/ThreeDotsLabs/watermill/message"
+)
+
+// BulkRead reads provided amount of messages from the provided channel, until a timeout occurs or the limit is reached.
+func BulkRead(messagesCh <-chan *message.Message, limit int, timeout time.Duration) (receivedMessages message.Messages, all bool) {
+MessagesLoop:
+	for len(receivedMessages) < limit {
+		select {
+		case msg, ok := <-messagesCh:
+			if !ok {
+				break MessagesLoop
+			}
+
+			receivedMessages = append(receivedMessages, msg)
+			msg.Ack()
+		case <-time.After(timeout):
+			break MessagesLoop
+		}
+	}
+
+	return receivedMessages, len(receivedMessages) == limit
+}
+
+// BulkReadWithDeduplication reads provided number of messages from the provided channel, ignoring duplicates,
+// until a timeout occurs or the limit is reached.
+func BulkReadWithDeduplication(messagesCh <-chan *message.Message, limit int, timeout time.Duration) (receivedMessages message.Messages, all bool) {
+	receivedIDs := map[string]struct{}{}
+
+MessagesLoop:
+	for len(receivedMessages) < limit {
+		select {
+		case msg, ok := <-messagesCh:
+			if !ok {
+				break MessagesLoop
+			}
+
+			if _, ok := receivedIDs[msg.UUID]; !ok {
+				receivedIDs[msg.UUID] = struct{}{}
+				receivedMessages = append(receivedMessages, msg)
+			}
+			msg.Ack()
+		case <-time.After(timeout):
+			break MessagesLoop
+		}
+	}
+
+	return receivedMessages, len(receivedMessages) == limit
+}