jy_docs/rpc/partnerlib/init/init.go

package init

import (
	"app.yhyue.com/moapp/jy_docs/rpc/partnerlib/internal/config"
	"app.yhyue.com/moapp/jy_docs/services/partner"
	jyDocsRpcUtil "app.yhyue.com/moapp/jy_docs/services/util"
	"flag"
	"github.com/zeromicro/go-zero/core/conf"
)

var (
	configFile    = flag.String("c", "etc/partner.yaml", "the partner file")
	interfaceFile = flag.String("i", "etc/interface.yaml", "the interface file")
	crontabFile   = flag.String("cron", "etc/crontab.yaml", "the crontab file")
	C             config.Config
	I             config.InterfaceInfo
	Cron          config.CrontabInfo
)

func init() {
	conf.MustLoad(*configFile, &C)
	conf.MustLoad(*interfaceFile, &I)
	conf.MustLoad(*crontabFile, &Cron)
	if C.JyDocsMysqlDB.DataSourceName != "" && C.EsConfig.Addr != "" {
		jyDocsRpcUtil.InitDB(C.JyDocsMysqlDB.DataSourceName, C.JyDocsMysqlDB.DriverName, C.JyDocsMysqlDB.MaxOpenConn, C.JyDocsMysqlDB.MaxIdleConn)
		jyDocsRpcUtil.InitEs(C.EsConfig.Addr, C.EsConfig.Pool, C.EsConfig.UserName, C.EsConfig.Password)
		jyDocsRpcUtil.InitRedis(C.Redis)
		jyDocsRpcUtil.InitMongo(C.Mongo)
	}
	//初始化 分类对应关系
	go partner.InitDocClass()
}