package init_client

import (
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"wms/conf"
	"wms/pkg/logx"
)

var (
	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
	}
}

func CloseClient() {
	if ApsConn != nil {
		ApsConn.Close()
	}
	if CrmConn != nil {
		CrmConn.Close()
	}
	if SrmConn != nil {
		SrmConn.Close()
	}
}