init.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. package init
  2. import (
  3. MC "app.yhyue.com/moapp/jybase/common"
  4. "flag"
  5. "github.com/zeromicro/go-zero/core/conf"
  6. "github.com/zeromicro/go-zero/core/logx"
  7. "jyBXBase/api/internal/config"
  8. "jyBXBase/entity"
  9. "log"
  10. "regexp"
  11. )
  12. var configFile = flag.String("cf", "etc/bxbase-api.yaml", "the config file")
  13. var C config.Config
  14. //
  15. var logFile = flag.String("lf", "etc/logs.yaml", "the config file")
  16. var logc entity.Logc
  17. //
  18. var dbFile = flag.String("df", "etc/db.yaml", "the db file")
  19. var DB config.Db
  20. //
  21. var routesFile = flag.String("rf", "etc/routes.yaml", "the routes file")
  22. var Routes config.Routes
  23. var ExcludeUrl = []*regexp.Regexp{}
  24. func init() {
  25. conf.MustLoad(*configFile, &C)
  26. log.Println("初始化配置") //
  27. //初始化日志信息
  28. conf.MustLoad(*logFile, &logc)
  29. if len(logc.Level) > 0 {
  30. for _, v := range logc.Level {
  31. logx.MustSetup(logx.LogConf{
  32. Mode: logc.Mode,
  33. Path: logc.Path,
  34. Level: v,
  35. KeepDays: logc.KeepDays,
  36. })
  37. logx.Info(v, "--日志记录")
  38. }
  39. }
  40. //读取数据库配置
  41. conf.MustLoad(*dbFile, &DB)
  42. //初始mongodb
  43. MongoDBInit(&DB.Mongo)
  44. //读取路由配置
  45. conf.MustLoad(*routesFile, &Routes)
  46. if len(Routes.ExcludeRoute) > 0 {
  47. for _, v := range Routes.ExcludeRoute {
  48. ExcludeUrl = append(ExcludeUrl, regexp.MustCompile(MC.ObjToString(v)))
  49. }
  50. }
  51. }