package init_client
|
|
import (
|
"google.golang.org/grpc"
|
"google.golang.org/grpc/credentials/insecure"
|
"wms/conf"
|
"wms/pkg/logx"
|
)
|
|
var (
|
AdminConn *grpc.ClientConn
|
ApsConn *grpc.ClientConn
|
CrmConn *grpc.ClientConn
|
SrmConn *grpc.ClientConn
|
)
|
|
func InitClient() {
|
var err error
|
ApsConn, err = grpc.Dial(conf.GrpcServerConf.ApsAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
|
if err != nil {
|
logx.Errorf("grpc dial product service error: %v", err.Error())
|
return
|
}
|
CrmConn, err = grpc.Dial(conf.GrpcServerConf.CrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
|
if err != nil {
|
logx.Errorf("grpc dial product service error: %v", err.Error())
|
return
|
}
|
SrmConn, err = grpc.Dial(conf.GrpcServerConf.SrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
|
if err != nil {
|
logx.Errorf("grpc dial product service error: %v", err.Error())
|
return
|
}
|
if len(conf.GrpcServerConf.AdminAddr) > 0 {
|
AdminConn, err = grpc.Dial(conf.GrpcServerConf.AdminAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
|
if err != nil {
|
logx.Errorf("grpc dial product service error: %v", err.Error())
|
return
|
}
|
}
|
}
|
|
func CloseClient() {
|
if AdminConn != nil {
|
AdminConn.Close()
|
}
|
if ApsConn != nil {
|
ApsConn.Close()
|
}
|
if CrmConn != nil {
|
CrmConn.Close()
|
}
|
if SrmConn != nil {
|
SrmConn.Close()
|
}
|
}
|