Apply .gitignore rules

.svn/pristine/33/335a13f982390c2ec9a716fcddcea8bc20697680.svn-base

@@ -0,0 +1,183 @@
+package mqMgr
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	amqp "github.com/rabbitmq/amqp091-go"
+)
+
+// Queue对象
+type HSYQueue struct {
+	conn    *amqp.Connection
+	channel *amqp.Channel
+	// 队列名称
+	queueName string
+	// 交换机
+	exchange string
+	// routing Key
+	routingKey string
+	//MQ链接字符串
+	mqurl string
+}
+
+// 消息队列配置对象
+type HSYQueueConfig struct {
+	QueueName  string
+	Exchange   string
+	RoutingKey string
+	Network    string
+	Port       int
+	UserName   string
+	Password   string
+}
+
+// 创建新的Queue对象
+func NewHSYQueue(queueName, exchange, routingKey, userName, password string, network string, port int) *HSYQueue {
+	queueConfigObj := &HSYQueueConfig{
+		QueueName:  queueName,
+		Exchange:   exchange,
+		RoutingKey: routingKey,
+		UserName:   userName,
+		Password:   password,
+		Network:    network,
+		Port:       port,
+	}
+
+	return NewHSYQueueByConfig(queueConfigObj)
+}
+
+// 通过队列配置对象创建新的Queue对象
+func NewHSYQueueByConfig(queueConfigObj *HSYQueueConfig) *HSYQueue {
+	queueObj := &HSYQueue{
+		queueName:  queueConfigObj.QueueName,
+		exchange:   queueConfigObj.Exchange,
+		routingKey: queueConfigObj.RoutingKey,
+		mqurl:      fmt.Sprintf("amqp://%s:%s@%s:%d/", queueConfigObj.UserName, queueConfigObj.Password, queueConfigObj.Network, queueConfigObj.Port),
+	}
+
+	if err := queueObj.GetQueueAttributes(); err != nil {
+		return nil
+	}
+
+	return queueObj
+}
+
+// 连接RabbitMQ服务器
+func (this *HSYQueue) GetQueueAttributes() (err error) {
+	this.conn, err = amqp.Dial(this.mqurl)
+	if err != nil {
+		return
+	}
+
+	this.channel, err = this.conn.Channel()
+	if err != nil {
+		return
+	}
+
+	return
+}
+
+// 释放连接
+func (this *HSYQueue) ReleaseRes() {
+	this.conn.Close()
+	this.channel.Close()
+}
+
+// SendMessage 发送单条消息
+func (this *HSYQueue) SendMessage(message string, ex string) (err error) {
+	// 声明队列
+	_, err = this.channel.QueueDeclare(
+		this.queueName, // 队列名
+		true,           // 是否持久化
+		false,          // 是否自动删除(前提是至少有一个消费者连接到这个队列，之后所有与这个队列连接的消费者都断开时，才会自动删除。注意：生产者客户端创建这个队列，或者没有消费者客户端与这个队列连接时，都不会自动删除这个队列)
+		false,          // 是否为排他队列（排他的队列仅对“首次”声明的conn可见[一个conn中的其他channel也能访问该队列]，conn结束后队列删除）
+		false,          // 是否阻塞
+		nil,            // 额外属性
+	)
+	if err != nil {
+		err = errors.New("声明队列失败")
+		return
+	}
+
+	// 声明交换器
+	err = this.channel.ExchangeDeclare(
+		this.exchange, //交换器名
+		ex,            //exchange type：一般用fanout、direct、topic
+		true,          // 是否持久化
+		false,         //是否自动删除（自动删除的前提是至少有一个队列或者交换器与这和交换器绑定，之后所有与这个交换器绑定的队列或者交换器都与此解绑）
+		false,         //设置是否内置的。true表示是内置的交换器，客户端程序无法直接发送消息到这个交换器中，只能通过交换器路由到交换器这种方式
+		false,         // 是否阻塞
+		nil,           // 额外属性
+	)
+	if err != nil {
+		err = errors.New("声明交换器失败")
+		return
+	}
+
+	// Binding
+	err = this.channel.QueueBind(
+		this.queueName,  // 绑定的队列名称
+		this.routingKey, // bindkey 用于消息路由分发的key
+		this.exchange,   // 绑定的exchange名
+		false,           // 是否阻塞
+		nil,             // 额外属性
+	)
+	if err != nil {
+		err = errors.New("绑定队列和交换器失败")
+		return
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	err = this.channel.PublishWithContext(ctx,
+		this.exchange,
+		this.routingKey,
+		false,
+		false,
+		amqp.Publishing{
+			ContentType: "text/plain",
+			Body:        []byte(message),
+		})
+	if err != nil {
+		err = errors.New("发布消息失败")
+		return
+	}
+
+	return
+}
+
+// ReceiveMessage 消费单条消息
+func (this *HSYQueue) ReceiveMessage() (msgs <-chan amqp.Delivery, err error) {
+	_, err = this.channel.QueueDeclare(
+		this.queueName, // 队列名
+		true,           // 是否持久化
+		false,          // 是否自动删除(前提是至少有一个消费者连接到这个队列，之后所有与这个队列连接的消费者都断开时，才会自动删除。注意：生产者客户端创建这个队列，或者没有消费者客户端与这个队列连接时，都不会自动删除这个队列)
+		false,          // 是否为排他队列（排他的队列仅对“首次”声明的conn可见[一个conn中的其他channel也能访问该队列]，conn结束后队列删除）
+		false,          // 是否阻塞
+		nil,            // 额外属性（我还不会用）
+	)
+	if err != nil {
+		err = errors.New("声明队列失败")
+		return
+	}
+
+	msgs, err = this.channel.Consume(
+		this.queueName, // 队列名
+		"",             // 消费者名，用来区分多个消费者，以实现公平分发或均等分发策略
+		true,           // 是否自动应答
+		false,          // 是否排他
+		false,          // 是否接收只同一个连接中的消息，若为true，则只能接收别的conn中发送的消息
+		true,           // 队列消费是否阻塞
+		nil,            // 额外属性
+	)
+	if err != nil {
+		err = errors.New("获取消息失败")
+		return
+	}
+
+	return
+}