init.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. package init
  2. import (
  3. "app.yhyue.com/moapp/jypkg/compatible"
  4. "app.yhyue.com/moapp/jypkg/middleground"
  5. "flag"
  6. _ "github.com/go-sql-driver/mysql"
  7. "github.com/zeromicro/go-zero/core/conf"
  8. "github.com/zeromicro/go-zero/core/logx"
  9. "jyBXBase/entity"
  10. "jyBXBase/rpc/internal/config"
  11. )
  12. var configFile = flag.String("cf", "etc/bxbase.yaml", "the config file")
  13. var C config.Config
  14. var err error
  15. //
  16. var dbFile = flag.String("df", "etc/db.yaml", "the db file")
  17. var DB config.Db
  18. //
  19. var logFile = flag.String("lf", "etc/logs.yaml", "the logs file")
  20. var logc entity.Logc
  21. var Middleground *middleground.Middleground
  22. var Compatible *compatible.Compatible
  23. func init() {
  24. //初始化日志信息
  25. conf.MustLoad(*logFile, &logc)
  26. if len(logc.Level) > 0 {
  27. for _, v := range logc.Level {
  28. logx.MustSetup(logx.LogConf{
  29. Mode: logc.Mode,
  30. Path: logc.Path,
  31. Level: v,
  32. KeepDays: logc.KeepDays,
  33. })
  34. logx.Info(v, "--日志记录")
  35. }
  36. }
  37. //基本配置
  38. conf.MustLoad(*configFile, &C)
  39. //数据库配置
  40. conf.MustLoad(*dbFile, &DB)
  41. //初始mongodb
  42. MongoDBInit(&DB.Mongo)
  43. //初始化msyql
  44. MysqlInit(&DB.Mysql)
  45. //初始redis
  46. RedisInit(&DB.Redis)
  47. //初始es
  48. EsInit(&DB.Es)
  49. Middleground = middleground.NewMiddleground(C.Etcd.Hosts).
  50. //Middleground = middleground.NewMiddleground([]string{"192.168.3.149:2379"}).
  51. RegUserCenter(C.UserCenterKey).
  52. RegPowerCheckCenter(C.PowerCheckCenterKey).
  53. RegEntManageApplication(C.EntManageApplication)
  54. Compatible = compatible.NewCompatible(&Mgo, BaseServiceMysql, MainMysql, Middleground)
  55. }