initial commit

pkg/rabbit/rabbitmq.go

@@ -0,0 +1,103 @@
+package rabbitmq
+
+import (
+	"context"
+	"time"
+
+	amqp "github.com/rabbitmq/amqp091-go"
+)
+
+const Version = "1.0.0"
+
+type Publisher interface {
+	Publish(ctx context.Context, exchange, routingKey string, msg *Message) error
+	Close() error
+}
+
+type Consumer interface {
+	Consume(ctx context.Context) error
+	Close() error
+}
+
+type MessageHandler func(ctx context.Context, msg *Message)
+
+type Client interface {
+	Publisher() Publisher
+	RegisterConsumer(handler MessageHandler, opts *ConsumerOptions) Consumer
+	DeclareQueue(name string, opts QueueOptions) error
+	DeclareExchange(name string, opts ExchangeOptions) error
+	BindQueue(queue, exchange, routingKey string) error
+	DeleteQueue(name string) error
+	DeleteExchange(name string) error
+	HealthCheck() error
+	Close() error
+}
+
+type ConnectionManager interface {
+	GetConnection() (*amqp.Connection, error)
+	GetChannel() (*amqp.Channel, error)
+	ReturnChannel(*amqp.Channel)
+	Close() error
+	IsConnected() bool
+	NotifyConnectionLoss() <-chan *amqp.Error
+}
+
+type QueueOptions struct {
+	Durable    bool
+	AutoDelete bool
+	Exclusive  bool
+	NoWait     bool
+	Args       amqp.Table
+}
+
+type ExchangeOptions struct {
+	Type       string
+	Durable    bool
+	AutoDelete bool
+	Internal   bool
+	NoWait     bool
+	Args       amqp.Table
+}
+
+type ConsumerOptions struct {
+	Queue         string
+	ConsumerTag   string
+	AutoAck       bool
+	Exclusive     bool
+	NoLocal       bool
+	NoWait        bool
+	PrefetchCount int
+	PrefetchSize  int
+	Args          amqp.Table
+	ReconnectWait time.Duration
+}
+
+type PublisherOptions struct {
+	ConfirmMode    bool
+	Mandatory      bool
+	Immediate      bool
+	RetryAttempts  int
+	RetryDelay     time.Duration
+	ConfirmTimeout time.Duration
+	Args           amqp.Table
+}
+
+const (
+	ExchangeTypeDirect  = "direct"
+	ExchangeTypeFanout  = "fanout"
+	ExchangeTypeTopic   = "topic"
+	ExchangeTypeHeaders = "headers"
+)
+
+const (
+	DeliveryModeTransient  = 1
+	DeliveryModePersistent = 2
+)
+
+const (
+	PriorityLowest  = 0
+	PriorityLow     = 64
+	PriorityNormal  = 128
+	PriorityHigh    = 192
+	PriorityHighest = 255
+)