From de4bcd1dead50b05f716bc5718be5540bdb96783 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期日, 28 四月 2024 17:23:29 +0800
Subject: [PATCH] fix
---
main.go | 44 ++++++++++++++++++++++++++++++++++++++++----
1 files changed, 40 insertions(+), 4 deletions(-)
diff --git a/main.go b/main.go
index 266e282..2b4c8d8 100644
--- a/main.go
+++ b/main.go
@@ -1,8 +1,14 @@
package main
import (
+ "fmt"
"go.uber.org/zap"
+ "google.golang.org/grpc"
+ "net"
"srm/api/v1/test"
+ "srm/proto/purchase_wms"
+ "srm/proto/qualityinspect"
+ "srm/proto/supplier"
"srm/core"
"srm/global"
@@ -23,11 +29,9 @@
// @BasePath /
func main() {
global.GVA_VP = core.Viper() // 鍒濆鍖朧iper
- initialize.OtherInit()
- global.GVA_LOG = core.Zap() // 鍒濆鍖杬ap鏃ュ織搴�
+ global.GVA_LOG = core.Zap() // 鍒濆鍖杬ap鏃ュ織搴�
zap.ReplaceGlobals(global.GVA_LOG)
global.GVA_DB = initialize.Gorm() // gorm杩炴帴鏁版嵁搴�
- initialize.Timer()
initialize.DBList()
if global.GVA_DB != nil {
initialize.RegisterTables() // 鍒濆鍖栬〃
@@ -35,8 +39,40 @@
db, _ := global.GVA_DB.DB()
defer db.Close()
}
+ //鍚姩grpc鏈嶅姟
+ go func() {
+ ln, err := net.Listen("tcp", ":"+global.GVA_CONFIG.System.GrpcPort)
+ if err != nil {
+ global.GVA_LOG.Error("grpc server init error: %v", zap.Error(err))
+ panic(fmt.Sprintf("grpc server init error: %v", err.Error()))
+ }
+ s := grpc.NewServer()
+ //todo 娣诲姞鍏蜂綋鏈嶅姟
+ purchase_wms.RegisterPurchaseServiceServer(s, &purchase_wms.Server{})
+ supplier.RegisterSupplierServiceServer(s, &supplier.Server{})
+ err = s.Serve(ln)
+ if err != nil {
+ global.GVA_LOG.Error("grpc server init error: %v", zap.Error(err))
+ panic(fmt.Sprintf("grpc server init error: %v", err.Error()))
+ }
+ }()
- go test.InitProductServiceConn()
+ go initGrpcClient()
+ defer closeGrpcClient()
core.RunWindowsServer()
}
+
+func initGrpcClient() {
+ test.InitProductServiceConn()
+ qualityinspect.InitConn()
+ purchase_wms.InitPurchaseConn()
+ test.InitProductServiceConn()
+}
+
+func closeGrpcClient() {
+ test.CloseProductServiceConn()
+ qualityinspect.CloseConn()
+ purchase_wms.ClosePurchaseConn()
+ test.CloseProductServiceConn()
+}
--
Gitblit v1.8.0