feat 改名

Server/gateway/app/app.go

@@ -0,0 +1,60 @@
+package app
+
+import (
+	"common/discover"
+	"common/log"
+	"fmt"
+	"gateway/config"
+	"github.com/judwhite/go-svc"
+)
+
+type Program struct {
+	moduleList []Module // 模块列表
+}
+
+type Module interface {
+	Init() error
+	Start() error
+	Stop() error
+}
+
+func (p *Program) Init(_ svc.Environment) error {
+	p.moduleList = append(p.moduleList, &ModuleBase{})
+	p.moduleList = append(p.moduleList, &ModuleDB{})
+	p.moduleList = append(p.moduleList, &ModuleWebServer{})
+	p.moduleList = append(p.moduleList, &ModuleWebsocketServer{})
+	p.moduleList = append(p.moduleList, &ModuleGrpcServer{})
+
+	for _, module := range p.moduleList {
+		if err := module.Init(); err != nil {
+			return err
+		}
+	}
+	log.Infof(fmt.Sprintf("%v Init successful...", config.Get().App.Name))
+	return nil
+}
+
+func (p *Program) Start() error {
+	for _, module := range p.moduleList {
+		if err := module.Start(); err != nil {
+			return err
+		}
+	}
+	discover.Listen()
+
+	log.Infof(fmt.Sprintf("%v Start successful...", config.Get().App.Name))
+	return nil
+}
+
+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 {
+			log.Errorf("module stop error: %v", err)
+		}
+	}
+
+	log.Infof(fmt.Sprintf("%v Stop successful...", config.Get().App.Name))
+	return nil
+}

Server/gateway/app/base.go

@@ -0,0 +1,33 @@
+package app
+
+import (
+	"common/log"
+	"common/utils"
+	"gateway/config"
+	"math/rand"
+)
+
+// ModuleBase 基础模块，或者一些零散的模块
+type ModuleBase struct {
+}
+
+func (p *ModuleBase) Init() error {
+	// 配置
+	if err := config.LoadConfig(); err != nil {
+		return err
+	}
+	cfg := config.Get()
+	// 日志
+	log.Init(cfg.Log.Debug, cfg.Log.MaxSize, cfg.Log.MaxBackups, cfg.Log.MaxAge, cfg.Log.Level)
+	// 雪花
+	utils.InitSnowflake(int64(rand.Intn(1000)))
+	return nil
+}
+
+func (p *ModuleBase) Start() error {
+	return nil
+}
+
+func (p *ModuleBase) Stop() error {
+	return nil
+}

Server/gateway/app/db.go

@@ -0,0 +1,31 @@
+package app
+
+import (
+	"common/db/etcd"
+	"common/log"
+	"gateway/config"
+)
+
+// ModuleDB 数据库模块
+type ModuleDB struct {
+}
+
+func (p *ModuleDB) Init() error {
+	cfg := config.Get()
+	// ETCD
+	if err := etcd.Init(cfg.DB.Etcd.Address); err != nil {
+		return err
+	}
+	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
+}

Server/gateway/app/grpc.go

@@ -0,0 +1,27 @@
+package app
+
+import (
+	"common/net/grpc/service"
+	"gateway/config"
+	"gateway/grpc_server/server"
+)
+
+// ModuleGrpcServer Grpc服务模块
+type ModuleGrpcServer struct {
+	server service.IService
+}
+
+func (m *ModuleGrpcServer) Init() error {
+	return nil
+}
+
+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 {
+	m.server.Close()
+	return nil
+}

Server/gateway/app/web.go

@@ -0,0 +1,49 @@
+package app
+
+import (
+	"common/log"
+	"errors"
+	"fmt"
+	"gateway/config"
+	"gateway/net/http_gateway"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sync"
+)
+
+// ModuleWebServer Web服务模块
+type ModuleWebServer struct {
+	wg     *sync.WaitGroup
+	server *http.Server
+	router *gin.Engine
+}
+
+func (m *ModuleWebServer) Init() error {
+	m.wg = &sync.WaitGroup{}
+	m.router = http_gateway.InitRouter()
+	return nil
+}
+
+func (m *ModuleWebServer) Start() error {
+	m.wg.Add(1)
+	go func() {
+		defer m.wg.Done()
+		m.server = &http.Server{
+			Addr:    fmt.Sprintf("%v:%v", config.Get().Serve.Http.Address, config.Get().Serve.Http.Port),
+			Handler: m.router,
+		}
+		if err := m.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+			log.Errorf("http server failed: %v", err.Error())
+		}
+		log.Infof("http server stop.")
+	}()
+	return nil
+}
+
+func (m *ModuleWebServer) Stop() error {
+	if err := m.server.Close(); err != nil {
+		log.Errorf("stop http server failed: %v", err)
+	}
+	m.wg.Wait()
+	return nil
+}

Server/gateway/app/websocket.go

@@ -0,0 +1,53 @@
+package app
+
+import (
+	"common/log"
+	"common/net/socket/websocket"
+	"fmt"
+	"gateway/config"
+	"gateway/net/ws_gateway"
+	"sync"
+	"time"
+)
+
+// ModuleWebsocketServer Websocket服务模块
+type ModuleWebsocketServer struct {
+	wg     *sync.WaitGroup
+	server *websocket.WSServer
+}
+
+func (m *ModuleWebsocketServer) Init() error {
+	m.wg = &sync.WaitGroup{}
+	m.server = websocket.NewWSServer(
+		&ws_gateway.GatewayWsServer{},
+		log.GetLogger().Named("ws_server"),
+		5*time.Second,
+	)
+	return nil
+}
+
+func (m *ModuleWebsocketServer) Start() error {
+	m.wg.Add(1)
+	go func() {
+		defer m.wg.Done()
+		_ = m.server.Run(
+			log.GetLogger().Named("GNET"),
+			fmt.Sprintf("tcp4://0.0.0.0:%v", config.Get().Serve.Socket.Web.Port),
+			true,
+			true,
+			false,
+			false,
+			true,
+			8,
+		)
+	}()
+	return nil
+}
+
+func (m *ModuleWebsocketServer) Stop() error {
+	if err := m.server.Stop(); err != nil {
+		log.Errorf("stop websocket server failed: %v", err)
+	}
+	m.wg.Wait()
+	return nil
+}