更新文档

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"
@@ -23,11 +29,9 @@
// @BasePath                    /
func main() {
   global.GVA_VP = core.Viper() // 初始化Viper
   initialize.OtherInit()
   global.GVA_LOG = core.Zap() // 初始化zap日志库
   global.GVA_LOG = core.Zap()  // 初始化zap日志库
   zap.ReplaceGlobals(global.GVA_LOG)
   global.GVA_DB = initialize.Gorm() // gorm连接数据库
   initialize.Timer()
   initialize.DBList()
   if global.GVA_DB != nil {
      initialize.RegisterTables() // 初始化表
@@ -35,8 +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()
   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()
}