Merge pull request #116 from kaytu-io/fix-query-map-param

fix: receive sequence number after producing message
opengovern · Sep 13, 2024 · 8c4a455 · 8c4a455
2 parents 4519f2a + 43aeeb6
commit 8c4a455
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pkg/jq/nats.go b/pkg/jq/nats.go
@@ -139,10 +139,11 @@ func (jq *JobQueue) ConsumeWithConfig(
 	return consumeCtx, nil
 }
 
-func (jq *JobQueue) Produce(ctx context.Context, topic string, data []byte, id string) error {
-	if _, err := jq.js.Publish(ctx, topic, data, jetstream.WithMsgID(id)); err != nil {
-		return err
+func (jq *JobQueue) Produce(ctx context.Context, topic string, data []byte, id string) (*uint64, error) {
+	pubAck, err := jq.js.Publish(ctx, topic, data, jetstream.WithMsgID(id))
+	if err != nil {
+		return nil, err
 	}
 
-	return nil
+	return &pubAck.Sequence, nil
 }