fix

initalize/rabbitmq.go

@@ -1,9 +1,11 @@
 package initalize
 
 import (
+	"context"
 	"crypto/tls"
 	"errors"
 	"fusenapi/constants"
+	"fusenapi/server/websocket/consumer"
 	"github.com/streadway/amqp"
 	"github.com/zeromicro/go-zero/core/logx"
 	"log"
@@ -82,11 +84,17 @@ func (h *RabbitMqHandle) SendMsg(queueName constants.RABBIT_MQ, message []byte)
 }
 
 // 消费消息
-func (h *RabbitMqHandle) Consume(queueName constants.RABBIT_MQ, handleFunc func(data []byte) error) error {
+func (h *RabbitMqHandle) Consume(ctx context.Context, queueName constants.RABBIT_MQ, handle consumer.MqHandle) error {
 	object, ok := mapMq[queueName]
 	if !ok {
 		return errors.New("unknown queue")
 	}
+	go func() {
+		select {
+		case <-ctx.Done():
+			panic("err ctx deadline")
+		}
+	}()
 	msgs, err := object.ch.Consume(
 		object.queue.Name, // 队列名
 		object.queue.Name, // 消费者名，如果为空，则是随机生成一个
@@ -115,7 +123,7 @@ func (h *RabbitMqHandle) Consume(queueName constants.RABBIT_MQ, handleFunc func(
 				<-limit
 				wait.Done()
 			}()
-			if err = handleFunc(m.Body); err != nil {
+			if err = handle.Run(m.Body); err != nil {
 				logx.Error("failed to deal with MQ message:", string(m.Body))
 				return
 			}