init.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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. "jyBXCore/api/internal/config"
  8. "jyBXCore/entity"
  9. "log"
  10. "regexp"
  11. )
  12. var configFile = flag.String("cf", "etc/bxcore-api.yaml", "the config file")
  13. var C config.Config
  14. //
  15. var logFile = flag.String("lf", "etc/logs.yaml", "the log file")
  16. var logc entity.Logc
  17. //
  18. var dbFile = flag.String("df", "etc/db.json", "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. log.Println("初始化配置")
  26. //读取基础配置
  27. conf.MustLoad(*configFile, &C)
  28. //读取数据库配置
  29. conf.MustLoad(*dbFile, &DB)
  30. //初始mongodb
  31. MongoDBInit(&DB)
  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. conf.MustLoad(*routesFile, &Routes)
  47. if len(Routes.ExcludeRoute) > 0 {
  48. for _, v := range Routes.ExcludeRoute {
  49. ExcludeUrl = append(ExcludeUrl, regexp.MustCompile(MC.ObjToString(v)))
  50. }
  51. }
  52. }