powerCheckCenter/rpc/init/init.go

package init

import (
	"flag"
	"log"

	"app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"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/discov"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/zrpc"
)

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()
	}
}