init.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  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 configFile = flag.String("cf", "etc/bxcore.yaml", "the config file")
  11. var C config.Config
  12. var err error
  13. //
  14. var dbFile = flag.String("df", "etc/db.yaml", "the db file")
  15. var DB config.Db
  16. //
  17. var logFile = flag.String("lf", "etc/logs.yaml", "the logs file")
  18. var logc entity.Logc
  19. func init() {
  20. //基本配置
  21. conf.MustLoad(*configFile, &C)
  22. //数据库配置
  23. conf.MustLoad(*dbFile, &DB)
  24. //初始mongodb
  25. MongoDBInit(&DB.Mongo)
  26. //初始化msyql
  27. MysqlInit(&DB.Mysql)
  28. //初始redis
  29. RedisInit(&DB.Redis)
  30. //初始es
  31. EsInit(&DB.Es)
  32. //初始化日志信息
  33. conf.MustLoad(*logFile, &logc)
  34. if len(logc.Level) > 0 {
  35. for _, v := range logc.Level {
  36. logx.MustSetup(logx.LogConf{
  37. Mode: logc.Mode,
  38. Path: logc.Path,
  39. Level: v,
  40. KeepDays: logc.KeepDays,
  41. })
  42. logx.Info(v, "--日志记录")
  43. }
  44. }
  45. //初始化标签
  46. LabelInit()
  47. }