From 99ffda4126a4217ecaf57f4eab2a5615ae353aff Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期二, 07 十一月 2023 14:30:51 +0800
Subject: [PATCH] crm获取库存信息

---
 main.go |   25 +++++++++++++++++++++++++
 1 files changed, 25 insertions(+), 0 deletions(-)

diff --git a/main.go b/main.go
index 3ee1279..4f7db20 100644
--- a/main.go
+++ b/main.go
@@ -2,14 +2,19 @@
 
 import (
 	"context"
+	"fmt"
+	"google.golang.org/grpc"
+	"net"
 	"net/http"
 	"os"
 	"os/signal"
 	"syscall"
 	"time"
 	"wms/conf"
+	"wms/controllers"
 	"wms/models"
 	"wms/pkg/logx"
+	"wms/proto/product_inventory"
 	"wms/router"
 )
 
@@ -38,6 +43,25 @@
 		WriteTimeout: 5 * time.Second,
 	}
 	go shutdown(server)
+	//鍚姩grpc瀹㈡埛绔�
+	go controllers.InitInventoryOrderServiceConn()
+	//鍚姩grpc鏈嶅姟
+	go func() {
+		ln, err := net.Listen("tcp", ":"+conf.WebConf.GrpcPort)
+		if err != nil {
+			logx.Errorf("grpc server init error: %v", err.Error())
+			panic(fmt.Sprintf("grpc server init error: %v", err.Error()))
+		}
+		fmt.Println("-----------鐩戝惉绔彛: ", conf.WebConf.GrpcPort)
+		s := grpc.NewServer()
+		//todo 娣诲姞鍏蜂綋鏈嶅姟
+		product_inventory.RegisterProductInventoryServiceServer(s, &product_inventory.Server{})
+		err = s.Serve(ln)
+		if err != nil {
+			logx.Errorf("grpc server init error: %v", err.Error())
+			panic(fmt.Sprintf("grpc server init error: %v", err.Error()))
+		}
+	}()
 	logx.Error(server.ListenAndServe().Error())
 }
 
@@ -51,6 +75,7 @@
 	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer cancel()
 
+	controllers.CloseInventoryOrderServiceConn()
 	// 鍏抽棴HTTP鏈嶅姟鍣�
 	if err := server.Shutdown(ctx); err != nil {
 		logx.Infof("鏈嶅姟浼橀泤閫�鍑哄け璐�: %v", err)

--
Gitblit v1.8.0