init.go 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package init
  2. import (
  3. "bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/rpc/powercheck"
  4. "flag"
  5. "fmt"
  6. _ "github.com/go-sql-driver/mysql"
  7. "github.com/zeromicro/go-zero/core/conf"
  8. "github.com/zeromicro/go-zero/core/discov"
  9. "github.com/zeromicro/go-zero/zrpc"
  10. "jyBXCore/entity"
  11. "jyBXCore/rpc/internal/config"
  12. )
  13. var configFile = flag.String("cf", "etc/bxcore.yaml", "the config file")
  14. var C config.Config
  15. var err error
  16. var dbFile = flag.String("df", "etc/db.yaml", "the db file")
  17. var DB config.Db
  18. // 全文或附件搜索限制
  19. var SearchLimitKey = "jy_limitSearchText_new"
  20. var SearchLimitFlag = "jy_limitSearchText_%s"
  21. func init() {
  22. //基本配置
  23. conf.MustLoad(*configFile, &C)
  24. //数据库配置
  25. conf.MustLoad(*dbFile, &DB)
  26. //初始mongodb
  27. MongoDBInit(&DB.Mongo)
  28. //初始化msyql
  29. MysqlInit(&DB.Mysql)
  30. //初始redis
  31. RedisInit(&DB.Redis)
  32. //初始es
  33. EsInit(&DB.Es)
  34. //初始化标签
  35. LabelInit()
  36. //
  37. SearchLimitKey = fmt.Sprintf(C.LimitSearchText.LimitKey, "jy_limitSearchText")
  38. //初始化权益中台相关
  39. powersClient := zrpc.MustNewClient(zrpc.RpcClientConf{
  40. Etcd: discov.EtcdConf{
  41. Hosts: C.PowersEtcdConf.Etcd.Hosts,
  42. Key: C.PowersEtcdConf.Etcd.Key,
  43. },
  44. })
  45. entity.PowerCheck = powercheck.NewPowerCheck(powersClient)
  46. }