go-db/factory/rabbitmq/rabbitmq_factory.go

// factory/rabbitmq_factory.go
package rabbitmq

import (
	"fmt"
	"sync"

	"git.x2erp.com/qdy/go-base/config"
	amqp "github.com/streadway/amqp"
)

// RabbitMQFactory RabbitMQ工厂
type RabbitMQFactory struct {
	mu      sync.RWMutex
	clients map[string]*amqp.Channel
	conn    *amqp.Connection
	config  config.RabbitMQConfig
}

// NewRabbitMQFactory 创建RabbitMQ工厂
func NewRabbitMQFactory() (*RabbitMQFactory, error) {
	cfg := config.GetConfig()

	if err := config.GetInitError(); err != nil {
		return nil, fmt.Errorf("failed to load config: %v", err)
	}

	rabbitConfig := cfg.GetRabbitMQ()
	if rabbitConfig.Host == "" || rabbitConfig.Port == 0 {
		return nil, fmt.Errorf("rabbitmq configuration is incomplete")
	}

	// 构建连接URL
	url := fmt.Sprintf("amqp://%s:%s@%s:%d/%s",
		rabbitConfig.Username,
		rabbitConfig.Password,
		rabbitConfig.Host,
		rabbitConfig.Port,
		rabbitConfig.Vhost)

	// 建立连接
	conn, err := amqp.Dial(url)
	if err != nil {
		return nil, fmt.Errorf("failed to connect to rabbitmq: %v", err)
	}

	return &RabbitMQFactory{
		conn:    conn,
		config:  rabbitConfig,
		clients: make(map[string]*amqp.Channel),
	}, nil
}

// CreateChannel 创建消息通道
func (f *RabbitMQFactory) CreateChannel(name string) (*amqp.Channel, error) {
	f.mu.Lock()
	defer f.mu.Unlock()

	// 如果已存在，直接返回
	if channel, ok := f.clients[name]; ok && channel != nil {
		return channel, nil
	}

	// 创建新通道
	channel, err := f.conn.Channel()
	if err != nil {
		return nil, fmt.Errorf("failed to create channel '%s': %v", name, err)
	}

	// 设置QoS（可选）
	if f.config.PrefetchCount > 0 {
		err = channel.Qos(
			f.config.PrefetchCount,
			f.config.PrefetchSize,
			f.config.Global,
		)
		if err != nil {
			channel.Close()
			return nil, fmt.Errorf("failed to set qos for channel '%s': %v", name, err)
		}
	}

	f.clients[name] = channel
	return channel, nil
}

// GetChannel 获取已创建的通道
func (f *RabbitMQFactory) GetChannel(name string) (*amqp.Channel, error) {
	f.mu.RLock()
	defer f.mu.RUnlock()

	channel, ok := f.clients[name]
	if !ok || channel == nil {
		return nil, fmt.Errorf("channel '%s' not found", name)
	}
	return channel, nil
}

// CloseChannel 关闭指定通道
func (f *RabbitMQFactory) CloseChannel(name string) error {
	f.mu.Lock()
	defer f.mu.Unlock()

	channel, ok := f.clients[name]
	if ok && channel != nil {
		if err := channel.Close(); err != nil {
			return fmt.Errorf("failed to close channel '%s': %v", name, err)
		}
		delete(f.clients, name)
	}
	return nil
}

// AddExchange 添加交换机
func (f *RabbitMQFactory) AddExchange(channelName, exchange, exchangeType string, durable bool) error {
	channel, err := f.GetChannel(channelName)
	if err != nil {
		return err
	}

	return channel.ExchangeDeclare(
		exchange,
		exchangeType, // "direct", "fanout", "topic", "headers"
		durable,      // durable
		false,        // auto-deleted
		false,        // internal
		false,        // no-wait
		nil,          // arguments
	)
}

// AddQueue 添加队列
func (f *RabbitMQFactory) AddQueue(channelName, queue string, durable, exclusive, autoDelete bool) error {
	channel, err := f.GetChannel(channelName)
	if err != nil {
		return err
	}

	_, err = channel.QueueDeclare(
		queue,
		durable,    // durable
		autoDelete, // delete when unused
		exclusive,  // exclusive
		false,      // no-wait
		nil,        // arguments
	)
	return err
}

// BindQueue 绑定队列到交换机
func (f *RabbitMQFactory) BindQueue(channelName, queue, exchange, routingKey string) error {
	channel, err := f.GetChannel(channelName)
	if err != nil {
		return err
	}

	return channel.QueueBind(
		queue,
		routingKey,
		exchange,
		false, // no-wait
		nil,   // arguments
	)
}

// Close 关闭所有连接
func (f *RabbitMQFactory) Close() error {
	f.mu.Lock()
	defer f.mu.Unlock()

	// 关闭所有通道
	var errs []error
	for name, channel := range f.clients {
		if channel != nil {
			if err := channel.Close(); err != nil {
				errs = append(errs, fmt.Errorf("close channel '%s' error: %v", name, err))
			}
		}
	}
	f.clients = make(map[string]*amqp.Channel)

	// 关闭连接
	if f.conn != nil {
		if err := f.conn.Close(); err != nil {
			errs = append(errs, fmt.Errorf("close connection error: %v", err))
		}
	}

	if len(errs) > 0 {
		return fmt.Errorf("errors closing rabbitmq: %v", errs)
	}
	return nil
}