package org.springblade.common.annotation;

import com.fasterxml.jackson.annotation.JacksonAnnotationsInside;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import org.springblade.common.config.SensitiveJsonSerializer;
import org.springblade.common.enums.SensitiveTypeEnum;

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

/**
 * 脫敏注解
 *
 * @author hf
 * @since 2023/6/14
 **/
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
@JacksonAnnotationsInside
@JsonSerialize(using = SensitiveJsonSerializer.class)
public @interface Desensitized {
    /**
     * 脫敏類(lèi)型(規(guī)則)
     */
    SensitiveTypeEnum type();
}

package org.springblade.common.enums;

import java.util.function.Function;

/**
 * 脫敏類(lèi)型枚舉
 *
 * @author hf
 * @since
 **/
public enum SensitiveTypeEnum {
    /**
     * 用戶(hù)名
     */
    USERNAME(s -> s.replaceAll("(\\S)\\S(\\S*)", "$1*$2")),
    /**
     * 身份證
     */
    ID_CARD(s -> s.replaceAll("(\\d{4})\\d{10}(\\w{4})", "$1****$2")),
    /**
     * 手機(jī)號(hào)
     */
    MOBILE_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> desensitizer;

    SensitiveTypeEnum(Function<String, String> desensitizer) {
        this.desensitizer = desensitizer;
    }

    public Function<String, String> desensitizer() {
        return desensitizer;
    }
}

package org.springblade.common.config;


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 org.springblade.common.annotation.Desensitized;
import org.springblade.common.enums.SensitiveTypeEnum;

import java.io.IOException;
import java.util.Objects;

/**
 * 序列化注解自定義實(shí)現(xiàn)
 * JsonSerializer<String>：指定String 類(lèi)型，serialize()方法用于將修改后的數(shù)據(jù)載入
 */
public class SensitiveJsonSerializer extends JsonSerializer<String> implements ContextualSerializer {
    private SensitiveTypeEnum strategy;

    @Override
    public void serialize(String value, JsonGenerator gen, SerializerProvider serializers) throws IOException {
        gen.writeString(strategy.desensitizer().apply(value));
    }

    /**
     * 獲取屬性上的注解屬性
     */
    @Override
    public JsonSerializer<?> createContextual(SerializerProvider prov, BeanProperty property) throws JsonMappingException {

        Desensitized annotation = property.getAnnotation(Desensitized.class);
        if (Objects.nonNull(annotation) && Objects.equals(String.class, property.getType().getRawClass())) {
            this.strategy = annotation.type();
            return this;
        }
        return prov.findValueSerializer(property.getType(), property);

    }
}

/**
	 * 手機(jī)
	 */
	@Desensitized(type = SensitiveTypeEnum.MOBILE_PHONE)
	private String phone;
	/**
	 * 身份證號(hào)
	 */
	@Desensitized(type = SensitiveTypeEnum.ID_CARD)
	private String idNo;

{
	"phone": "13924369234",
	"idNo": "1210103199603282410",
}

{
	"phone": "139****9234",
	"idNo": "1210************10",
}