first commit

apps/api/internal/svc/servicecontext.go

@@ -0,0 +1,50 @@
+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 中
+	}
+}