wangxianzhang
3 years ago
14 changed files with 250 additions and 3 deletions
@ -0,0 +1,44 @@ |
|||||
|
package com.epmet.commons.tools.annotation; |
||||
|
|
||||
|
import com.epmet.commons.tools.enums.RequirePermissionEnum; |
||||
|
import javassist.runtime.Inner; |
||||
|
|
||||
|
import java.lang.annotation.*; |
||||
|
import java.util.function.Function; |
||||
|
|
||||
|
/** |
||||
|
* 标记一个接口，它的返回值中的某些字段需要打掩码 |
||||
|
*/ |
||||
|
@Target(ElementType.METHOD) |
||||
|
@Retention(RetentionPolicy.RUNTIME) |
||||
|
@Documented |
||||
|
public @interface MaskResponse { |
||||
|
|
||||
|
/** |
||||
|
* 掩码类型 |
||||
|
*/ |
||||
|
String MASK_TYPE_ID_CARD = "ID_CARD"; |
||||
|
String MASK_TYPE_MOBILE = "MOBILE"; |
||||
|
|
||||
|
///**
|
||||
|
// * 默认的一些字段，如果没有手动指定，就会使用默认的。如果手动指定了，就不再使用默认的
|
||||
|
// */
|
||||
|
//String[] defaultFieldnames = {"idCard","mobile","phone"};
|
||||
|
//
|
||||
|
///**
|
||||
|
// * 默认字段对应的掩码类型
|
||||
|
// */
|
||||
|
//String[] defaultFieldsMaskType = { MASK_TYPE_ID_CARD, MASK_TYPE_MOBILE, MASK_TYPE_MOBILE };
|
||||
|
|
||||
|
/** |
||||
|
* 要打码的字段列表。会递归的着这些字段 |
||||
|
* @return |
||||
|
*/ |
||||
|
String[] fieldNames() default {"idCard","mobile","phone"}; |
||||
|
|
||||
|
/** |
||||
|
* 要打码的类型 |
||||
|
* @return |
||||
|
*/ |
||||
|
String[] fieldsMaskType() default { MASK_TYPE_ID_CARD, MASK_TYPE_MOBILE, MASK_TYPE_MOBILE }; |
||||
|
} |
||||
@ -0,0 +1,36 @@ |
|||||
|
package com.epmet.commons.tools.aspect; |
||||
|
|
||||
|
import com.epmet.commons.tools.annotation.MaskResponse; |
||||
|
import com.epmet.commons.tools.exception.EpmetErrorCode; |
||||
|
import com.epmet.commons.tools.exception.EpmetException; |
||||
|
import com.epmet.commons.tools.processor.MaskProcessor; |
||||
|
import com.epmet.commons.tools.utils.Result; |
||||
|
import org.aspectj.lang.JoinPoint; |
||||
|
import org.aspectj.lang.annotation.AfterReturning; |
||||
|
import org.aspectj.lang.annotation.Aspect; |
||||
|
import org.aspectj.lang.reflect.MethodSignature; |
||||
|
import org.springframework.core.annotation.Order; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
|
||||
|
@Aspect |
||||
|
@Component |
||||
|
@Order(0) |
||||
|
public class MaskResponseAspect { |
||||
|
|
||||
|
@AfterReturning(pointcut = "@annotation(com.epmet.commons.tools.annotation.MaskResponse)", returning = "result") |
||||
|
public Object proceed(JoinPoint point, Result result) throws Throwable { |
||||
|
MethodSignature signature = (MethodSignature) point.getSignature(); |
||||
|
MaskResponse maskResponseAnno = signature.getMethod().getAnnotation(MaskResponse.class); |
||||
|
|
||||
|
String[] fieldNames = maskResponseAnno.fieldNames(); |
||||
|
String[] fieldsMaskType = maskResponseAnno.fieldsMaskType(); |
||||
|
|
||||
|
if (fieldNames.length != fieldsMaskType.length) { |
||||
|
String msg = "掩码配置错误"; |
||||
|
throw new EpmetException(EpmetErrorCode.SERVER_ERROR.getCode(), msg, msg); |
||||
|
} |
||||
|
|
||||
|
new MaskProcessor(fieldNames, fieldsMaskType).mask(result); |
||||
|
return null; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,142 @@ |
|||||
|
package com.epmet.commons.tools.processor; |
||||
|
|
||||
|
import cn.hutool.core.util.StrUtil; |
||||
|
import com.epmet.commons.tools.annotation.MaskResponse; |
||||
|
import com.epmet.commons.tools.exception.ExceptionUtils; |
||||
|
import com.epmet.commons.tools.page.PageData; |
||||
|
import com.epmet.commons.tools.utils.Result; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.lang3.StringUtils; |
||||
|
|
||||
|
import java.lang.reflect.Field; |
||||
|
import java.util.Arrays; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.regex.Pattern; |
||||
|
|
||||
|
@Slf4j |
||||
|
public class MaskProcessor { |
||||
|
|
||||
|
|
||||
|
public static final String EPMET_PACKAGE_PREFIX = "com.epmet"; |
||||
|
|
||||
|
private List<String> fieldNames; |
||||
|
private List<String> fieldsMaskType; |
||||
|
|
||||
|
public MaskProcessor(String[] fields, String[] fieldsMaskType) { |
||||
|
if (fields != null && fields.length > 0) { |
||||
|
this.fieldNames = Arrays.asList(fields); |
||||
|
this.fieldsMaskType = Arrays.asList(fieldsMaskType); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 为dto中的属性打掩码 |
||||
|
* @param object |
||||
|
*/ |
||||
|
public void mask(Object object) { |
||||
|
if (object == null) { |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
if (object instanceof Result) { |
||||
|
mask(((Result<?>) object).getData()); |
||||
|
return; |
||||
|
} else if (object instanceof PageData) { |
||||
|
mask(((PageData<?>) object).getList()); |
||||
|
return; |
||||
|
} else if (object instanceof List) { |
||||
|
((List)object).forEach(e -> mask(e)); |
||||
|
return; |
||||
|
} else if (object instanceof Map) { |
||||
|
((Map) object).values().forEach(v -> mask(v)); |
||||
|
return; |
||||
|
} else if (object.getClass().getName().startsWith(EPMET_PACKAGE_PREFIX)) { |
||||
|
// 自定义bean，走反射
|
||||
|
maskEpmetBean(object); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 反射 |
||||
|
* @param object |
||||
|
*/ |
||||
|
private void maskEpmetBean(Object object) { |
||||
|
Field[] declaredFields = object.getClass().getDeclaredFields(); |
||||
|
for (Field currentField : declaredFields) { |
||||
|
currentField.setAccessible(true); |
||||
|
try { |
||||
|
String fieldName = currentField.getName(); |
||||
|
Object value = currentField.get(object); |
||||
|
// 是epmet的类，继续下钻
|
||||
|
if (currentField.getClass().getName().startsWith(EPMET_PACKAGE_PREFIX)) { |
||||
|
maskEpmetBean(value); |
||||
|
continue; |
||||
|
} |
||||
|
|
||||
|
// 是字符串
|
||||
|
String fieldValue; |
||||
|
if (value instanceof String && StringUtils.isNotBlank(fieldValue = (String) value)) { |
||||
|
int fieldIndexInAnnoAttrs = fieldNames.indexOf(fieldName); |
||||
|
if (fieldIndexInAnnoAttrs != -1) { |
||||
|
String product = maskString(fieldValue, fieldsMaskType.get(fieldIndexInAnnoAttrs)); |
||||
|
currentField.set(object, product); |
||||
|
} |
||||
|
continue; |
||||
|
} |
||||
|
|
||||
|
// 非字符串，非epmet类的其他类型
|
||||
|
mask(value); |
||||
|
} catch (IllegalAccessException e) { |
||||
|
log.error("【mask一些字段报错】{}", ExceptionUtils.getErrorStackTrace(e)); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 把字符串变更为掩码 |
||||
|
* @param originString |
||||
|
* @return |
||||
|
*/ |
||||
|
public String maskString(String originString, String maskType) { |
||||
|
if (MaskResponse.MASK_TYPE_ID_CARD.equals(maskType)) { |
||||
|
return maskIdCard(originString); |
||||
|
} else if (MaskResponse.MASK_TYPE_MOBILE.equals(maskType)) { |
||||
|
return maskMobile(originString); |
||||
|
} else { |
||||
|
return originString; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 将明文字符串打码变为掩码。保留前6，后面打码 |
||||
|
* @param originString |
||||
|
* @return |
||||
|
*/ |
||||
|
private String maskIdCard(String originString) { |
||||
|
// 仅将6位之后的全都打码
|
||||
|
int length = originString.length(); |
||||
|
if (length <= 6) { |
||||
|
return originString; |
||||
|
} |
||||
|
|
||||
|
return originString.replace(originString.substring(6), StrUtil.repeatByLength("*", length - 6)); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 将明文字符串打码变为掩码。保留前3后4，中间打码 |
||||
|
* 187****3461 |
||||
|
* @param originString |
||||
|
* @return |
||||
|
*/ |
||||
|
private String maskMobile(String originString) { |
||||
|
int length = originString.length(); |
||||
|
if (length <= 7) { |
||||
|
return originString; |
||||
|
} |
||||
|
|
||||
|
String maskStr = StrUtil.repeatByLength("*", length - 7); |
||||
|
|
||||
|
return originString.replaceAll("^(1\\d{2})\\d*(\\d{4})$", new StringBuilder("$1").append(maskStr).append("$2").toString()); |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue