f

app/main/api/internal/service/apirequestService.go

@@ -33,19 +33,33 @@ func generateAuthDateRange() string {
 }
 
 type ApiRequestService struct {
-	config              config.Config
-	featureModel        model.FeatureModel
-	productFeatureModel model.ProductFeatureModel
-	tianyuanapi         *tianyuanapi.Client
+	config                    config.Config
+	featureModel              model.FeatureModel
+	productFeatureModel       model.ProductFeatureModel
+	userFeatureWhitelistModel model.UserFeatureWhitelistModel
+	whitelistService          *WhitelistService
+	tianyuanapi               *tianyuanapi.Client
+	tianyuanapiCallLogService *TianyuanapiCallLogService
 }
 
-// NewApiRequestService 是一个构造函数，用于初始化 ApiRequestService
-func NewApiRequestService(c config.Config, featureModel model.FeatureModel, productFeatureModel model.ProductFeatureModel, tianyuanapi *tianyuanapi.Client) *ApiRequestService {
+// NewApiRequestService 构造函数
+func NewApiRequestService(
+	c config.Config,
+	featureModel model.FeatureModel,
+	productFeatureModel model.ProductFeatureModel,
+	userFeatureWhitelistModel model.UserFeatureWhitelistModel,
+	tianyuanapi *tianyuanapi.Client,
+	tianyuanapiCallLogService *TianyuanapiCallLogService,
+	whitelistService *WhitelistService,
+) *ApiRequestService {
 	return &ApiRequestService{
-		config:              c,
-		featureModel:        featureModel,
-		productFeatureModel: productFeatureModel,
-		tianyuanapi:         tianyuanapi,
+		config:                    c,
+		featureModel:              featureModel,
+		productFeatureModel:       productFeatureModel,
+		userFeatureWhitelistModel: userFeatureWhitelistModel,
+		tianyuanapi:               tianyuanapi,
+		tianyuanapiCallLogService: tianyuanapiCallLogService,
+		whitelistService:          whitelistService,
 	}
 }
 
@@ -61,23 +75,33 @@ type APIResponseData struct {
 func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([]byte, error) {
 	var ctx, cancel = context.WithCancel(context.Background())
 	defer cancel()
-    build := a.productFeatureModel.SelectBuilder().Where(squirrel.Eq{
-        "product_id": productID,
-    })
+
+	// 白名单：已下架模块直接返回占位成功，不调用天远 API
+	idCard := gjson.GetBytes(params, "id_card").String()
+	var whitelistedFeatureApiIds map[string]bool
+	if a.whitelistService != nil {
+		whitelistedFeatureApiIds, _ = a.whitelistService.GetWhitelistedFeatureApisByIdCard(ctx, idCard)
+	} else {
+		whitelistedFeatureApiIds = make(map[string]bool)
+	}
+
+	build := a.productFeatureModel.SelectBuilder().Where(squirrel.Eq{
+		"product_id": productID,
+	})
 	productFeatureList, findProductFeatureErr := a.productFeatureModel.FindAll(ctx, build, "")
 	if findProductFeatureErr != nil {
 		return nil, findProductFeatureErr
 	}
-    var featureIDs []string
-    isImportantMap := make(map[string]int64, len(productFeatureList))
+	var featureIDs []string
+	isImportantMap := make(map[string]int64, len(productFeatureList))
 	for _, pf := range productFeatureList {
-        featureIDs = append(featureIDs, pf.FeatureId)
-        isImportantMap[pf.FeatureId] = pf.IsImportant
+		featureIDs = append(featureIDs, pf.FeatureId)
+		isImportantMap[pf.FeatureId] = pf.IsImportant
 	}
 	if len(featureIDs) == 0 {
 		return nil, errors.New("featureIDs 是空的")
 	}
-    builder := a.featureModel.SelectBuilder().Where(squirrel.Eq{"id": featureIDs})
+	builder := a.featureModel.SelectBuilder().Where(squirrel.Eq{"id": featureIDs})
 	featureList, findFeatureErr := a.featureModel.FindAll(ctx, builder, "")
 	if findFeatureErr != nil {
 		return nil, findFeatureErr
@@ -109,12 +133,19 @@ func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([]
 				Success: false,
 			}
 			timestamp := time.Now().Format("2006-01-02 15:04:05")
+			// 白名单：已下架模块直接返回占位成功
+			if whitelistedFeatureApiIds[feature.ApiId] {
+				result.Success = true
+				result.Timestamp = timestamp
+				result.Data = []byte("null")
+				resultsCh <- result
+				return
+			}
 			var (
 				resp          json.RawMessage
 				preprocessErr error
 			)
-			// 若 isImportantMap[feature.ID] == 1，则表示需要在出错时重试
-            isImportant := isImportantMap[feature.Id] == 1
+			isImportant := isImportantMap[feature.Id] == 1
 			tryCount := 0
 			for {
 				tryCount++