initial commit

pkg/watermill/azsb/azbus.go

@@ -0,0 +1,80 @@
+package azsb
+
+import (
+	"context"
+	"fmt"
+	"sync"
+
+	"github.com/Azure/azure-sdk-for-go/sdk/azidentity"
+	"github.com/Azure/azure-sdk-for-go/sdk/messaging/azservicebus"
+	"github.com/ThreeDotsLabs/watermill/message"
+	"github.com/rs/zerolog"
+)
+
+type AzBus struct {
+	client      *azservicebus.Client
+	logger      zerolog.Logger
+	closed      bool
+	closedMutex sync.RWMutex
+}
+
+type Config struct {
+	ConnectionString   string
+	UseManagedIdentity bool
+	Namespace          string
+}
+
+// NewAzBus creates a new Azure Service Bus publisher and subscriber
+func NewAzBus(cfg Config, logger zerolog.Logger) (message.Subscriber, message.Publisher, error) {
+	var client *azservicebus.Client
+	var err error
+
+	if cfg.UseManagedIdentity {
+		// Use managed identity
+		if cfg.Namespace == "" {
+			return nil, nil, fmt.Errorf("azure service bus namespace is required when using managed identity")
+		}
+
+		credential, credErr := azidentity.NewDefaultAzureCredential(nil)
+		if credErr != nil {
+			return nil, nil, fmt.Errorf("failed to create azure credential: %w", credErr)
+		}
+
+		namespace := cfg.Namespace
+		client, err = azservicebus.NewClient(namespace, credential, nil)
+		if err != nil {
+			return nil, nil, fmt.Errorf("failed to create azure service bus client: %w", err)
+		}
+	} else {
+		// Use connection string
+		if cfg.ConnectionString == "" {
+			return nil, nil, fmt.Errorf("azure service bus connection string is not configured")
+		}
+
+		client, err = azservicebus.NewClientFromConnectionString(cfg.ConnectionString, nil)
+		if err != nil {
+			return nil, nil, fmt.Errorf("failed to create azure service bus client: %w", err)
+		}
+	}
+
+	azb := &AzBus{client: client, logger: logger, closed: false, closedMutex: sync.RWMutex{}}
+
+	return azb, azb, nil
+}
+
+func (a *AzBus) Close() error {
+	if a.closed {
+		return nil
+	}
+
+	if a.client != nil {
+		if err := a.client.Close(context.Background()); err != nil {
+			a.logger.Error().Err(err).Msg("failed to close azure service bus client")
+			return err
+		}
+	}
+
+	a.closed = true
+	a.logger.Info().Msg("azure service bus publisher closed")
+	return nil
+}