49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package rabbitmq_amqp
|
|
|
|
import (
|
|
"context"
|
|
"github.com/Azure/go-amqp"
|
|
)
|
|
|
|
type PublishResult struct {
|
|
Outcome amqp.DeliveryState
|
|
Message *amqp.Message
|
|
}
|
|
|
|
type Publisher struct {
|
|
sender *amqp.Sender
|
|
}
|
|
|
|
func newPublisher(sender *amqp.Sender) *Publisher {
|
|
return &Publisher{sender: sender}
|
|
}
|
|
|
|
// Publish sends a message to the destination address.
|
|
// The message is sent to the destination address and the outcome of the operation is returned.
|
|
// The outcome is a DeliveryState that indicates if the message was accepted or rejected.
|
|
// RabbitMQ supports the following DeliveryState types:
|
|
// - StateAccepted
|
|
// - StateReleased
|
|
// - StateRejected
|
|
// See: https://www.rabbitmq.com/docs/next/amqp#outcomes for more information.
|
|
func (m *Publisher) Publish(ctx context.Context, message *amqp.Message) (*PublishResult, error) {
|
|
r, err := m.sender.SendWithReceipt(ctx, message, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
state, err := r.Wait(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
publishResult := &PublishResult{
|
|
Message: message,
|
|
Outcome: state,
|
|
}
|
|
return publishResult, err
|
|
}
|
|
|
|
// Close closes the publisher.
|
|
func (m *Publisher) Close(ctx context.Context) error {
|
|
return m.sender.Close(ctx)
|
|
}
|