feat: add client.PublishOption for setting message delivery mode

micro · Oct 22, 2023 · 09bc84c · 09bc84c
1 parent bc05fb0
commit 09bc84c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/v4/broker/rabbitmq/options.go b/v4/broker/rabbitmq/options.go
@@ -5,6 +5,7 @@ import (
 	"time"
 
 	"go-micro.dev/v4/broker"
+	"go-micro.dev/v4/client"
 )
 
 type durableQueueKey struct{}
@@ -164,3 +165,14 @@ type ackSuccessKey struct{}
 func AckOnSuccess() broker.SubscribeOption {
 	return setSubscribeOption(ackSuccessKey{}, true)
 }
+
+// PublishDeliveryMode client.PublishOption for setting message "delivery mode"
+// mode , Transient (0 or 1) or Persistent (2)
+func PublishDeliveryMode(mode uint8) client.PublishOption {
+	return func(o *client.PublishOptions) {
+		if o.Context == nil {
+			o.Context = context.Background()
+		}
+		o.Context = context.WithValue(o.Context, deliveryMode{}, mode)
+	}
+}