service_context.go 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
  1. package svc
  2. import (
  3. "github.com/bwmarrin/snowflake"
  4. "github.com/zeromicro/go-zero/core/collection"
  5. "gorm.io/gorm"
  6. "time"
  7. "wechat-api/database"
  8. "wechat-api/database/dao/wechat/query"
  9. "wechat-api/internal/config"
  10. "wechat-api/internal/middleware"
  11. "wechat-api/internal/pkg/wechat_ws"
  12. "wechat-api/mongo_model"
  13. "github.com/redis/go-redis/v9"
  14. "wechat-api/ent"
  15. _ "wechat-api/ent/runtime"
  16. "github.com/suyuan32/simple-admin-core/rpc/coreclient"
  17. "github.com/zeromicro/go-zero/core/logx"
  18. "github.com/casbin/casbin/v2"
  19. "github.com/zeromicro/go-zero/rest"
  20. "github.com/zeromicro/go-zero/zrpc"
  21. )
  22. type ServiceContext struct {
  23. Config config.Config
  24. Casbin *casbin.Enforcer
  25. Authority rest.Middleware
  26. OpenAuthority rest.Middleware
  27. Miniprogram rest.Middleware
  28. DB *ent.Client
  29. WechatDB *gorm.DB
  30. WechatQ *query.Query
  31. CoreRpc coreclient.Core
  32. Rds redis.UniversalClient
  33. WechatWs map[string]*wechat_ws.WechatWsClient
  34. Cache *collection.Cache
  35. NodeID *snowflake.Node
  36. MongoModel *mongo_model.AllMongoModel
  37. }
  38. func NewServiceContext(c config.Config) *ServiceContext {
  39. rds := c.RedisConf.MustNewUniversalRedis()
  40. cbn := c.CasbinConf.MustNewCasbinWithOriginalRedisWatcher(c.CasbinDatabaseConf.Type, c.CasbinDatabaseConf.GetDSN(), c.RedisConf)
  41. db := ent.NewClient(
  42. ent.Log(logx.Info), // logger
  43. ent.Driver(c.DatabaseConf.NewNoCacheDriver()),
  44. ent.Debug(), // debug mode
  45. )
  46. // 初始化 MongoDB 客户端
  47. all_mongo_model := mongo_model.SetupMongoModel(c.FastgptMongoConf.Url, c.FastgptMongoConf.DBName)
  48. // gorm 数据库连接
  49. wechatDb, err := database.InitWechatDB(c.DatabaseConf, c.Mode)
  50. if err != nil {
  51. logx.Error(err)
  52. panic("gorm 数据库连接失败")
  53. }
  54. wechatQ := query.Use(wechatDb)
  55. coreRpc := coreclient.NewCore(zrpc.NewClientIfEnable(c.CoreRpc))
  56. // 初始化微信ws客户端
  57. // todo 现在配置是从 config.yaml中读取的,后续需要改成从数据库中读取,以便匹配不同的微信号
  58. wechatWs := make(map[string]*wechat_ws.WechatWsClient)
  59. for _, ws := range c.WebSocket {
  60. if ws.Type == "wechat" {
  61. client, err := wechat_ws.NewWechatWsClient(ws.Url, ws.Name, ws.Type)
  62. if err != nil {
  63. logx.Error(err)
  64. } else {
  65. logx.Info("建立ws client成功~", ws.Name)
  66. go client.ReadPump()
  67. go client.WritePump()
  68. wechatWs[ws.Name] = client
  69. }
  70. }
  71. }
  72. cache, err := collection.NewCache(time.Hour * 24 * 365)
  73. if err != nil {
  74. logx.Error(err)
  75. panic("本地缓存实例失败")
  76. }
  77. node, err := snowflake.NewNode(1)
  78. if err != nil {
  79. logx.Error(err)
  80. panic("雪花算法实例失败")
  81. }
  82. return &ServiceContext{
  83. Config: c,
  84. Authority: middleware.NewAuthorityMiddleware(cbn, rds, coreRpc).Handle,
  85. OpenAuthority: middleware.NewOpenAuthorityMiddleware(db, rds, c).Handle,
  86. Miniprogram: middleware.NewMiniprogramMiddleware(cbn, rds, coreRpc, c).Handle,
  87. DB: db,
  88. WechatDB: wechatDb,
  89. WechatQ: wechatQ,
  90. CoreRpc: coreRpc,
  91. Rds: rds,
  92. WechatWs: wechatWs,
  93. Cache: cache,
  94. NodeID: node,
  95. MongoModel: all_mongo_model,
  96. }
  97. }