init.go 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  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. "jyBXBase/entity"
  8. "jyBXBase/rpc/internal/config"
  9. )
  10. var configFile = flag.String("cf", "etc/bxbase.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(*logFile, &logc)
  22. if len(logc.Level) > 0 {
  23. for _, v := range logc.Level {
  24. logx.MustSetup(logx.LogConf{
  25. Mode: logc.Mode,
  26. Path: logc.Path,
  27. Level: v,
  28. KeepDays: logc.KeepDays,
  29. })
  30. logx.Info(v, "--日志记录")
  31. }
  32. }
  33. //基本配置
  34. conf.MustLoad(*configFile, &C)
  35. //数据库配置
  36. conf.MustLoad(*dbFile, &DB)
  37. //初始mongodb
  38. MongoDBInit(&DB.Mongo)
  39. //初始化msyql
  40. MysqlInit(&DB.Mysql)
  41. //初始redis
  42. RedisInit(&DB.Redis)
  43. //初始es
  44. EsInit(&DB.Es)
  45. }