发布v2.5.3版本，更新内容请查看：https://github.com/bufanyun/hotgo/blob/v2.0/docs/guide-zh-CN/start-update-log.md

server/internal/library/queue/queue.go

@@ -0,0 +1,217 @@
+// Package queue
+// @Link  https://github.com/bufanyun/hotgo
+// @Copyright  Copyright (c) 2023 HotGo CLI
+// @Author  Ms <133814250@qq.com>
+// @License  https://github.com/bufanyun/hotgo/blob/master/LICENSE
+package queue
+
+import (
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gctx"
+	"hotgo/internal/library/queue/disk"
+	"hotgo/utility/charset"
+	"sync"
+	"time"
+)
+
+type MqProducer interface {
+	SendMsg(topic string, body string) (mqMsg MqMsg, err error)
+	SendByteMsg(topic string, body []byte) (mqMsg MqMsg, err error)
+	SendDelayMsg(topic string, body string, delaySecond int64) (mqMsg MqMsg, err error)
+}
+
+type MqConsumer interface {
+	ListenReceiveMsgDo(topic string, receiveDo func(mqMsg MqMsg)) (err error)
+}
+
+const (
+	_ = iota
+	SendMsg
+	ReceiveMsg
+)
+
+type Config struct {
+	Switch    bool   `json:"switch"`
+	Driver    string `json:"driver"`
+	Retry     int    `json:"retry"`
+	GroupName string `json:"groupName"`
+	Redis     RedisConf
+	Rocketmq  RocketmqConf
+	Kafka     KafkaConf
+	Disk      *disk.Config
+}
+
+type RedisConf struct {
+	Timeout int64 `json:"timeout"`
+}
+type RocketmqConf struct {
+	Address  []string `json:"address"`
+	LogLevel string   `json:"logLevel"`
+}
+
+type KafkaConf struct {
+	Address       []string `json:"address"`
+	Version       string   `json:"version"`
+	RandClient    bool     `json:"randClient"`
+	MultiConsumer bool     `json:"multiConsumer"`
+}
+
+type MqMsg struct {
+	RunType   int       `json:"run_type"`
+	Topic     string    `json:"topic"`
+	MsgId     string    `json:"msg_id"`
+	Offset    int64     `json:"offset"`
+	Partition int32     `json:"partition"`
+	Timestamp time.Time `json:"timestamp"`
+	Body      []byte    `json:"body"`
+}
+
+var (
+	ctx                   = gctx.New()
+	mqProducerInstanceMap map[string]MqProducer
+	mqConsumerInstanceMap map[string]MqConsumer
+	mutex                 sync.Mutex
+	config                Config
+)
+
+func init() {
+	mqProducerInstanceMap = make(map[string]MqProducer)
+	mqConsumerInstanceMap = make(map[string]MqConsumer)
+	if err := g.Cfg().MustGet(ctx, "queue").Scan(&config); err != nil {
+		g.Log().Warningf(ctx, "queue init err:%+v", err)
+	}
+}
+
+// InstanceConsumer 实例化消费者
+func InstanceConsumer() (mqClient MqConsumer, err error) {
+	return NewConsumer(config.GroupName)
+}
+
+// InstanceProducer 实例化生产者
+func InstanceProducer() (mqClient MqProducer, err error) {
+	return NewProducer(config.GroupName)
+}
+
+// NewProducer 初始化生产者实例
+func NewProducer(groupName string) (mqClient MqProducer, err error) {
+	if item, ok := mqProducerInstanceMap[groupName]; ok {
+		return item, nil
+	}
+
+	if groupName == "" {
+		err = gerror.New("mq groupName is empty.")
+		return
+	}
+
+	switch config.Driver {
+	case "rocketmq":
+		if len(config.Rocketmq.Address) == 0 {
+			err = gerror.New("queue rocketmq address is not support")
+			return
+		}
+		mqClient, err = RegisterRocketProducer(config.Rocketmq.Address, groupName, config.Retry)
+	case "kafka":
+		if len(config.Kafka.Address) == 0 {
+			err = gerror.New("queue kafka address is not support")
+			return
+		}
+		mqClient, err = RegisterKafkaProducer(KafkaConfig{
+			Brokers: config.Kafka.Address,
+			GroupID: groupName,
+			Version: config.Kafka.Version,
+		})
+	case "redis":
+		if _, err = g.Redis().Do(ctx, "ping"); err == nil {
+			mqClient = RegisterRedisMqProducer(RedisOption{
+				Timeout: config.Redis.Timeout,
+			}, groupName)
+		}
+
+	case "disk":
+		config.Disk.GroupName = groupName
+		mqClient, err = RegisterDiskMqProducer(config.Disk)
+	default:
+		err = gerror.New("queue driver is not support")
+	}
+
+	if err != nil {
+		return
+	}
+
+	mutex.Lock()
+	defer mutex.Unlock()
+	mqProducerInstanceMap[groupName] = mqClient
+
+	return
+}
+
+// NewConsumer 初始化消费者实例
+func NewConsumer(groupName string) (mqClient MqConsumer, err error) {
+	if groupName == "" {
+		err = gerror.New("mq groupName is empty.")
+		return
+	}
+
+	switch config.Driver {
+	case "rocketmq":
+		if len(config.Rocketmq.Address) == 0 {
+			err = gerror.New("queue.rocketmq.address is empty.")
+			return
+		}
+		mqClient, err = RegisterRocketConsumer(config.Rocketmq.Address, groupName)
+	case "kafka":
+		if len(config.Kafka.Address) == 0 {
+			err = gerror.New("queue kafka address is not support")
+			return
+		}
+
+		randTag := string(charset.RandomCreateBytes(6))
+		// 是否支持创建多个消费者
+		if config.Kafka.MultiConsumer == false {
+			randTag = "001"
+		}
+
+		if item, ok := mqConsumerInstanceMap[groupName+"-"+randTag]; ok {
+			return item, nil
+		}
+
+		clientId := "HOTGO-Consumer-" + groupName
+		if config.Kafka.RandClient {
+			clientId += "-" + randTag
+		}
+
+		mqClient, err = RegisterKafkaMqConsumer(KafkaConfig{
+			Brokers:  config.Kafka.Address,
+			GroupID:  groupName,
+			Version:  config.Kafka.Version,
+			ClientId: clientId,
+		})
+	case "redis":
+		if _, err = g.Redis().Do(ctx, "ping"); err == nil {
+			mqClient = RegisterRedisMqConsumer(RedisOption{
+				Timeout: config.Redis.Timeout,
+			}, groupName)
+		}
+	case "disk":
+		config.Disk.GroupName = groupName
+		mqClient, err = RegisterDiskMqConsumer(config.Disk)
+	default:
+		err = gerror.New("queue driver is not support")
+	}
+
+	if err != nil {
+		return
+	}
+
+	mutex.Lock()
+	defer mutex.Unlock()
+	mqConsumerInstanceMap[groupName] = mqClient
+
+	return
+}
+
+// BodyString 返回消息体
+func (m *MqMsg) BodyString() string {
+	return string(m.Body)
+}