jyMicroservices/jyBXBase/rpc/init/db.go

/*
初始数据库应用
*/
package init

import (
	"strings"

	"github.com/zeromicro/go-zero/core/logx"

	"jyBXBase/entity"

	"app.yhyue.com/moapp/jybase/esv1"
	"app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"app.yhyue.com/moapp/jybase/redis"
)

var (
	MainMysql          *mysql.Mysql
	PushMysql          *mysql.Mysql
	BigmemberPushMysql *mysql.Mysql
	EntnichePushMysql  *mysql.Mysql
	BaseServiceMysql   *mysql.Mysql
	Mgo                mongodb.MongodbSim
	MgoEnt             mongodb.MongodbSim
	MgoBidding         mongodb.MongodbSim
)

//
func MongoDBInit(em *entity.Mongo) {
	//初始化 mongodb
	if em.Main.Address != "" {
		logx.Info("--初始化 mongodb--")
		Mgo = mongodb.MongodbSim{
			MongodbAddr: em.Main.Address,
			Size:        em.Main.Size,
			DbName:      em.Main.DbName,
		}
		Mgo.InitPool()
	}
	//
	if em.Bidding.Address != "" {
		logx.Info("--初始化 mongodb bidding--")
		MgoBidding = mongodb.MongodbSim{
			MongodbAddr: em.Bidding.Address,
			Size:        em.Bidding.Size,
			DbName:      em.Bidding.DbName,
			UserName:    em.Bidding.UserName,
			Password:    em.Bidding.Password,
		}
		MgoBidding.InitPool()
	}
	//
	if em.Ent.Address != "" {
		logx.Info("--初始化 mongodb ent--")
		MgoEnt = mongodb.MongodbSim{
			MongodbAddr: em.Ent.Address,
			Size:        em.Ent.Size,
			DbName:      em.Ent.DbName,
			UserName:    em.Ent.UserName,
			Password:    em.Ent.Password,
		}
		MgoEnt.InitPool()
	}
}

//
func MysqlInit(em *entity.Mysql) {
	//初始化 mysql-main
	if em.Main.Address != "" {
		logx.Info("--初始化 main mysql--")
		MainMysql = &mysql.Mysql{
			Address:      em.Main.Address,
			UserName:     em.Main.UserName,
			PassWord:     em.Main.Password,
			DBName:       em.Main.DbName,
			MaxOpenConns: em.Main.MaxOpenConns,
			MaxIdleConns: em.Main.MaxIdleConns,
		}
		MainMysql.Init()
	}
	//初始化 mysql-push
	if em.Push.Address != "" {
		logx.Info("--初始化 push mysql--")
		PushMysql = &mysql.Mysql{
			Address:      em.Push.Address,
			UserName:     em.Push.UserName,
			PassWord:     em.Push.Password,
			DBName:       em.Push.DbName,
			MaxOpenConns: em.Push.MaxOpenConns,
			MaxIdleConns: em.Push.MaxIdleConns,
		}
		PushMysql.Init()
	}
	//初始化 mysql-大会员
	if em.MemberPush.Address != "" {
		logx.Info("--初始化 MemberPush mysql--")
		BigmemberPushMysql = &mysql.Mysql{
			Address:      em.MemberPush.Address,
			UserName:     em.MemberPush.UserName,
			PassWord:     em.MemberPush.Password,
			DBName:       em.MemberPush.DbName,
			MaxOpenConns: em.MemberPush.MaxOpenConns,
			MaxIdleConns: em.MemberPush.MaxIdleConns,
		}
		BigmemberPushMysql.Init()
	}
	//初始化 mysql-商机管理
	if em.EntnichePush.Address != "" {
		logx.Info("--初始化 EntnichePush mysql--")
		EntnichePushMysql = &mysql.Mysql{
			Address:      em.EntnichePush.Address,
			UserName:     em.EntnichePush.UserName,
			PassWord:     em.EntnichePush.Password,
			DBName:       em.EntnichePush.DbName,
			MaxOpenConns: em.EntnichePush.MaxOpenConns,
			MaxIdleConns: em.EntnichePush.MaxIdleConns,
		}
		EntnichePushMysql.Init()
	}
	if em.BaseService.Address != "" {
		logx.Info("--初始化 BaseServiceMysql tidb--")
		BaseServiceMysql = &mysql.Mysql{
			Address:      em.BaseService.Address,
			UserName:     em.BaseService.UserName,
			PassWord:     em.BaseService.Password,
			DBName:       em.BaseService.DbName,
			MaxOpenConns: em.BaseService.MaxOpenConns,
			MaxIdleConns: em.BaseService.MaxIdleConns,
		}
		BaseServiceMysql.Init()
	}
}

//
func RedisInit(rm *entity.RedisStuct) {
	//初始化 redis
	if len(rm.Addr) > 0 {
		logx.Info("--初始化 redis--")
		redis.InitRedisBySize(strings.Join(rm.Addr, ","), 100, 30, 300)
	}
}

//
func EsInit(es *entity.EsStruct) {
	//初始化 elasticsearch
	if es.Addr != "" {
		logx.Info("--初始化 elasticsearch--")
		elastic.InitElasticSize(es.Addr, es.Size)
	}
}