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"
	BigServiceUser      = "bigmember_service_user"
)

// 订单简单信息
type VipSimpleMsg struct {
	Area             *map[string]interface{}  `json:"area"`             //选择地区
	Industry         []string                 `json:"industry"`         //选择行业
	Cyclecount       int                      `json:"cyclecount"`       //日期数量(订单详情展示使用)
	Cycleunit        int                      `json:"cycleunit"`        //日期单位(订单详情展示使用)
	Buyset           *SubvipBuySet            `json:"buyset"`           //旧购买详情
	NewBuyset        *SubvipBuySet            `json:"newBuyset"`        //新购买详情
	UpgradeSubtotail []map[string]interface{} `json:"upgradeSubtotail"` //升级清单(订单详情计价清单展示使用)
	OrderType        int                      `json:"ordertype"`        //1,3,5   类型 1:订单 2:续费 3:立即升级 4:下月升级 5:即将到期（可升降级续费）6:升级订单未生效再次升级
	DisWord          string                   `json:"disWord"`          //分销系统 口令
	ContractStatus   int                      `json:"contractStatus"`   //合同状态
	Source           string                   `json:"source"`           //
	Badge            string                   `json:"badge"`            //
	Remark           string                   `json:"remark"`           //备注字段
}

// 购买内容
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)
}