feat mysql

Server/scene/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 {
@@ -24,7 +24,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
 		}
 	}
@@ -34,7 +34,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
 		}
 	}
@@ -48,7 +48,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/scene/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/scene/app/db.go

@@ -1,31 +1,23 @@
 package app
 
 import (
-	"common/db/etcd"
-	"common/log"
+	"common/db"
 	"scene/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/scene/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
 }