init.go 890 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. package init
  2. import (
  3. "flag"
  4. _ "github.com/go-sql-driver/mysql"
  5. "github.com/zeromicro/go-zero/core/conf"
  6. "github.com/zeromicro/go-zero/core/logx"
  7. "jyBXCore/entity"
  8. "jyBXCore/rpc/internal/config"
  9. )
  10. var configF = flag.String("f", "etc/bxsubscribe.yaml", "the config file")
  11. var C config.Config
  12. var err error
  13. //
  14. var logFile = flag.String("lf", "etc/logs.yaml", "the config file")
  15. var logc entity.Logc
  16. func init() {
  17. conf.MustLoad(*configF, &C)
  18. //初始mongodb
  19. MongoDBInit(&C.MongoDB)
  20. //初始化msyql
  21. MysqlInit(&C.Mysql)
  22. //初始redis
  23. RedisInit(&C.RedisDB)
  24. //初始es
  25. EsInit(&C.Es)
  26. //初始化日志信息
  27. conf.MustLoad(*logFile, &logc)
  28. if len(logc.Level) > 0 {
  29. for _, v := range logc.Level {
  30. logx.MustSetup(logx.LogConf{
  31. Mode: logc.Mode,
  32. Path: logc.Path,
  33. Level: v,
  34. KeepDays: logc.KeepDays,
  35. })
  36. logx.Info(v, "--日志记录")
  37. }
  38. }
  39. }