leadGeneration/public/db.go

package public

import (
	util "app.yhyue.com/moapp/jybase/common"
	elastic "app.yhyue.com/moapp/jybase/es"
	m "app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"app.yhyue.com/moapp/jybase/redis"
	"log"
)

var (
	DbConf        *dbConf
	MainMysql     *mysql.Mysql //主程序订单表
	UserAnalyseDb *mysql.Mysql //用户分析
	MQFW          m.MongodbSim
	Mgo_Log       m.MongodbSim
)

type dbConf struct {
	Mongodb struct {
		Main    *mgoConf
		Log     *mgoConf
		Ent     *mgoConf
		Qyfw    *mgoConf
		Bidding *mgoConf
	}
	Elasticsearch struct {
		Main *esConf
	}
	Redis struct {
		Main *redisConf
	}
	Mysql struct {
		Main   *mysqlConf //大会员&订单
		Active *mysqlConf //活跃用户
	}
}
type mgoConf struct {
	Address         string
	Size            int
	DbName          string
	ReplSet         string
	UserName        string
	Password        string
	Collection      string
	Collection_back string
}

type esConf struct {
	Version  string
	Address  string
	Size     int
	UserName string
	Password string
}
type redisConf struct {
	Address string
}
type mysqlConf struct {
	DbName       string
	Address      string
	UserName     string
	PassWord     string
	MaxOpenConns int
	MaxIdleConns int
}

func init() {
	util.ReadConfig("./db.json", &DbConf)
	if DbConf != nil {
		if DbConf.Elasticsearch.Main != nil {
			elastic.NewEs(DbConf.Elasticsearch.Main.Version, DbConf.Elasticsearch.Main.Address, DbConf.Elasticsearch.Main.Size, DbConf.Elasticsearch.Main.UserName, DbConf.Elasticsearch.Main.Password)
			log.Println("初始化 elasticsearch")
		}
		//初始化redis
		if DbConf.Redis.Main != nil {
			log.Println("[SYSTEM]初始化 redis")
			redis.InitRedisBySize(DbConf.Redis.Main.Address, 100, 30, 300)
		}
		//
		if DbConf.Mongodb.Main != nil {
			log.Println("[SYSTEM]初始化 mongodb main")
			MQFW = m.MongodbSim{
				MongodbAddr: DbConf.Mongodb.Main.Address,
				Size:        DbConf.Mongodb.Main.Size,
				DbName:      DbConf.Mongodb.Main.DbName,
				ReplSet:     DbConf.Mongodb.Main.ReplSet,
			}
			MQFW.InitPool()
		}
		if DbConf.Mongodb.Log != nil {
			log.Println("[SYSTEM]初始化 mongodb log")
			Mgo_Log = m.MongodbSim{
				MongodbAddr: DbConf.Mongodb.Log.Address,
				Size:        DbConf.Mongodb.Log.Size,
				DbName:      DbConf.Mongodb.Log.DbName,
				ReplSet:     DbConf.Mongodb.Log.ReplSet,
				UserName:    DbConf.Mongodb.Log.UserName,
				Password:    DbConf.Mongodb.Log.Password,
			}
			Mgo_Log.InitPool()
		}

		if DbConf.Mysql.Main != nil {
			log.Println("[SYSTEM]初始化 main mysql")
			MainMysql = &mysql.Mysql{
				Address:      DbConf.Mysql.Main.Address,
				UserName:     DbConf.Mysql.Main.UserName,
				PassWord:     DbConf.Mysql.Main.PassWord,
				DBName:       DbConf.Mysql.Main.DbName,
				MaxOpenConns: DbConf.Mysql.Main.MaxOpenConns,
				MaxIdleConns: DbConf.Mysql.Main.MaxIdleConns,
			}
			MainMysql.Init()
		}

		if DbConf.Mysql.Active != nil {
			log.Println("[SYSTEM]初始化 active mysql")
			UserAnalyseDb = &mysql.Mysql{
				Address:      DbConf.Mysql.Active.Address,
				UserName:     DbConf.Mysql.Active.UserName,
				PassWord:     DbConf.Mysql.Active.PassWord,
				DBName:       DbConf.Mysql.Active.DbName,
				MaxOpenConns: DbConf.Mysql.Active.MaxOpenConns,
				MaxIdleConns: DbConf.Mysql.Active.MaxIdleConns,
			}
			UserAnalyseDb.Init()
		}
	}
}