package init import ( "flag" "log" "strings" "app.yhyue.com/moapp/jybase/mongodb" "app.yhyue.com/moapp/jybase/mysql" "app.yhyue.com/moapp/jybase/redis" "bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/entity" "bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/rpc/internal/config" _ "github.com/go-sql-driver/mysql" "github.com/zeromicro/go-zero/core/conf" "github.com/zeromicro/go-zero/core/logx" ) var logFile = flag.String("lf", "etc/log.yaml", "the config file") var configF = flag.String("ff", "etc/powercheck.yaml", "the config file") var logc entity.Logc var C config.Config 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, "--日志记录") } } //初始化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() } nn := C.BaseMysqlMain if nn.Address != "" { log.Println("--初始化 mysql--") entity.BaseMysql = &mysql.Mysql{ Address: nn.Address, UserName: nn.UserName, PassWord: nn.PassWord, DBName: nn.DbName, MaxOpenConns: nn.MaxOpenConns, MaxIdleConns: nn.MaxIdleConns, } entity.BaseMysql.Init() } //初始化 mongodb if C.Mongo.Main.Address != "" { log.Println("初始化 mongo main") entity.MgoJy = &mongodb.MongodbSim{ MongodbAddr: C.Mongo.Main.Address, Size: C.Mongo.Main.Size, DbName: C.Mongo.Main.DbName, } entity.MgoJy.InitPool() } // log.Println("初始化 redis") redis.InitRedisBySize(strings.Join(C.RedisAddrees, ","), 100, 30, 300) }