com.feizhaiyou.encrypt.handler.CommonSensitiveHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of encrypt-spring-boot-starter Show documentation

整合SpringBoot可对Web项目的HTTP接口参数进行脱敏与加解密，通过注解的方式直接使用加解密，支持AES、RSA，可自定义脱敏与加解密配置。

The newest version!

package com.feizhaiyou.encrypt.handler;

import com.feizhaiyou.encrypt.annotation.Sensitive;
import com.feizhaiyou.encrypt.format.SensitiveProcessor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.Assert;

import java.lang.annotation.Annotation;

/**
 * @author ls
 * @since 2023-07-27
 */
@Slf4j
public class CommonSensitiveHandler implements SensitiveHandler {

    private SensitiveProcessor sensitiveProcessor;

    public CommonSensitiveHandler(SensitiveProcessor sensitiveProcessor) {
        Assert.notNull(sensitiveProcessor, "sensitiveProcessor could not be null");
        this.sensitiveProcessor = sensitiveProcessor;
    }

    @Override
    public Sensitive acquire(Annotation[] annotations) {
        Sensitive sensitive = null;
        for (int i = 0; i < annotations.length; i++) {
            if (annotations[i] instanceof Sensitive) {
                sensitive = (Sensitive) annotations[i];
                if (sensitive.required()) {
                    // 注解存在且required为true
                    return sensitive;
                }
            }
        }
        return null;
    }

    @Override
    public String format(String source, Sensitive annotation) {
        if (annotation.required()) {
            try {
                return sensitiveProcessor.format(source, annotation.type());
            } catch (Exception e) {
                log.error("sensitive fail: {}, source field value: {}", e.getMessage(), source);
            }
        }
        return source;
    }
}