init.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. package init
  2. import (
  3. "flag"
  4. "fmt"
  5. _ "github.com/go-sql-driver/mysql"
  6. "github.com/zeromicro/go-zero/core/conf"
  7. "github.com/zeromicro/go-zero/core/logx"
  8. "jyBXCore/entity"
  9. "jyBXCore/rpc/internal/config"
  10. )
  11. var configFile = flag.String("cf", "etc/bxcore.yaml", "the config file")
  12. var C config.Config
  13. var err error
  14. //
  15. var dbFile = flag.String("df", "etc/db.yaml", "the db file")
  16. var DB config.Db
  17. //
  18. var logFile = flag.String("lf", "etc/logs.yaml", "the logs file")
  19. var logc entity.Logc
  20. //全文或附件搜索限制
  21. var SearchLimitKey = "jy_limitSearchText_new"
  22. var SearchLimitFlag = "jy_limitSearchText_%s"
  23. func init() {
  24. //基本配置
  25. conf.MustLoad(*configFile, &C)
  26. //数据库配置
  27. conf.MustLoad(*dbFile, &DB)
  28. //初始mongodb
  29. MongoDBInit(&DB.Mongo)
  30. //初始化msyql
  31. MysqlInit(&DB.Mysql)
  32. //初始redis
  33. RedisInit(&DB.Redis)
  34. //初始es
  35. EsInit(&DB.Es)
  36. //初始化日志信息
  37. conf.MustLoad(*logFile, &logc)
  38. if len(logc.Level) > 0 {
  39. for _, v := range logc.Level {
  40. logx.MustSetup(logx.LogConf{
  41. Mode: logc.Mode,
  42. Path: logc.Path,
  43. Level: v,
  44. KeepDays: logc.KeepDays,
  45. })
  46. logx.Info(v, "--日志记录")
  47. }
  48. }
  49. //初始化标签
  50. LabelInit()
  51. //
  52. SearchLimitKey = fmt.Sprintf(C.LimitSearchText.LimitKey, "jy_limitSearchText")
  53. }