feat mysql

Server/gateway/app/app.go

@@ -13,9 +13,9 @@ type Program struct {
 }
 
 type Module interface {
-	Init() error
-	Start() error
-	Stop() error
+	init() error
+	start() error
+	stop() error
 }
 
 func (p *Program) Init(_ svc.Environment) error {
@@ -27,7 +27,7 @@ func (p *Program) Init(_ svc.Environment) error {
 	p.moduleList = append(p.moduleList, &ModuleGrpcServer{})
 
 	for _, module := range p.moduleList {
-		if err := module.Init(); err != nil {
+		if err := module.init(); err != nil {
 			return err
 		}
 	}
@@ -37,7 +37,7 @@ func (p *Program) Init(_ svc.Environment) error {
 
 func (p *Program) Start() error {
 	for _, module := range p.moduleList {
-		if err := module.Start(); err != nil {
+		if err := module.start(); err != nil {
 			return err
 		}
 	}
@@ -51,7 +51,7 @@ func (p *Program) Stop() error {
 	discover.Close()
 	for i := len(p.moduleList) - 1; i >= 0; i-- {
 		module := p.moduleList[i]
-		if err := module.Stop(); err != nil {
+		if err := module.stop(); err != nil {
 			log.Errorf("module stop error: %v", err)
 		}
 	}

Server/gateway/app/base.go

@@ -11,7 +11,7 @@ import (
 type ModuleBase struct {
 }
 
-func (p *ModuleBase) Init() error {
+func (p *ModuleBase) init() error {
 	// 配置
 	if err := config.LoadConfig(); err != nil {
 		return err
@@ -24,10 +24,10 @@ func (p *ModuleBase) Init() error {
 	return nil
 }
 
-func (p *ModuleBase) Start() error {
+func (p *ModuleBase) start() error {
 	return nil
 }
 
-func (p *ModuleBase) Stop() error {
+func (p *ModuleBase) stop() error {
 	return nil
 }

Server/gateway/app/db.go

@@ -1,31 +1,23 @@
 package app
 
 import (
-	"common/db/etcd"
-	"common/log"
+	"common/db"
 	"gateway/config"
 )
 
 // ModuleDB 数据库模块
 type ModuleDB struct {
+	dbModule *db.ModuleDB
 }
 
-func (p *ModuleDB) Init() error {
-	cfg := config.Get()
-	// ETCD
-	if err := etcd.Init(cfg.DB.Etcd.Address); err != nil {
-		return err
-	}
+func (p *ModuleDB) init() error {
+	return p.dbModule.Init(config.Get().DB)
+}
+
+func (p *ModuleDB) start() error {
 	return nil
 }
 
-func (p *ModuleDB) Start() error {
-	return nil
-}
-
-func (p *ModuleDB) Stop() error {
-	if err := etcd.Close(); err != nil {
-		log.Errorf("close etcd failed: %v", err)
-	}
-	return nil
+func (p *ModuleDB) stop() error {
+	return p.dbModule.Stop()
 }

Server/gateway/app/grpc.go

@@ -11,17 +11,17 @@ type ModuleGrpcServer struct {
 	server service.IService
 }
 
-func (m *ModuleGrpcServer) Init() error {
+func (m *ModuleGrpcServer) init() error {
 	return nil
 }
 
-func (m *ModuleGrpcServer) Start() error {
+func (m *ModuleGrpcServer) start() error {
 	m.server = server.NewServer(config.Get().Serve.Grpc.TTL)
 	m.server.Init(config.Get().Serve.Grpc.Address, config.Get().Serve.Grpc.Port)
 	return nil
 }
 
-func (m *ModuleGrpcServer) Stop() error {
+func (m *ModuleGrpcServer) stop() error {
 	m.server.Close()
 	return nil
 }

Server/gateway/app/prometheus.go

@@ -17,12 +17,12 @@ type ModulePrometheus struct {
 	server *http.Server
 }
 
-func (m *ModulePrometheus) Init() error {
+func (m *ModulePrometheus) init() error {
 	m.wg = &sync.WaitGroup{}
 	return nil
 }
 
-func (m *ModulePrometheus) Start() error {
+func (m *ModulePrometheus) start() error {
 	m.wg.Add(1)
 	go func() {
 		defer m.wg.Done()
@@ -38,7 +38,7 @@ func (m *ModulePrometheus) Start() error {
 	return nil
 }
 
-func (m *ModulePrometheus) Stop() error {
+func (m *ModulePrometheus) stop() error {
 	if err := m.server.Shutdown(context.Background()); err != nil {
 		log.Errorf("stop prometheus server failed: %v", err)
 	}

Server/gateway/app/web.go

@@ -19,13 +19,13 @@ type ModuleWebServer struct {
 	router *gin.Engine
 }
 
-func (m *ModuleWebServer) Init() error {
+func (m *ModuleWebServer) init() error {
 	m.wg = &sync.WaitGroup{}
 	m.router = http_gateway.InitRouter()
 	return nil
 }
 
-func (m *ModuleWebServer) Start() error {
+func (m *ModuleWebServer) start() error {
 	m.wg.Add(1)
 	go func() {
 		defer m.wg.Done()
@@ -41,7 +41,7 @@ func (m *ModuleWebServer) Start() error {
 	return nil
 }
 
-func (m *ModuleWebServer) Stop() error {
+func (m *ModuleWebServer) stop() error {
 	if err := m.server.Shutdown(context.Background()); err != nil {
 		log.Errorf("stop http server failed: %v", err)
 	}

Server/gateway/app/websocket.go

@@ -16,7 +16,7 @@ type ModuleWebsocketServer struct {
 	server *websocket.WSServer
 }
 
-func (m *ModuleWebsocketServer) Init() error {
+func (m *ModuleWebsocketServer) init() error {
 	m.wg = &sync.WaitGroup{}
 	m.server = websocket.NewWSServer(
 		&ws_gateway.GatewayWsServer{},
@@ -26,7 +26,7 @@ func (m *ModuleWebsocketServer) Init() error {
 	return nil
 }
 
-func (m *ModuleWebsocketServer) Start() error {
+func (m *ModuleWebsocketServer) start() error {
 	m.wg.Add(1)
 	go func() {
 		defer m.wg.Done()
@@ -44,7 +44,7 @@ func (m *ModuleWebsocketServer) Start() error {
 	return nil
 }
 
-func (m *ModuleWebsocketServer) Stop() error {
+func (m *ModuleWebsocketServer) stop() error {
 	if err := m.server.Stop(); err != nil {
 		log.Errorf("stop websocket server failed: %v", err)
 	}