From 727d16a6d0336daa6c2f9541564d1500444e44a2 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期四, 16 十一月 2023 09:43:47 +0800
Subject: [PATCH] 采购单与wms通信

---
 main.go |   38 +++++++++++++++++++++++++++++++++++---
 1 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index a8d8ebe..af4d670 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,10 +38,37 @@
 		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()
-	initialize.InitRpcClient()
-	defer initialize.CloseRpcClient()
+	go initGrpcClient()
+	defer closeGrpcClient()
 
 	core.RunWindowsServer()
 }
+
+func initGrpcClient() {
+	test.InitProductServiceConn()
+	qualityinspect.InitConn()
+	purchase_wms.InitPurchaseConn()
+}
+
+func closeGrpcClient() {
+	test.CloseProductServiceConn()
+	qualityinspect.CloseConn()
+	purchase_wms.ClosePurchaseConn()
+}

--
Gitblit v1.8.0