v1.1

app/main/api/internal/logic/query/query_common.go

@@ -0,0 +1,157 @@
+package query
+
+import (
+	"context"
+	"database/sql"
+	"encoding/hex"
+	"encoding/json"
+	"fmt"
+	"ycc-server/common/xerr"
+	"ycc-server/pkg/lzkit/crypto"
+	"ycc-server/pkg/lzkit/lzUtils"
+
+	"github.com/jinzhu/copier"
+	"github.com/pkg/errors"
+
+	"ycc-server/app/main/api/internal/svc"
+	"ycc-server/app/main/api/internal/types"
+	"ycc-server/app/main/model"
+)
+
+func ProcessQueryData(queryData sql.NullString, target *[]types.QueryItem, key []byte) error {
+	queryDataStr := lzUtils.NullStringToString(queryData)
+	if queryDataStr == "" {
+		return nil
+	}
+	decryptedData, decryptErr := crypto.AesDecrypt(queryDataStr, key)
+	if decryptErr != nil {
+		return decryptErr
+	}
+	var decryptedArray []map[string]interface{}
+	unmarshalErr := json.Unmarshal(decryptedData, &decryptedArray)
+	if unmarshalErr != nil {
+		return unmarshalErr
+	}
+	if len(*target) == 0 {
+		*target = make([]types.QueryItem, len(decryptedArray))
+	}
+	for i := 0; i < len(decryptedArray); i++ {
+		(*target)[i].Data = decryptedArray[i]
+	}
+	return nil
+}
+
+func ProcessQueryParams(QueryParams string, target *map[string]interface{}, key []byte) error {
+	decryptedData, decryptErr := crypto.AesDecrypt(QueryParams, key)
+	if decryptErr != nil {
+		return decryptErr
+	}
+	unmarshalErr := json.Unmarshal(decryptedData, target)
+	if unmarshalErr != nil {
+		return unmarshalErr
+	}
+	return nil
+}
+
+func UpdateFeatureAndProductFeature(ctx context.Context, svcCtx *svc.ServiceContext, productID string, target *[]types.QueryItem) error {
+	for i := len(*target) - 1; i >= 0; i-- {
+		queryItem := &(*target)[i]
+		data, ok := queryItem.Data.(map[string]interface{})
+		if !ok {
+			return fmt.Errorf("queryItem.Data 必须是 map[string]interface{} 类型")
+		}
+		apiID, ok := data["apiID"].(string)
+		if !ok {
+			return fmt.Errorf("queryItem.Data 中的 apiID 必须是字符串类型")
+		}
+		feature, err := svcCtx.FeatureModel.FindOneByApiId(ctx, apiID)
+		if err != nil {
+			*target = append((*target)[:i], (*target)[i+1:]...)
+			continue
+		}
+		builder := svcCtx.ProductFeatureModel.SelectBuilder().Where("product_id = ?", productID)
+		productFeatures, err := svcCtx.ProductFeatureModel.FindAll(ctx, builder, "")
+		if err != nil {
+			return fmt.Errorf("查询 ProductFeatureModel 错误: %v", err)
+		}
+		sort := 0
+		for _, pf := range productFeatures {
+			if pf.FeatureId == feature.Id {
+				sort = int(pf.Sort)
+				break
+			}
+		}
+		featureData := map[string]interface{}{
+			"featureName": feature.Name,
+			"sort":        sort,
+		}
+		queryItem.Feature = featureData
+	}
+	return nil
+}
+
+func BuildEncryptedQuery(ctx context.Context, svcCtx *svc.ServiceContext, queryModel *model.Query) (string, error) {
+	var query types.Query
+	query.CreateTime = queryModel.CreateTime.Format("2006-01-02 15:04:05")
+	query.UpdateTime = queryModel.UpdateTime.Format("2006-01-02 15:04:05")
+	secretKey := svcCtx.Config.Encrypt.SecretKey
+	key, decodeErr := hex.DecodeString(secretKey)
+	if decodeErr != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取AES解密解药失败, %v", decodeErr)
+	}
+	processParamsErr := ProcessQueryParams(queryModel.QueryParams, &query.QueryParams, key)
+	if processParamsErr != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告参数处理失败: %v", processParamsErr)
+	}
+	processErr := ProcessQueryData(queryModel.QueryData, &query.QueryData, key)
+	if processErr != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", processErr)
+	}
+	updateErr := UpdateFeatureAndProductFeature(ctx, svcCtx, queryModel.ProductId, &query.QueryData)
+	if updateErr != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", updateErr)
+	}
+	err := copier.Copy(&query, queryModel)
+	if err != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结构体复制失败, %v", err)
+	}
+	product, err := svcCtx.ProductModel.FindOne(ctx, queryModel.ProductId)
+	if err != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取商品信息失败, %v", err)
+	}
+	query.Product = product.ProductEn
+	query.ProductName = product.ProductName
+	queryBytes, marshalErr := json.Marshal(query)
+	if marshalErr != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 序列化查询结果失败: %v", marshalErr)
+	}
+	encryptedQuery, encryptErr := crypto.AesEncrypt(queryBytes, key)
+	if encryptErr != nil {
+		return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 加密查询结果失败: %v", encryptErr)
+	}
+	return encryptedQuery, nil
+}
+
+func IsOrderAgent(ctx context.Context, svcCtx *svc.ServiceContext, userId string, orderId string) (bool, error) {
+	agent, err := svcCtx.AgentModel.FindOneByUserId(ctx, userId)
+	if err != nil {
+		if errors.Is(err, model.ErrNotFound) {
+			return false, nil
+		}
+		return false, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err)
+	}
+	if agent == nil {
+		return false, nil
+	}
+	agentOrder, err := svcCtx.AgentOrderModel.FindOneByOrderId(ctx, orderId)
+	if err != nil {
+		if errors.Is(err, model.ErrNotFound) {
+			return false, nil
+		}
+		return false, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理订单失败, %v", err)
+	}
+	if agentOrder == nil {
+		return false, nil
+	}
+	return agentOrder.AgentId == agent.Id, nil
+}