74 lines
2.3 KiB
Go
74 lines
2.3 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"qnc-server/app/main/api/internal/svc"
|
|
"qnc-server/app/main/api/internal/types"
|
|
"qnc-server/app/main/model"
|
|
"qnc-server/common/xerr"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type CheckFeatureWhitelistStatusLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewCheckFeatureWhitelistStatusLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CheckFeatureWhitelistStatusLogic {
|
|
return &CheckFeatureWhitelistStatusLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *CheckFeatureWhitelistStatusLogic) CheckFeatureWhitelistStatus(req *types.CheckFeatureWhitelistStatusReq) (resp *types.CheckFeatureWhitelistStatusResp, err error) {
|
|
if req.IdCard == "" {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("身份证号不能为空"), "")
|
|
}
|
|
if req.FeatureApiId == "" {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("模块API标识不能为空"), "")
|
|
}
|
|
mainApiId := req.FeatureApiId
|
|
if idx := strings.Index(req.FeatureApiId, "_"); idx > 0 {
|
|
mainApiId = req.FeatureApiId[:idx]
|
|
}
|
|
feature, err := l.svcCtx.FeatureModel.FindOneByApiId(l.ctx, mainApiId)
|
|
if err != nil {
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("模块不存在"), "")
|
|
}
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询模块信息失败, %v", err)
|
|
}
|
|
whitelistBuilder := l.svcCtx.UserFeatureWhitelistModel.SelectBuilder().
|
|
Where("id_card = ? AND feature_api_id = ? AND status = ?", req.IdCard, mainApiId, 1)
|
|
whitelists, err := l.svcCtx.UserFeatureWhitelistModel.FindAll(l.ctx, whitelistBuilder, "")
|
|
if err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询白名单记录失败, %v", err)
|
|
}
|
|
isWhitelisted := len(whitelists) > 0
|
|
dataDeleted := false
|
|
if req.QueryId != "" {
|
|
containsFeature, err := l.svcCtx.WhitelistService.CheckQueryDataContainsFeature(l.ctx, req.QueryId, req.FeatureApiId)
|
|
if err != nil {
|
|
logx.Errorf("检查报告数据是否包含模块失败：%v", err)
|
|
dataDeleted = true
|
|
} else {
|
|
dataDeleted = !containsFeature
|
|
}
|
|
} else {
|
|
dataDeleted = true
|
|
}
|
|
return &types.CheckFeatureWhitelistStatusResp{
|
|
IsWhitelisted: isWhitelisted,
|
|
WhitelistPrice: feature.WhitelistPrice,
|
|
FeatureId: feature.Id,
|
|
DataDeleted: dataDeleted,
|
|
}, nil
|
|
}
|