fix

initalize/rabbitmq.go

@@ -7,6 +7,7 @@ import (
 	"github.com/streadway/amqp"
 	"github.com/zeromicro/go-zero/core/logx"
 	"log"
+	"sync"
 )
 
 type RabbitMqHandle struct {
@@ -100,16 +101,19 @@ func (h *RabbitMqHandle) Consume(queueName constants.RABBIT_MQ, handleFunc func(
 	}
 	//允许20的并发
 	limit := make(chan struct{}, 20)
+	wait := sync.WaitGroup{}
 	defer close(limit)
 	// 消费消息
 	for msg := range msgs {
 		limit <- struct{}{}
+		wait.Add(1)
 		go func(m amqp.Delivery) {
 			if err := recover(); err != nil {
 				logx.Error(err)
 			}
 			defer func() {
 				<-limit
+				wait.Done()
 			}()
 			if err = handleFunc(m.Body); err != nil {
 				logx.Error("failed to deal with MQ message:", string(m.Body))
@@ -122,5 +126,6 @@ func (h *RabbitMqHandle) Consume(queueName constants.RABBIT_MQ, handleFunc func(
 			}
 		}(msg)
 	}
+	wait.Wait()
 	return nil
 }