biService/vipTask/config/config.go

package config

import (
	. "app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"flag"
	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/core/logx"
)

var UserMgo *MongodbSim
var DbConf *dbConf
var JianYu *mysql.Mysql
var Subjectdb *mysql.Mysql
var logc Logc

type dbConf struct {
	User          *MgoConf
	InitConfig    []*ExpireConfig
	DailyConfig   []*ExpireConfig
	StatisticTime string
	JianYu        *MysqlConf
	Subjectdb     *MysqlConf
}
type MysqlConf struct {
	DbName       string
	Address      string
	UserName     string
	PassWord     string
	MaxOpenConns int
	MaxIdleConns int
}
type ExpireConfig struct {
	CycleDays int64
	EndDays   int64
	StartDays int64
}
type MgoConf struct {
	Address  string
	Size     int
	DbName   string
	UserName string
	Password string
}

// 日志信息
type Logc struct {
	Mode     string
	Path     string
	Level    []string
	KeepDays int
}

var logFile = flag.String("lf", "./logs.yaml", "the config file")
var configFile = flag.String("rf", "./config.yaml", "the config file")

func init() {
	//util.ReadConfig(&DbConf)
	//初始化日志信息
	//日志初始化
	conf.MustLoad(*logFile, &logc)
	conf.MustLoad(*configFile, &DbConf)
	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, "--日志记录")
		}
	}
	//初始化mongo用户库
	UserMgo = &MongodbSim{
		MongodbAddr: DbConf.User.Address,
		Size:        DbConf.User.Size,
		DbName:      DbConf.User.DbName,
		UserName:    DbConf.User.UserName,
		Password:    DbConf.User.Password,
	}
	UserMgo.InitPool()
	JianYu = &mysql.Mysql{
		Address:      DbConf.JianYu.Address,
		UserName:     DbConf.JianYu.UserName,
		PassWord:     DbConf.JianYu.PassWord,
		DBName:       DbConf.JianYu.DbName,
		MaxOpenConns: DbConf.JianYu.MaxOpenConns,
		MaxIdleConns: DbConf.JianYu.MaxIdleConns,
	}
	JianYu.Init()
	Subjectdb = &mysql.Mysql{
		Address:      DbConf.Subjectdb.Address,
		UserName:     DbConf.Subjectdb.UserName,
		PassWord:     DbConf.Subjectdb.PassWord,
		DBName:       DbConf.Subjectdb.DbName,
		MaxOpenConns: DbConf.Subjectdb.MaxOpenConns,
		MaxIdleConns: DbConf.Subjectdb.MaxIdleConns,
	}
	Subjectdb.Init()
}