451 lines
14 KiB
Go
451 lines
14 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"encoding/hex"
|
||
"encoding/json"
|
||
"strings"
|
||
|
||
"github.com/google/uuid"
|
||
"github.com/pkg/errors"
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
||
"qnc-server/app/main/api/internal/config"
|
||
"qnc-server/app/main/model"
|
||
"qnc-server/common/xerr"
|
||
"qnc-server/pkg/lzkit/crypto"
|
||
"qnc-server/pkg/lzkit/lzUtils"
|
||
)
|
||
|
||
// WhitelistService 白名单领域服务
|
||
type WhitelistService struct {
|
||
config config.Config
|
||
UserFeatureWhitelistModel model.UserFeatureWhitelistModel
|
||
WhitelistOrderModel model.WhitelistOrderModel
|
||
WhitelistOrderItemModel model.WhitelistOrderItemModel
|
||
QueryModel model.QueryModel
|
||
FeatureModel model.FeatureModel
|
||
}
|
||
|
||
// NewWhitelistService 创建白名单服务
|
||
func NewWhitelistService(
|
||
c config.Config,
|
||
userFeatureWhitelistModel model.UserFeatureWhitelistModel,
|
||
whitelistOrderModel model.WhitelistOrderModel,
|
||
whitelistOrderItemModel model.WhitelistOrderItemModel,
|
||
queryModel model.QueryModel,
|
||
featureModel model.FeatureModel,
|
||
) *WhitelistService {
|
||
return &WhitelistService{
|
||
config: c,
|
||
UserFeatureWhitelistModel: userFeatureWhitelistModel,
|
||
WhitelistOrderModel: whitelistOrderModel,
|
||
WhitelistOrderItemModel: whitelistOrderItemModel,
|
||
QueryModel: queryModel,
|
||
FeatureModel: featureModel,
|
||
}
|
||
}
|
||
|
||
// EnsureFreeWhitelist 免费下架：如果还没有生效白名单，则创建一条免费白名单记录
|
||
func (s *WhitelistService) EnsureFreeWhitelist(
|
||
ctx context.Context,
|
||
session sqlx.Session,
|
||
idCard string,
|
||
feature *model.Feature,
|
||
userId string,
|
||
orderId string,
|
||
) error {
|
||
builder := s.UserFeatureWhitelistModel.SelectBuilder().
|
||
Where("id_card = ? AND feature_id = ?", idCard, feature.Id)
|
||
records, err := s.UserFeatureWhitelistModel.FindAll(ctx, builder, "")
|
||
if err != nil {
|
||
return errors.Wrap(err, "查询白名单记录失败")
|
||
}
|
||
for _, r := range records {
|
||
if r.Status == 1 {
|
||
return nil
|
||
}
|
||
}
|
||
wl := &model.UserFeatureWhitelist{
|
||
Id: uuid.NewString(),
|
||
IdCard: idCard,
|
||
FeatureId: feature.Id,
|
||
FeatureApiId: feature.ApiId,
|
||
UserId: userId,
|
||
OrderId: lzUtils.StringToNullString(orderId),
|
||
WhitelistOrderId: lzUtils.StringToNullString(""),
|
||
Amount: 0,
|
||
Status: 1,
|
||
}
|
||
_, err = s.UserFeatureWhitelistModel.Insert(ctx, session, wl)
|
||
if err != nil {
|
||
return errors.Wrap(err, "创建免费白名单记录失败")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// CreateWhitelistByPaidOrder 根据已支付的白名单订单，创建对应的白名单记录
|
||
func (s *WhitelistService) CreateWhitelistByPaidOrder(
|
||
ctx context.Context,
|
||
session sqlx.Session,
|
||
order *model.Order,
|
||
whitelistOrder *model.WhitelistOrder,
|
||
) error {
|
||
if whitelistOrder.Status != 2 {
|
||
return nil
|
||
}
|
||
itemBuilder := s.WhitelistOrderItemModel.SelectBuilder().
|
||
Where("order_id = ?", whitelistOrder.Id)
|
||
items, err := s.WhitelistOrderItemModel.FindAll(ctx, itemBuilder, "")
|
||
if err != nil {
|
||
return errors.Wrap(err, "查询白名单订单明细失败")
|
||
}
|
||
for _, item := range items {
|
||
wl := &model.UserFeatureWhitelist{
|
||
Id: uuid.NewString(),
|
||
IdCard: whitelistOrder.IdCard,
|
||
FeatureId: item.FeatureId,
|
||
FeatureApiId: item.FeatureApiId,
|
||
UserId: whitelistOrder.UserId,
|
||
OrderId: lzUtils.StringToNullString(order.Id),
|
||
WhitelistOrderId: lzUtils.StringToNullString(whitelistOrder.Id),
|
||
Amount: item.Price,
|
||
Status: 1,
|
||
}
|
||
if _, err := s.UserFeatureWhitelistModel.Insert(ctx, session, wl); err != nil {
|
||
return errors.Wrap(err, "创建白名单记录失败")
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// GetWhitelistedFeatureApisByIdCard 获取某个身份证号已下架的 feature_api_id 集合
|
||
func (s *WhitelistService) GetWhitelistedFeatureApisByIdCard(
|
||
ctx context.Context,
|
||
idCard string,
|
||
) (map[string]bool, error) {
|
||
result := make(map[string]bool)
|
||
if s == nil || idCard == "" {
|
||
return result, nil
|
||
}
|
||
builder := s.UserFeatureWhitelistModel.SelectBuilder().
|
||
Where("id_card = ? AND status = ?", idCard, 1)
|
||
list, err := s.UserFeatureWhitelistModel.FindAll(ctx, builder, "")
|
||
if err != nil {
|
||
return nil, errors.Wrap(err, "查询白名单失败")
|
||
}
|
||
for _, wl := range list {
|
||
result[wl.FeatureApiId] = true
|
||
}
|
||
return result, nil
|
||
}
|
||
|
||
// CheckWhitelistExists 检查指定身份证号和模块是否已有生效的白名单记录
|
||
func (s *WhitelistService) CheckWhitelistExists(
|
||
ctx context.Context,
|
||
idCard string,
|
||
featureId string,
|
||
) (bool, error) {
|
||
if idCard == "" || featureId == "" {
|
||
return false, nil
|
||
}
|
||
builder := s.UserFeatureWhitelistModel.SelectBuilder().
|
||
Where("id_card = ? AND feature_id = ? AND status = ?", idCard, featureId, 1)
|
||
list, err := s.UserFeatureWhitelistModel.FindAll(ctx, builder, "")
|
||
if err != nil {
|
||
return false, errors.Wrap(err, "查询白名单记录失败")
|
||
}
|
||
return len(list) > 0, nil
|
||
}
|
||
|
||
// ProcessOfflineFeature 统一下架处理
|
||
func (s *WhitelistService) ProcessOfflineFeature(
|
||
ctx context.Context,
|
||
session sqlx.Session,
|
||
idCard string,
|
||
featureApiId string,
|
||
userId string,
|
||
orderId string,
|
||
) (needPay bool, amount float64, whitelistCreated bool, err error) {
|
||
mainApiId := s.extractMainApiId(featureApiId)
|
||
feature, err := s.getFeatureByApiId(ctx, mainApiId)
|
||
if err != nil {
|
||
return false, 0, false, err
|
||
}
|
||
if feature.WhitelistPrice < 0 {
|
||
return false, 0, false, errors.Wrapf(xerr.NewErrMsg("该模块不支持下架"), "")
|
||
}
|
||
exists, err := s.CheckWhitelistExists(ctx, idCard, feature.Id)
|
||
if err != nil {
|
||
return false, 0, false, err
|
||
}
|
||
if exists {
|
||
return false, 0, true, nil
|
||
}
|
||
price := feature.WhitelistPrice
|
||
if price <= 0 {
|
||
if err := s.EnsureFreeWhitelist(ctx, session, idCard, feature, userId, orderId); err != nil {
|
||
return false, 0, false, err
|
||
}
|
||
return false, 0, true, nil
|
||
}
|
||
paidOrderId, err := s.findPaidWhitelistOrder(ctx, userId, idCard, feature.Id)
|
||
if err != nil {
|
||
return false, 0, false, err
|
||
}
|
||
if paidOrderId != "" {
|
||
if err := s.createWhitelistFromPaidOrder(ctx, session, idCard, feature, userId, orderId, paidOrderId, price); err != nil {
|
||
return false, 0, false, err
|
||
}
|
||
return false, price, true, nil
|
||
}
|
||
return true, price, false, nil
|
||
}
|
||
|
||
func (s *WhitelistService) extractMainApiId(featureApiId string) string {
|
||
if idx := strings.Index(featureApiId, "_"); idx > 0 {
|
||
return featureApiId[:idx]
|
||
}
|
||
return featureApiId
|
||
}
|
||
|
||
func (s *WhitelistService) getFeatureByApiId(ctx context.Context, apiId string) (*model.Feature, error) {
|
||
feature, err := s.FeatureModel.FindOneByApiId(ctx, apiId)
|
||
if err != nil {
|
||
if errors.Is(err, model.ErrNotFound) {
|
||
return nil, errors.Wrap(err, "模块不存在")
|
||
}
|
||
return nil, errors.Wrap(err, "查询模块信息失败")
|
||
}
|
||
return feature, nil
|
||
}
|
||
|
||
func (s *WhitelistService) findPaidWhitelistOrder(
|
||
ctx context.Context,
|
||
userId string,
|
||
idCard string,
|
||
featureId string,
|
||
) (string, error) {
|
||
orderBuilder := s.WhitelistOrderModel.SelectBuilder().
|
||
Where("user_id = ? AND id_card = ? AND status = ?", userId, idCard, 2)
|
||
orders, err := s.WhitelistOrderModel.FindAll(ctx, orderBuilder, "")
|
||
if err != nil {
|
||
return "", errors.Wrap(err, "查询白名单订单失败")
|
||
}
|
||
for _, order := range orders {
|
||
itemBuilder := s.WhitelistOrderItemModel.SelectBuilder().
|
||
Where("order_id = ? AND feature_id = ?", order.Id, featureId)
|
||
items, itemErr := s.WhitelistOrderItemModel.FindAll(ctx, itemBuilder, "")
|
||
if itemErr != nil {
|
||
return "", errors.Wrap(itemErr, "查询白名单订单明细失败")
|
||
}
|
||
if len(items) > 0 {
|
||
return order.Id, nil
|
||
}
|
||
}
|
||
return "", nil
|
||
}
|
||
|
||
func (s *WhitelistService) createWhitelistFromPaidOrder(
|
||
ctx context.Context,
|
||
session sqlx.Session,
|
||
idCard string,
|
||
feature *model.Feature,
|
||
userId string,
|
||
orderId string,
|
||
paidOrderId string,
|
||
price float64,
|
||
) error {
|
||
wl := &model.UserFeatureWhitelist{
|
||
Id: uuid.NewString(),
|
||
IdCard: idCard,
|
||
FeatureId: feature.Id,
|
||
FeatureApiId: feature.ApiId,
|
||
UserId: userId,
|
||
OrderId: lzUtils.StringToNullString(orderId),
|
||
WhitelistOrderId: lzUtils.StringToNullString(paidOrderId),
|
||
Amount: price,
|
||
Status: 1,
|
||
}
|
||
if _, err := s.UserFeatureWhitelistModel.Insert(ctx, session, wl); err != nil {
|
||
return errors.Wrap(err, "根据已支付订单创建白名单记录失败")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// DeleteFeatureFromQueryData 从报告数据中删除指定模块的数据
|
||
func (s *WhitelistService) DeleteFeatureFromQueryData(
|
||
ctx context.Context,
|
||
session sqlx.Session,
|
||
queryId string,
|
||
featureApiId string,
|
||
) error {
|
||
queryModel, err := s.getQueryModel(ctx, queryId)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if queryModel == nil {
|
||
return nil
|
||
}
|
||
mainApiId := s.extractMainApiId(featureApiId)
|
||
dataArray, key, err := s.decryptQueryData(queryModel)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
modifiedArray, hasModified := s.clearFeatureData(dataArray, mainApiId)
|
||
if !hasModified {
|
||
logx.Infof("删除报告数据：查询记录 %s 中未找到模块 %s 的数据，跳过删除", queryId, featureApiId)
|
||
return nil
|
||
}
|
||
if err := s.updateQueryData(ctx, session, queryModel, modifiedArray, key); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (s *WhitelistService) getQueryModel(ctx context.Context, queryId string) (*model.Query, error) {
|
||
queryModel, err := s.QueryModel.FindOne(ctx, queryId)
|
||
if err != nil {
|
||
if errors.Is(err, model.ErrNotFound) {
|
||
return nil, nil
|
||
}
|
||
return nil, errors.Wrap(err, "查询报告记录失败")
|
||
}
|
||
if !queryModel.QueryData.Valid || queryModel.QueryData.String == "" {
|
||
return nil, nil
|
||
}
|
||
return queryModel, nil
|
||
}
|
||
|
||
func (s *WhitelistService) decryptQueryData(queryModel *model.Query) ([]map[string]interface{}, []byte, error) {
|
||
secretKey := s.config.Encrypt.SecretKey
|
||
key, decodeErr := hex.DecodeString(secretKey)
|
||
if decodeErr != nil {
|
||
return nil, nil, errors.Wrap(decodeErr, "获取AES密钥失败")
|
||
}
|
||
decryptedData, decryptErr := crypto.AesDecrypt(queryModel.QueryData.String, key)
|
||
if decryptErr != nil {
|
||
return nil, nil, errors.Wrap(decryptErr, "解密报告数据失败")
|
||
}
|
||
var dataArray []map[string]interface{}
|
||
if unmarshalErr := json.Unmarshal(decryptedData, &dataArray); unmarshalErr != nil {
|
||
return nil, nil, errors.Wrap(unmarshalErr, "解析报告数据失败")
|
||
}
|
||
return dataArray, key, nil
|
||
}
|
||
|
||
func (s *WhitelistService) clearFeatureData(dataArray []map[string]interface{}, mainApiId string) ([]map[string]interface{}, bool) {
|
||
modifiedArray := make([]map[string]interface{}, 0, len(dataArray))
|
||
hasModified := false
|
||
for _, item := range dataArray {
|
||
newItem := make(map[string]interface{})
|
||
for k, v := range item {
|
||
newItem[k] = v
|
||
}
|
||
apiID, ok := item["apiID"].(string)
|
||
if !ok {
|
||
modifiedArray = append(modifiedArray, newItem)
|
||
continue
|
||
}
|
||
if s.extractMainApiId(apiID) == mainApiId {
|
||
newItem["data"] = nil
|
||
hasModified = true
|
||
}
|
||
modifiedArray = append(modifiedArray, newItem)
|
||
}
|
||
return modifiedArray, hasModified
|
||
}
|
||
|
||
func (s *WhitelistService) updateQueryData(
|
||
ctx context.Context,
|
||
session sqlx.Session,
|
||
queryModel *model.Query,
|
||
filteredArray []map[string]interface{},
|
||
key []byte,
|
||
) error {
|
||
filteredBytes, marshalErr := json.Marshal(filteredArray)
|
||
if marshalErr != nil {
|
||
return errors.Wrap(marshalErr, "序列化过滤后的报告数据失败")
|
||
}
|
||
encryptedData, encryptErr := crypto.AesEncrypt(filteredBytes, key)
|
||
if encryptErr != nil {
|
||
return errors.Wrap(encryptErr, "加密过滤后的报告数据失败")
|
||
}
|
||
queryModel.QueryData = sql.NullString{String: encryptedData, Valid: true}
|
||
if updateErr := s.QueryModel.UpdateWithVersion(ctx, session, queryModel); updateErr != nil {
|
||
return errors.Wrap(updateErr, "更新报告数据失败")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// CheckQueryDataContainsFeature 检查报告数据中是否包含指定的模块
|
||
func (s *WhitelistService) CheckQueryDataContainsFeature(
|
||
ctx context.Context,
|
||
queryId string,
|
||
featureApiId string,
|
||
) (bool, error) {
|
||
queryModel, err := s.getQueryModel(ctx, queryId)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
if queryModel == nil {
|
||
return false, nil
|
||
}
|
||
mainApiId := s.extractMainApiId(featureApiId)
|
||
dataArray, _, err := s.decryptQueryData(queryModel)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
for _, item := range dataArray {
|
||
apiID, ok := item["apiID"].(string)
|
||
if !ok {
|
||
continue
|
||
}
|
||
if s.extractMainApiId(apiID) == mainApiId {
|
||
dataValue, exists := item["data"]
|
||
if !exists || dataValue == nil {
|
||
return false, nil
|
||
}
|
||
return true, nil
|
||
}
|
||
}
|
||
return false, nil
|
||
}
|
||
|
||
// ProcessPaidWhitelistOrder 处理已支付的白名单订单
|
||
func (s *WhitelistService) ProcessPaidWhitelistOrder(
|
||
ctx context.Context,
|
||
session sqlx.Session,
|
||
order *model.Order,
|
||
whitelistOrder *model.WhitelistOrder,
|
||
) error {
|
||
if whitelistOrder.Status != 2 {
|
||
return nil
|
||
}
|
||
itemBuilder := s.WhitelistOrderItemModel.SelectBuilder().
|
||
Where("order_id = ?", whitelistOrder.Id)
|
||
items, err := s.WhitelistOrderItemModel.FindAll(ctx, itemBuilder, "")
|
||
if err != nil {
|
||
return errors.Wrap(err, "查询白名单订单明细失败")
|
||
}
|
||
for _, item := range items {
|
||
wl := &model.UserFeatureWhitelist{
|
||
Id: uuid.NewString(),
|
||
IdCard: whitelistOrder.IdCard,
|
||
FeatureId: item.FeatureId,
|
||
FeatureApiId: item.FeatureApiId,
|
||
UserId: whitelistOrder.UserId,
|
||
OrderId: lzUtils.StringToNullString(""),
|
||
WhitelistOrderId: lzUtils.StringToNullString(whitelistOrder.Id),
|
||
Amount: item.Price,
|
||
Status: 1,
|
||
}
|
||
if _, err := s.UserFeatureWhitelistModel.Insert(ctx, session, wl); err != nil {
|
||
return errors.Wrap(err, "创建白名单记录失败")
|
||
}
|
||
logx.Infof("白名单订单支付成功：订单 %s，模块 %s，已创建白名单记录", whitelistOrder.OrderNo, item.FeatureApiId)
|
||
}
|
||
return nil
|
||
}
|