init.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. package init
  2. import (
  3. "app.yhyue.com/moapp/jyInfo/entity"
  4. "app.yhyue.com/moapp/jyInfo/rpc/consumer/internal/config"
  5. "app.yhyue.com/moapp/jyInfo/rpc/model"
  6. Oss "app.yhyue.com/moapp/jyInfo/rpc/model/oss"
  7. "flag"
  8. "github.com/zeromicro/go-zero/core/conf"
  9. "github.com/zeromicro/go-zero/core/logx"
  10. )
  11. var configFile = flag.String("cf", "etc/consumer.yaml", "the config file")
  12. var C config.Config
  13. //
  14. var logFile = flag.String("lf", "etc/logs.yaml", "the config file")
  15. var logc entity.Logc
  16. func init() {
  17. conf.MustLoad(*configFile, &C)
  18. //初始化msyql
  19. mm := C.MysqlMain
  20. model.InitMysql(&mm)
  21. //初始redis
  22. rm := C.Redis
  23. model.InitRedis(&rm)
  24. //初始nsq
  25. nm := C.Nsq
  26. model.InitNsqConfig(&nm)
  27. //初始SupplyTotal
  28. sm := C.SupplyTotal
  29. model.SupplyTotalConfig(sm)
  30. //初始特别行政地区
  31. pc := C.PublishCity
  32. model.PublishCityInfo(pc)
  33. //初始redis 时间
  34. sd := C.SupplyDay
  35. model.SupplyDay(sd)
  36. //初始es
  37. es := C.Es
  38. model.InitEs(&es)
  39. //初始化oss
  40. oss := C.Oss
  41. Oss.InitOss(oss.OssEndpoint, oss.OssAccessKeyId, oss.OssAccessKeySecret)
  42. //
  43. //初始化日志信息
  44. conf.MustLoad(*logFile, &logc)
  45. if len(logc.Level) > 0 {
  46. for _, v := range logc.Level {
  47. logx.MustSetup(logx.LogConf{
  48. Mode: logc.Mode,
  49. Path: logc.Path,
  50. Level: v,
  51. KeepDays: logc.KeepDays,
  52. })
  53. logx.Info(v, "--日志记录")
  54. }
  55. }
  56. }