前置代理

api/biservice.go

@@ -4,15 +4,15 @@ import (
 	"flag"
 	"fmt"
 
-	// "os"
-	// "os/signal"
-	// "syscall"
+	"os"
+	"os/signal"
+	"syscall"
 
 	"bp.jydev.jianyu360.cn/BaseService/biService/api/internal/config"
 	"bp.jydev.jianyu360.cn/BaseService/biService/api/internal/handler"
 	"bp.jydev.jianyu360.cn/BaseService/biService/api/internal/svc"
 
-	// "bp.jydev.jianyu360.cn/BaseService/gateway/core/node"
+	"bp.jydev.jianyu360.cn/BaseService/gateway/core/node"
 	"github.com/zeromicro/go-zero/core/conf"
 	"github.com/zeromicro/go-zero/core/logx"
 	"github.com/zeromicro/go-zero/rest"
@@ -28,12 +28,12 @@ func main() {
 	// logx 根据配置初始化
 	logx.MustSetup(c.Logx)
 	//前置代理相关
-	// var serverCode string = "biservice"
-	// var serverPort int = c.Port
-	// closeNotify, err := node.NewNode(c.GatewayRpcConf.Etcd.Hosts[0]).Register(serverCode, fmt.Sprintf("%d", serverPort))
-	// if err != nil {
-	// 	panic(err)
-	// }
+	var serverCode string = "biservice"
+	var serverPort int = c.Port
+	closeNotify, err := node.NewNode(c.GatewayRpcConf.Etcd.Hosts[0]).Register(serverCode, fmt.Sprintf("%d", serverPort))
+	if err != nil {
+		panic(err)
+	}
 
 	ctx := svc.NewServiceContext(c)
 	server := rest.MustNewServer(c.RestConf)
@@ -44,8 +44,8 @@ func main() {
 	fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
 	server.Start()
 
-	// quit := make(chan os.Signal, 1)
-	// signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
-	// <-quit
-	// closeNotify()
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+	closeNotify()
 }