自定义注解实现数据脱敏

实现方案: 利用jackson提供的JsonSerializer 在返回response对象序列化时 处理数据.

1. 定义枚举类, 提供不通的脱敏策略
   这里用一个函数来做属性, 实在是妙!

/**
 * 敏感数据处理策略枚举
 *
 * @author Lynn
 * @date 2022/10/9 15:36
 */
@Getter
@AllArgsConstructor
public enum SensitiveStrategyEnum {

    /**
     * 用户名
     */
    USERNAME(s -> s.replaceAll("(\\S)\\S(\\S*)", "$1*$2")),

    /**
     * 身份证
     */
    ID_CARD(s -> s.replaceAll("(\\d{4})\\d{10}(\\w{4})", "$1****$2")),

    /**
     * 手机号
     */
    PHONE(s -> s.replaceAll("(\\d{3})\\d{4}(\\d{4})", "$1****$2")),

    /**
     * 地址
     */
    ADDRESS(s -> s.replaceAll("(\\S{3})\\S{2}(\\S*)\\S{2}", "$1****$2****"));

    private final Function<String, String> desensitise;
}

2.定义序列化类, 继承jackson的JsonSerializer, 具体的序列化过程, 由此类完成

/**
 * @author Lynn
 * @date 2022/10/9 15:47
 */
public class SensitiveJsonSerializer extends JsonSerializer<String> implements ContextualSerializer {

    private SensitiveStrategyEnum strategy;

    @Override
    public void serialize(String value, JsonGenerator jsonGenerator, SerializerProvider serializerProvider) throws IOException {
        jsonGenerator.writeString(strategy.getDesensitise().apply(value));
    }

    @Override
    public JsonSerializer<?> createContextual(SerializerProvider serializerProvider, BeanProperty beanProperty) throws JsonMappingException {
        Sensitive annotation = beanProperty.getAnnotation(Sensitive.class);
        if (Objects.nonNull(annotation) && Objects.equals(String.class, beanProperty.getType().getRawClass())) {
            this.strategy = annotation.strategy();
            return this;
        }
        return serializerProvider.findValueSerializer(beanProperty.getType(), beanProperty);
    }
}

3.自定义枚举

/**
 * @author Lynn
 * @date 2022/10/9 15:35
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
@JacksonAnnotationsInside
@JsonSerialize(using = SensitiveJsonSerializer.class)
public @interface Sensitive {
    SensitiveStrategyEnum strategy();
}

4.具体使用

/**
 * @author Lynn
 * @date 2022/10/9 14:47
 */
@Data
public class AddressListResponse {

    /**
     * 收货人姓名
     */
    private String receiverName;

    /**
     * 收货人手机号
     */
    @Sensitive(strategy = SensitiveStrategyEnum.PHONE)
    private String receiverPhone;
}