51 lines
1.9 KiB
Go
51 lines
1.9 KiB
Go
|
package svc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/zeromicro/go-queue/kq"
|
||
|
|
"github.com/zeromicro/go-zero/core/stores/redis"
|
||
|
|
"github.com/zeromicro/go-zero/rest"
|
||
|
|
"github.com/zeromicro/go-zero/zrpc"
|
||
|
|
"tianyuan-api/apps/api/internal/config"
|
||
|
|
"tianyuan-api/apps/api/internal/middleware"
|
||
|
|
"tianyuan-api/apps/api/internal/service"
|
||
|
|
"tianyuan-api/apps/sentinel/sentinel"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ServiceContext struct {
|
||
|
|
Config config.Config
|
||
|
|
ApiAuthInterceptor rest.Middleware
|
||
|
|
Redis *redis.Redis
|
||
|
|
WhitelistRpc sentinel.WhitelistClient
|
||
|
|
SecretRpc sentinel.SecretClient
|
||
|
|
ProductRpc sentinel.ProductClient
|
||
|
|
UserProductRpc sentinel.UserProductClient
|
||
|
|
KqPusherClient *kq.Pusher
|
||
|
|
WestDexService *service.WestDexService
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewServiceContext(c config.Config) *ServiceContext {
|
||
|
|
redisConf := redis.RedisConf{
|
||
|
|
Host: c.CacheRedis[0].Host,
|
||
|
|
Pass: c.CacheRedis[0].Pass,
|
||
|
|
Type: c.CacheRedis[0].Type, // Redis 节点类型，如 "node"
|
||
|
|
}
|
||
|
|
// 使用 MustNewRedis 来初始化 Redis 客户端
|
||
|
|
rds := redis.MustNewRedis(redisConf)
|
||
|
|
|
||
|
|
productRpc := sentinel.NewProductClient(zrpc.MustNewClient(c.SentinelRpc).Conn())
|
||
|
|
userProductRpc := sentinel.NewUserProductClient(zrpc.MustNewClient(c.SentinelRpc).Conn())
|
||
|
|
whitelistRpc := sentinel.NewWhitelistClient(zrpc.MustNewClient(c.SentinelRpc).Conn())
|
||
|
|
secretRpc := sentinel.NewSecretClient(zrpc.MustNewClient(c.SentinelRpc).Conn())
|
||
|
|
return &ServiceContext{
|
||
|
|
Config: c,
|
||
|
|
Redis: rds,
|
||
|
|
WhitelistRpc: whitelistRpc,
|
||
|
|
SecretRpc: secretRpc,
|
||
|
|
ProductRpc: productRpc,
|
||
|
|
UserProductRpc: userProductRpc,
|
||
|
|
ApiAuthInterceptor: middleware.NewApiAuthInterceptorMiddleware(whitelistRpc, secretRpc, userProductRpc, rds).Handle,
|
||
|
|
KqPusherClient: kq.NewPusher(c.KqPusherConf.Brokers, c.KqPusherConf.Topic),
|
||
|
|
WestDexService: service.NewWestDexService(c.WestConfig.Key, c.WestConfig.SecretId), // 假设你将密钥和 ID 配置在 config 中
|
||
|
|
}
|
||
|
|
}
|