加入网络层

Server/Gateway/app/app.go

@@ -0,0 +1,76 @@
+package app
+
+import (
+	"common/db/etcd"
+	"common/discover"
+	"common/log"
+	"common/net/grpc/service"
+	"common/net/socket/websocket"
+	"fmt"
+	"gateway/config"
+	"gateway/grpc_server"
+	"gateway/handler/ws_handler"
+	"github.com/gin-gonic/gin"
+	"github.com/judwhite/go-svc"
+	"runtime/debug"
+	"sync"
+)
+
+type Program struct {
+	wg        *sync.WaitGroup
+	server    service.IService    // grpc服务
+	webServer *gin.Engine         // web服务
+	wsServer  *websocket.WSServer // websocket服务
+	stop      chan bool
+}
+
+func (p *Program) Init(_ svc.Environment) error {
+	if cfg, err := p.initBase(); err != nil {
+		return err
+	} else {
+		log.Infof(fmt.Sprintf("%v starting...", cfg.App.Name))
+		if err := p.initDB(cfg); err != nil {
+			return err
+		}
+		if err := p.initWebServer(cfg); err != nil {
+			return err
+		}
+		if err := p.initWsServer(cfg); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (p *Program) Start() error {
+	defer func() {
+		if err := recover(); err != nil {
+			fmt.Printf("Start err: %v", err)
+			debug.PrintStack()
+			_ = p.Stop()
+		}
+	}()
+
+	discover.Listen()
+	p.server = grpc_server.NewServer(config.Get().Grpc.Registry.TTL)
+	p.server.Init(config.Get().Grpc.Registry.Address, config.Get().Grpc.Registry.Port)
+
+	ws_handler.NewClient(123, nil)
+
+	return nil
+}
+
+func (p *Program) Stop() error {
+	defer func() {
+		if err := recover(); err != nil {
+			fmt.Printf("Stop err: %v", err)
+			debug.PrintStack()
+		}
+	}()
+
+	discover.Close()
+	p.server.Close()
+	_ = etcd.Close()
+
+	return nil
+}