package init import ( "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 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) //初始化mysql mm := C.MysqlMain if mm.Address != "" { log.Println("--初始化 mysql--") entity.Mysql = &mysql.Mysql{ Address: mm.Address, UserName: mm.UserName, PassWord: mm.PassWord, DBName: mm.DbName, MaxOpenConns: mm.MaxOpenConns, MaxIdleConns: mm.MaxIdleConns, } entity.Mysql.Init() } //初始化数据库配置 conf.MustLoad(*dbFile, &DB) rm := DB.Redis if len(rm.Addr) > 0 { logx.Info("--初始化 redis--") redis.InitRedisBySize(strings.Join(rm.Addr, ","), 100, 30, 300) } //初始化日志信息 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, "--日志记录") } } entity.SurvivalTime = C.SurvivalTime }