feat kafka

module/db.go

@@ -3,6 +3,7 @@ package module
 import (
 	"git.hlsq.asia/mmorpg/service-common/config"
 	"git.hlsq.asia/mmorpg/service-common/db/etcd"
+	"git.hlsq.asia/mmorpg/service-common/db/kafka"
 	"git.hlsq.asia/mmorpg/service-common/db/mysql"
 	"git.hlsq.asia/mmorpg/service-common/db/redis"
 	"git.hlsq.asia/mmorpg/service-common/log"
@@ -11,7 +12,8 @@ import (
 // DB 数据库模块
 type DB struct {
 	DefaultModule
-	cfg *config.DBConfig
+	cfg         *config.DBConfig
+	serviceName string
 }
 
 func (m *DB) Init() error {
@@ -33,6 +35,12 @@ func (m *DB) Init() error {
 			return err
 		}
 	}
+	// KAFKA
+	if m.cfg.Kafka != nil {
+		if err := kafka.Init(m.cfg.Kafka, m.serviceName); err != nil {
+			return err
+		}
+	}
 	return nil
 }
 
@@ -46,6 +54,9 @@ func (m *DB) Stop() error {
 	if err := redis.Close(); err != nil {
 		log.Errorf("close redis failed: %v", err)
 	}
+	if err := kafka.Close(); err != nil {
+		log.Errorf("close kafka failed: %v", err)
+	}
 	return nil
 }
 
@@ -56,5 +67,8 @@ func (m *DB) Bind(data ...any) Module {
 	if cfg, ok := data[0].(*config.DBConfig); ok {
 		m.cfg = cfg
 	}
+	if name, ok := data[1].(string); ok {
+		m.serviceName = name
+	}
 	return m
 }