Java自定义注解实现数据脱敏

在实际开发中经常会遇到有一些信息不能全部展示用户，需要隐藏（可以叫脱敏），所以本文为大家分享了利用自定义注解实现数据脱敏的示例代码，需要的可以参考下

在实际开发中经常会遇到有一些信息不能全部展示用户，需要隐藏（可以叫脱敏）一部分的情况比如地址，电话，手机号，身份证等。脱敏的做法目前我知道的方法有：

1）：业务代码脱敏：顾名思义就是拿到数据需要脱敏的那个字段进过一系列的脱敏规则替换成自己想要的格式。

2）：自定义注解+aop切面的方式去完成字段脱敏的目的。

3）：自定义注解+序列化的方式对数据进行脱敏。

其中效率最慢的当属于代码业务代码方式，最优的方式当属于自定义注解的方式，减少代码量，提高工作效率，需要脱敏的字段仅需要加一个注解就可以达到目的。

今天就记录下第三种方式，自定义注解+序列化的方式。

首先先定义一个需要脱敏的策略

 

package com.***.sensitive;

import java.util.function.Function;

public enum SensitiveType {
    /**
     * 名称脱敏
     */
    USERNAME(s -> s.replaceAll("(\\S)\\S(\\S*)", "$1*$2"))
    ,
    /**
     * 手机号脱敏
     */
    MOBILE_PHONE(s -> s.replaceAll("(\\d{3})\\d{4}(\\d{4})", "$1****$2"))
    ,
    /**
     * 身份证号脱敏
     */
    ID_NUMBER(s -> s.replaceAll("(\\w{6})\\w*(\\w{3})", "$1*********$2"))
    ,
    /**
     *地址脱敏
     */
    ADDRESS(s -> s.replaceAll("(\\S{3})\\S{2}(\\S*)\\S{2}", "$1****$2****"))
    ;
    /**
     * 成员变量  是一个接口类型
     */
    private Function<String, String> function;
    SensitiveType(Function<String, String> function) {
        this.function = function;
    }
    public Function<String, String> function() {
        return this.function;
    }
}

然后自定义一个注解

package com.***.sensitive;

import com.fasterxml.jackson.annotation.JacksonAnnotationsInside;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
@JacksonAnnotationsInside //标明序列化
@JsonSerialize(using = DesensitizationDataSerialize.class)
public @interface SensitiveInfo {
    SensitiveType type();
}

然后自定义一个序列化器

package com.***.sensitive;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.BeanProperty;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.ser.ContextualSerializer;
import lombok.NoArgsConstructor;
import java.io.IOException;
import java.util.Objects;
/**
 * @description: 数据脱敏序列化器
 * @author: 
 * @date: 2023/7/31 19:17
 * @Version: 1.0
 */
@NoArgsConstructor
public class DesensitizationDataSerialize extends JsonSerializer<String> implements ContextualSerializer {
    private SensitiveType type;
    @Override
    public JsonSerializer<?> createContextual(SerializerProvider serializerProvider, BeanProperty beanProperty) throws JsonMappingException {
        SensitiveInfo annotation = beanProperty.getAnnotation(SensitiveInfo.class);
        if(Objects.nonNull(annotation) && Objects.equals(beanProperty.getType().getRawClass(),String.class)){
            this.type = annotation.type();
            return this;
        }
        return serializerProvider.findValueSerializer(beanProperty.getType(),beanProperty);
    }
    @Override
    public void serialize(String s, JsonGenerator jsonGenerator, SerializerProvider serializerProvider) throws IOException {
        jsonGenerator.writeString(type.function().apply(s));
    }
}

这样一个自定义脱敏注解就完成了。

使用方法：

只需要在返回参数需要脱敏的字段上加上响应注解就可以了

public class User implements Serializable {
    @TableId("id")
    private Integer id;
    @JsonFormat(pattern = "yyyy-MM-dd",timezone = "GMT+8")
    @ApiModelProperty(value = "出生年月")
    private Date birthday;
    @ApiModelProperty(value = "性别")
    private String gender;
    @ApiModelProperty(value = "姓名")
    private String username;
    @ApiModelProperty(value = "密码")
    private String password;
    @ApiModelProperty(value = "备注")
    private String remark;
    @ApiModelProperty(value = "状态")
    private String station;
    @ApiModelProperty(value = "联系电话")
    //手机号脱敏
    @DesensitizationData(function = DesensitizationEnum.MOBILE_PHONE)
    private String telephone;
}

结果。