jyMicroservices/jyBXBase/rpc/init/init.go

package init

import (
	"flag"
	_ "github.com/go-sql-driver/mysql"
	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/core/logx"
	"jyBXBase/entity"
	"jyBXBase/rpc/internal/config"
)

var configFile = flag.String("cf", "etc/bxbase.yaml", "the config file")
var C config.Config
var err error

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

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

func init() {
	//基本配置
	conf.MustLoad(*configFile, &C)
	//数据库配置
	conf.MustLoad(*dbFile, &DB)
	//初始mongodb
	MongoDBInit(&DB.MongoDB)
	//初始化msyql
	MysqlInit(&DB.MysqlMain)
	//初始redis
	RedisInit(&DB.RedisDB)
	//初始es
	EsInit(&DB.Es)
	//初始化日志信息
	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, "--日志记录")
		}
	}
}