init.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. package init
  2. import (
  3. "app.yhyue.com/moapp/jy_docs/rpc/partnerlib/internal/config"
  4. "app.yhyue.com/moapp/jy_docs/services/partner"
  5. jyDocsRpcUtil "app.yhyue.com/moapp/jy_docs/services/util"
  6. "app.yhyue.com/moapp/jypkg/common/src/qfw/util/jy"
  7. "flag"
  8. "github.com/zeromicro/go-zero/core/conf"
  9. )
  10. var (
  11. configFile = flag.String("c", "etc/partnerlib.yaml", "the partnerlib file")
  12. interfaceFile = flag.String("i", "etc/interface.yaml", "the interface file")
  13. crontabFile = flag.String("cron", "etc/crontab.yaml", "the crontab file")
  14. warnFile = flag.String("warn", "etc/warn.yaml", "the warn file")
  15. C config.Config
  16. I config.InterfaceInfo
  17. Cron config.CrontabInfo
  18. Warn config.Warn
  19. )
  20. func init() {
  21. conf.MustLoad(*configFile, &C)
  22. conf.MustLoad(*interfaceFile, &I)
  23. conf.MustLoad(*crontabFile, &Cron)
  24. conf.MustLoad(*warnFile, &Warn)
  25. if C.JyDocsMysqlDB.DataSourceName != "" && C.EsConfig.Addr != "" {
  26. jyDocsRpcUtil.InitDB(C.JyDocsMysqlDB.DataSourceName, C.JyDocsMysqlDB.DriverName, C.JyDocsMysqlDB.MaxOpenConn, C.JyDocsMysqlDB.MaxIdleConn)
  27. jyDocsRpcUtil.InitESV7(C.EsConfig.Addr, C.EsConfig.Pool, C.EsConfig.UserName, C.EsConfig.Password)
  28. jyDocsRpcUtil.InitRedis(C.RedisAddr)
  29. jyDocsRpcUtil.InitMongo(C.Mongo)
  30. jyDocsRpcUtil.InitOss(C.FileSystemConf)
  31. }
  32. jy.OrderCodePoolInit()
  33. //初始化 分类对应关系
  34. go partner.InitDocClass()
  35. }