jyMicroservices/jyBXCore/rpc/init/init.go

package init

import (
	"bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/rpc/powercheck"
	"flag"
	"fmt"
	_ "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"
	"jyBXCore/entity"
	"jyBXCore/rpc/internal/config"
)

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

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

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

// 全文或附件搜索限制
var SearchLimitKey = "jy_limitSearchText_new"
var SearchLimitFlag = "jy_limitSearchText_%s"

func init() {
	//基本配置
	conf.MustLoad(*configFile, &C)
	//数据库配置
	conf.MustLoad(*dbFile, &DB)
	//初始mongodb
	MongoDBInit(&DB.Mongo)
	//初始化msyql
	MysqlInit(&DB.Mysql)
	//初始redis
	RedisInit(&DB.Redis)
	//初始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, "--日志记录")
		}
	}
	//初始化标签
	LabelInit()
	//
	SearchLimitKey = fmt.Sprintf(C.LimitSearchText.LimitKey, "jy_limitSearchText")
	//初始化权益中台相关
	powersClient := zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: C.PowersEtcdConf.Etcd.Hosts,
			Key:   C.PowersEtcdConf.Etcd.Key,
		},
	})
	entity.PowerCheck = powercheck.NewPowerCheck(powersClient)
}