偷懒式自定义注解脱敏

枚举

@AllArgsConstructor
public enum SensitiveStrategy {
    
    

    /**
     * 身份证脱敏
     */
    ID_CARD(s -> DesensitizedUtil.idCardNum(s, 3, 4)),

    /**
     * 手机号脱敏
     */
    PHONE(DesensitizedUtil::mobilePhone),

    /**
     * 地址脱敏
     */
    ADDRESS(s -> DesensitizedUtil.address(s, 8)),

    /**
     * 邮箱脱敏
     */
    EMAIL(DesensitizedUtil::email),

    /**
     * 银行卡
     */
    BANK_CARD(DesensitizedUtil::bankCard);

    //可自行添加其他脱敏策略

    private final Function<String, String> desensitizer;

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

自定义JsonSerializer序列化

@Slf4j
public class SensitiveJsonSerializer extends JsonSerializer<String> implements ContextualSerializer {
    
    

    private SensitiveStrategy strategy;

    @Override
    public void serialize(String value, JsonGenerator gen, SerializerProvider serializers) throws IOException {
    
    
           try {
    
    
//            SensitiveService sensitiveService = SpringUtils.getBean(SensitiveService.class);
//            if (ObjectUtil.isNotNull(sensitiveService) && sensitiveService.isSensitive()) {
    
    
                gen.writeString(strategy.desensitizer().apply(value));
//            } else {
    
    
//                gen.writeString(value);
//            }
        } catch (BeansException e) {
    
    
            log.error("脱敏实现不存在, 采用默认处理 => {}", e.getMessage());
            gen.writeString(value);
        }
    }

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

自定义注解

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
//注解的作用是将多个注解组合到一起
@JacksonAnnotationsInside
// 定义新的序列化注解
@JsonSerialize(using = SensitiveJsonSerializer.class)
public @interface Sensitive {
    
    
    SensitiveStrategy strategy();
}

测试

public class TestSensitiveController extends BaseController {
    
    

    /**
     * 测试数据脱敏
     */
    @GetMapping("/test")
    public R<TestSensitive> test() {
    
    
        TestSensitive testSensitive = new TestSensitive();
        testSensitive.setIdCard("210397198608215431");
        testSensitive.setPhone("17640125371");
        testSensitive.setAddress("北京市朝阳区某某四合院1203室");
        testSensitive.setEmail("[email protected]");
        testSensitive.setBankCard("6226456952351452853");
        return R.ok(testSensitive);
    }

    public static void main(String[] args) throws JsonProcessingException {
    
    
        TestSensitive testSensitive = new TestSensitive();
        testSensitive.setIdCard("371523199811216292");
        testSensitive.setPhone("17640125371");
        testSensitive.setAddress("北京市朝阳区某某四合院1203室");
        testSensitive.setEmail("[email protected]");
        testSensitive.setBankCard("6226456952351452853");
          // 此处应该用jackson,因为自定义的序列化方法是jackson
        JsonMapper jsonMapper = new JsonMapper();
        System.err.println(jsonMapper.writeValueAsString(testSensitive));
    }

    @Data
    static class TestSensitive {
    
    

        /**
         * 身份证
         */
        @Sensitive(strategy = SensitiveStrategy.ID_CARD)
        private String idCard;

        /**
         * 电话
         */
        @Sensitive(strategy = SensitiveStrategy.PHONE)
        private String phone;

        /**
         * 地址
         */
        @Sensitive(strategy = SensitiveStrategy.ADDRESS)
        private String address;

        /**
         * 邮箱
         */
        @Sensitive(strategy = SensitiveStrategy.EMAIL)
        private String email;

        /**
         * 银行卡
         */
        @Sensitive(strategy = SensitiveStrategy.BANK_CARD)
        private String bankCard;

    }

}

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_43861630/article/details/129202377