temp

apps/api/internal/middleware/apiauthinterceptormiddleware.go

@@ -14,12 +14,15 @@ import (
 	"tianyuan-api/apps/sentinel/client/userproduct"
 	"tianyuan-api/apps/sentinel/client/whitelist"
 	"tianyuan-api/apps/sentinel/sentinel"
+	"tianyuan-api/apps/user/user"
+	"tianyuan-api/pkg/crypto"
 )
 
 type ApiAuthInterceptorMiddleware struct {
 	WhitelistRpc   sentinel.WhitelistClient
 	SecretRpc      sentinel.SecretClient
 	UserProductRpc sentinel.UserProductClient
+	UserRpc        user.UserClient
 	Rds            *redis.Redis
 }
 
@@ -27,11 +30,13 @@ func NewApiAuthInterceptorMiddleware(
 	whitelistRpc sentinel.WhitelistClient,
 	secretRpc sentinel.SecretClient,
 	userProductRpc sentinel.UserProductClient,
+	userRpc user.UserClient,
 	rds *redis.Redis) *ApiAuthInterceptorMiddleware {
 	return &ApiAuthInterceptorMiddleware{
 		WhitelistRpc:   whitelistRpc,
 		SecretRpc:      secretRpc,
 		UserProductRpc: userProductRpc,
+		UserRpc:        userRpc,
 		Rds:            rds,
 	}
 }
@@ -83,8 +88,21 @@ func (m *ApiAuthInterceptorMiddleware) Handle(next http.HandlerFunc) http.Handle
 			return
 		}
 
-		// 3、是否有开通该产品
 		userId := secrets.UserId
+
+		// 3、额度是否冻结
+		info, err := m.UserRpc.GetUserInfo(r.Context(), &user.UserInfoReq{UserId: userId})
+		if err != nil {
+			xhttp.JsonBaseResponseCtx(r.Context(), w, errors.New("系统错误，请联系管理员"))
+			return
+		}
+
+		if info.QuotaExceeded == 1 {
+			xhttp.JsonBaseResponseCtx(r.Context(), w, errors.New("账户余额不足，无法请求"))
+			return
+		}
+
+		// 4、是否有开通该产品
 		pathParts := strings.Split(r.URL.Path, "/")
 		productCode := pathParts[len(pathParts)-1]
 		userProductRedisKey := fmt.Sprintf("user_products:%d", userId)
@@ -106,7 +124,13 @@ func (m *ApiAuthInterceptorMiddleware) Handle(next http.HandlerFunc) http.Handle
 
 		// 将 userId 存入 context，供后续逻辑使用
 		ctx := context.WithValue(r.Context(), "userId", userId)
-		ctx = context.WithValue(r.Context(), "secretKey", secrets.AesKey)
+		ctx = context.WithValue(ctx, "secretKey", secrets.AesKey)
+		ctx = context.WithValue(ctx, "productCode", productCode)
+
+		// 生成流水号
+		transactionID := crypto.GenerateTransactionID()
+		ctx = context.WithValue(ctx, "transactionID", transactionID)
+
 		next(w, r.WithContext(ctx))
 	}
 }