init.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. package init
  2. import (
  3. MC "app.yhyue.com/moapp/jybase/common"
  4. "app.yhyue.com/moapp/jypkg/compatible"
  5. "app.yhyue.com/moapp/jypkg/middleground"
  6. "bp.jydev.jianyu360.cn/BaseService/jyMicroservices/jyBXBase/entity"
  7. "bp.jydev.jianyu360.cn/BaseService/jyMicroservices/jyBXBase/rpc/internal/config"
  8. "flag"
  9. _ "github.com/go-sql-driver/mysql"
  10. "github.com/zeromicro/go-zero/core/conf"
  11. )
  12. var configFile = flag.String("cf", "etc/bxbase.yaml", "the config file")
  13. var C config.Config
  14. var err error
  15. var dbFile = flag.String("df", "etc/db.yaml", "the db file")
  16. var DB config.Db
  17. var logFile = flag.String("lf", "etc/logs.yaml", "the logs file")
  18. var Middleground *middleground.Middleground
  19. var Compatible *compatible.Compatible
  20. func init() {
  21. //基本配置
  22. conf.MustLoad(*configFile, &C)
  23. //数据库配置
  24. conf.MustLoad(*dbFile, &DB)
  25. //初始mongodb
  26. MongoDBInit(&DB.Mongo)
  27. //初始化msyql
  28. MysqlInit(&DB.Mysql)
  29. //初始redis
  30. RedisInit(&DB.Redis)
  31. //初始es
  32. EsInit(&DB.Es)
  33. Middleground = middleground.NewMiddleground(C.Etcd.Hosts).
  34. //Middleground = middleground.NewMiddleground([]string{"192.168.3.149:2379"}).
  35. RegUserCenter(C.UserCenterKey).
  36. RegPowerCheckCenter(C.PowerCheckCenterKey).
  37. RegEntManageApplication(C.EntManageApplication)
  38. Compatible = compatible.NewCompatible(&Mgo, BaseServiceMysql, MainMysql, Middleground)
  39. //创建执行池 等待池
  40. num := MC.If(C.NewConcurrencyNum > 0, C.NewConcurrencyNum, 3).(int)
  41. entity.NewLimit(num)
  42. }