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"
|
"srm/initialize"
|
)
|
|
//go:generate go env -w GO111MODULE=on
|
//go:generate go env -w GOPROXY=https://goproxy.cn,direct
|
//go:generate go mod tidy
|
//go:generate go mod download
|
|
// @title Swagger Example API
|
// @version 0.0.1
|
// @description This is a sample Server pets
|
// @securityDefinitions.apikey ApiKeyAuth
|
// @in header
|
// @name x-token
|
// @BasePath /
|
func main() {
|
global.GVA_VP = core.Viper() // 初始化Viper
|
global.GVA_LOG = core.Zap() // 初始化zap日志库
|
zap.ReplaceGlobals(global.GVA_LOG)
|
global.GVA_DB = initialize.Gorm() // gorm连接数据库
|
initialize.DBList()
|
if global.GVA_DB != nil {
|
initialize.RegisterTables() // 初始化表
|
// 程序结束前关闭数据库链接
|
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 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()
|
}
|
