entManageApplication/entity/entity.go

package entity

import (
	"log"

	"app.yhyue.com/moapp/jyResourcesCenter/rpc/resourcesCenterclient"
	"app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/rpc/powercheck"
	"bp.jydev.jianyu360.cn/BaseService/resourceCenter/rpc/resource"
	"bp.jydev.jianyu360.cn/BaseService/userCenter/rpc/usercenter"
	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/zrpc"
)

const (
	NewOther     = "newother"
	PowerCacheDb = "other"

	EntnicheInfo        = "entniche_info"
	EntnicheUser        = "entniche_user"
	EntnichePower       = "entniche_power"
	EntnicheWaitEmpower = "entniche_wait_empower"
	EntnicheOrder       = "entniche_order"
	Order               = "dataexport_order"
	JyOrderDetail       = "jy_order_detail"
	BigServiceUser      = "bigmember_service_user"
)

// 订单简单信息
type VipCycleFilter struct {
	BuyAccountCount  int `json:"buyAccountCount"`  //购买数量(仅购买主体是企业)
	GiftAccountCount int `json:"giftAccountCount"` //赠送数量 要算售价(仅购买主体是企业)
	CutAccount       int `json:"cutAccount"`       //不计入成本的子账号数量(仅购买主体是企业)
	BuyCycle         int `json:"buy_cycle"`        //购买周期
	BuyType          int `json:"buy_type"`         //购买周期 类型 1天 2月 3年 4季度
	GiftCycle        int `json:"give_cycle"`       //赠送周期
	GiftType         int `json:"give_type"`        //赠送周期 类型 1天 2月 3年 4季度

	FinalAccountCount int `json:"finalAccountCount"` //当前套餐最终
	FinalAreaCount    int `json:"finalAreaCount"`

	Supplement string `json:"supplement" doc:"补充服务"`
}

// 购买内容
type SubvipBuySet struct {
	Upgrade         int   `json:"upgrade"`         //是否是升级版;1是 其他不是
	AreaCount       int   `json:"areacount"`       //-1 全国  >0 省份数量
	NewCitys        []int `json:"newcitys"`        //城市数量(新)
	BuyerclassCount int   `json:"buyerclasscount"` //行业数
}

type MysqlDb struct {
	Address      string //数据库地址:端口
	UserName     string //用户名
	PassWord     string //密码
	DBName       string //数据库名
	MaxOpenConns int    //用于设置最大打开的连接数，默认值为0表示不限制。
	MaxIdleConns int    //用于设置闲置的连接数。
}

var (
	JyMysql             *mysql.Mysql
	Mgo                 *mongodb.MongodbSim
	UserCenterRpc       usercenter.UserCenter
	PowerCheckCenterRpc powercheck.PowerCheck
	ResourceCenterRpc   resourcesCenterclient.ResourcesCenter
	ResourceRpc         resource.Resource
)

func InitMysql(n *MysqlDb) {
	JyMysql = &mysql.Mysql{
		Address:      n.Address,
		UserName:     n.UserName,
		PassWord:     n.PassWord,
		DBName:       n.DBName,
		MaxOpenConns: n.MaxOpenConns,
		MaxIdleConns: n.MaxIdleConns,
	}
	JyMysql.Init()
	logx.Info("初始化mysql")
}

func InitMongo(addr, db string, size int) {
	Mgo = mongodb.NewMgo(addr, db, size)
}

func InitRpc(hosts []string, userCenterRpc, powerCheckRpc, resourceCenterRpc, resourceRpc string) {
	uc, err := zrpc.NewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: hosts,
			Key:   userCenterRpc,
		},
	})
	if err != nil {
		log.Fatalln(err)
	}
	UserCenterRpc = usercenter.NewUserCenter(uc)
	//
	pc, err := zrpc.NewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: hosts,
			Key:   powerCheckRpc,
		},
	})
	if err != nil {
		log.Fatalln(err)
	}
	PowerCheckCenterRpc = powercheck.NewPowerCheck(pc)
	//
	rc, err := zrpc.NewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: hosts,
			Key:   resourceCenterRpc,
		},
	})
	if err != nil {
		log.Fatalln(err)
	}
	ResourceCenterRpc = resourcesCenterclient.NewResourcesCenter(rc)
	//
	r, err := zrpc.NewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: hosts,
			Key:   resourceRpc,
		},
	})
	if err != nil {
		log.Fatalln(err)
	}
	ResourceRpc = resource.NewResource(r)
}