代码恢复

liujiandao

2023-11-16 645e6cae2bdc4102c351c014bcc443b29fdd1dc1

 main.go

@@ -15,6 +15,7 @@
   "wms/models"
   "wms/pkg/logx"
   "wms/proto/product_inventory"
   "wms/proto/purchase_wms"
   "wms/router"
)

@@ -46,6 +47,7 @@
   //启动grpc客户端
   go controllers.InitInventoryOrderServiceConn()
   go controllers.InitProductInventoryServiceConn()
   go controllers.InitCodeServiceConn()
   //启动grpc服务
   go func() {
      ln, err := net.Listen("tcp", ":"+conf.WebConf.GrpcPort)
@@ -57,6 +59,7 @@
      s := grpc.NewServer()
      //todo 添加具体服务
      product_inventory.RegisterProductInventoryServiceServer(s, &product_inventory.Server{})
      purchase_wms.RegisterPurchaseServiceServer(s, &purchase_wms.Server{})
      err = s.Serve(ln)
      if err != nil {
         logx.Errorf("grpc server init error: %v", err.Error())
@@ -78,6 +81,7 @@

   controllers.CloseInventoryOrderServiceConn()
   controllers.CloseProductInventoryServiceConn()
   controllers.CloseCodeServiceConn()
   // 关闭HTTP服务器
   if err := server.Shutdown(ctx); err != nil {
      logx.Infof("服务优雅退出失败: %v", err)