mongo和sqlServer数据表字段加解密

一. mongoDB

        实现BeforeConvertCallback  AfterConvertCallback  AfterSaveCallback，做钩子函数，然后拦截,加密算法使用自定义的加密算法，也可以用现有的。

import org.bson.Document;
import org.jetbrains.annotations.NotNull;
import org.springframework.data.mongodb.core.mapping.event.AfterConvertCallback;
import org.springframework.data.mongodb.core.mapping.event.AfterSaveCallback;
import org.springframework.data.mongodb.core.mapping.event.BeforeConvertCallback;
import org.springframework.stereotype.Component;

//注册成组件
@Component
public class DocumentCallback implements BeforeConvertCallback<IEncryptDocument>, AfterConvertCallback<IEncryptDocument>, AfterSaveCallback<IEncryptDocument> {

    @NotNull
    @Override
    public IEncryptDocument onAfterConvert(@NotNull IEncryptDocument entity, @NotNull Document document, @NotNull String collection) {
        encryptOrDecryptDocument(entity, false);

        return entity;
    }

    @NotNull
    @Override
    public IEncryptDocument onAfterSave(@NotNull IEncryptDocument entity, @NotNull Document document, @NotNull String collection) {
        encryptOrDecryptDocument(entity, false);

        return entity;
    }

    @NotNull
    @Override
    public IEncryptDocument onBeforeConvert(@NotNull IEncryptDocument entity, @NotNull String collection) {
        encryptOrDecryptDocument(entity, true);

        return entity;
    }

    private void encryptOrDecryptDocument(IEncryptDocument entity, boolean doEncrypt) {

        if (null == entity) {
            return;
        }

        Class<?> entityClass = entity.getClass();

        if (ICustomEncryptDocument.class.isAssignableFrom(entityClass)) {
            ICustomEncryptDocument iCustomEncryptDocument = (ICustomEncryptDocument) entity;
            iCustomEncryptDocument.handleEncrypt(doEncrypt);
            return;
        }

        EncryptFieldUtil.handleField(entity, doEncrypt);
    }
}

2.sqlserver

@Convert(converter = NeedEncryptAttributeConverter.class)  

加上这个注解就能，NeedEncryptAttributeConverter(实现AttributeConverter,重写两个方法)只要写好这个类

import javax.persistence.AttributeConverter;

public class NeedEncryptAttributeConverter implements AttributeConverter<String, String> {

    private EncryptService encryptService;

    @Override
    public String convertToDatabaseColumn(String attribute) {
        init();
        return encryptService.encrypt(attribute);
    }

    @Override
    public String convertToEntityAttribute(String dbData) {
        init();
        return encryptService.decrypt(dbData);
    }

    private void init() {
        if (encryptService == null) {
            encryptService = ApplicationContextHolder.getBean("encryptService", EncryptService.class);
        }
    }
}