更新文档

zhangqian

2024-04-19 cdb38521ea1f662b53bafb87412c38dfd0d5e11d

 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"
@@ -33,11 +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()
   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()
}