1
0
mirror of https://github.com/go-micro/go-micro.git synced 2025-06-30 22:33:49 +02:00

add rmq message properties (#2177)

Co-authored-by: dtitov <dtitov@might24.ru>
This commit is contained in:
Dmitry Titov
2021-06-08 16:34:47 +07:00
committed by GitHub
parent a1e9b88495
commit 52bb3845f6
2 changed files with 102 additions and 0 deletions

View File

@ -2,6 +2,7 @@ package rabbitmq
import ( import (
"context" "context"
"time"
"github.com/asim/go-micro/v3/broker" "github.com/asim/go-micro/v3/broker"
) )
@ -15,6 +16,16 @@ type exchangeKey struct{}
type requeueOnErrorKey struct{} type requeueOnErrorKey struct{}
type deliveryMode struct{} type deliveryMode struct{}
type priorityKey struct{} type priorityKey struct{}
type contentType struct{}
type contentEncoding struct{}
type correlationID struct{}
type replyTo struct{}
type expiration struct{}
type messageID struct{}
type timestamp struct{}
type typeMsg struct{}
type userID struct{}
type appID struct{}
type externalAuth struct{} type externalAuth struct{}
type durableExchange struct{} type durableExchange struct{}
@ -68,6 +79,56 @@ func Priority(value uint8) broker.PublishOption {
return setPublishOption(priorityKey{}, value) return setPublishOption(priorityKey{}, value)
} }
// ContentType sets a property MIME content type for publishing
func ContentType(value string) broker.PublishOption {
return setPublishOption(contentType{}, value)
}
// ContentEncoding sets a property MIME content encoding for publishing
func ContentEncoding(value string) broker.PublishOption {
return setPublishOption(contentEncoding{}, value)
}
// CorrelationID sets a property correlation ID for publishing
func CorrelationID(value string) broker.PublishOption {
return setPublishOption(correlationID{}, value)
}
// ReplyTo sets a property address to to reply to (ex: RPC) for publishing
func ReplyTo(value string) broker.PublishOption {
return setPublishOption(replyTo{}, value)
}
// Expiration sets a property message expiration spec for publishing
func Expiration(value string) broker.PublishOption {
return setPublishOption(expiration{}, value)
}
// MessageId sets a property message identifier for publishing
func MessageId(value string) broker.PublishOption {
return setPublishOption(messageID{}, value)
}
// Timestamp sets a property message timestamp for publishing
func Timestamp(value time.Time) broker.PublishOption {
return setPublishOption(timestamp{}, value)
}
// TypeMsg sets a property message type name for publishing
func TypeMsg(value string) broker.PublishOption {
return setPublishOption(typeMsg{}, value)
}
// UserID sets a property user id for publishing
func UserID(value string) broker.PublishOption {
return setPublishOption(userID{}, value)
}
// AppID sets a property application id for publishing
func AppID(value string) broker.PublishOption {
return setPublishOption(appID{}, value)
}
func ExternalAuth() broker.Option { func ExternalAuth() broker.Option {
return setBrokerOption(externalAuth{}, ExternalAuthentication{}) return setBrokerOption(externalAuth{}, ExternalAuthentication{})
} }

View File

@ -162,6 +162,47 @@ func (r *rbroker) Publish(topic string, msg *broker.Message, opts ...broker.Publ
if value, ok := options.Context.Value(priorityKey{}).(uint8); ok { if value, ok := options.Context.Value(priorityKey{}).(uint8); ok {
m.Priority = value m.Priority = value
} }
if value, ok := options.Context.Value(contentType{}).(string); ok {
m.ContentType = value
}
if value, ok := options.Context.Value(contentEncoding{}).(string); ok {
m.ContentEncoding = value
}
if value, ok := options.Context.Value(correlationID{}).(string); ok {
m.CorrelationId = value
}
if value, ok := options.Context.Value(replyTo{}).(string); ok {
m.ReplyTo = value
}
if value, ok := options.Context.Value(expiration{}).(string); ok {
m.Expiration = value
}
if value, ok := options.Context.Value(messageID{}).(string); ok {
m.MessageId = value
}
if value, ok := options.Context.Value(timestamp{}).(time.Time); ok {
m.Timestamp = value
}
if value, ok := options.Context.Value(typeMsg{}).(string); ok {
m.Type = value
}
if value, ok := options.Context.Value(userID{}).(string); ok {
m.UserId = value
}
if value, ok := options.Context.Value(appID{}).(string); ok {
m.AppId = value
}
} }
for k, v := range msg.Header { for k, v := range msg.Header {