From e51d4c53de6e9a3ecfe9542d4bfdef215d613229 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 16 十一月 2023 10:56:55 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/aps/SRM

---
 main.go |   41 +++++++++++++++++++++++++++++++++++++----
 1 files changed, 37 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 50fa538..fe6d232 100644
--- a/main.go
+++ b/main.go
@@ -1,8 +1,13 @@
 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/core"
 	"srm/global"
@@ -33,11 +38,39 @@
 		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{})
+		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()
-	defer test.CloseProductServiceConn()
-	initialize.InitRpcClient()
-	defer initialize.CloseRpcClient()
+	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