messageCenter/rpc/messagecenter/init/init.go

package init

import (
	"app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"app.yhyue.com/moapp/jybase/redis"
	"bp.jydev.jianyu360.cn/SocialPlatform/messageCenter/entity"
	"bp.jydev.jianyu360.cn/SocialPlatform/messageCenter/rpc/messagecenter/internal/config"
	"flag"
	_ "github.com/go-sql-driver/mysql"
	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/core/logx"
	"log"
	"strings"
)

var (
	MainMysql    *mysql.Mysql
	BaseMysql    *mysql.Mysql
	SurvivalTime int
	Mgo          mongodb.MongodbSim
)
var configF = flag.String("ff", "etc/messagecenter.yaml", "the config file")
var C config.Config

var logFile = flag.String("lf", "etc/logs.yaml", "the config file")
var logc entity.Logc

var dbFile = flag.String("df", "etc/db.yaml", "the db file")
var DB config.Db

func init() {
	conf.MustLoad(*configF, &C)
	//初始化日志信息
	conf.MustLoad(*logFile, &logc)
	if len(logc.Level) > 0 {
		for _, v := range logc.Level {
			logx.MustSetup(logx.LogConf{
				Mode:     logc.Mode,
				Path:     logc.Path,
				Level:    v,
				KeepDays: logc.KeepDays,
			})
			logx.Info(v, "--日志记录")
		}
	}
	conf.MustLoad(*dbFile, &DB)
	//初始化mysql
	if DB.Mysql.Main.Address != "" {
		log.Println("--初始化 mysql--")
		MainMysql = &mysql.Mysql{
			Address:      DB.Mysql.Main.Address,
			UserName:     DB.Mysql.Main.UserName,
			PassWord:     DB.Mysql.Main.Password,
			DBName:       DB.Mysql.Main.DbName,
			MaxOpenConns: DB.Mysql.Main.MaxOpenConns,
			MaxIdleConns: DB.Mysql.Main.MaxIdleConns,
		}
		MainMysql.Init()
	}
	if DB.Mysql.BaseService.Address != "" {
		log.Println("--初始化 mysql--")
		BaseMysql = &mysql.Mysql{
			Address:      DB.Mysql.BaseService.Address,
			UserName:     DB.Mysql.BaseService.UserName,
			PassWord:     DB.Mysql.BaseService.Password,
			DBName:       DB.Mysql.BaseService.DbName,
			MaxOpenConns: DB.Mysql.BaseService.MaxOpenConns,
			MaxIdleConns: DB.Mysql.BaseService.MaxIdleConns,
		}
		BaseMysql.Init()
	}
	//初始化数据库配置
	rm := DB.Redis
	if len(rm.Addr) > 0 {
		logx.Info("--初始化 redis--")
		redis.InitRedisBySize(strings.Join(rm.Addr, ","), 100, 30, 300)
	}
	if DB.Mongo.Main.Address != "" {
		logx.Info("--初始化 mongodb--")
		Mgo = mongodb.MongodbSim{
			MongodbAddr: DB.Mongo.Main.Address,
			Size:        DB.Mongo.Main.Size,
			DbName:      DB.Mongo.Main.DbName,
		}
		Mgo.InitPool()
	}
	SurvivalTime = C.SurvivalTime
}