使用mybatis或mybatisplus存储数据库加密，解密

最新推荐文章于 2025-01-06 22:18:16 发布

原创最新推荐文章于 2025-01-06 22:18:16 发布 · 342 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#1024程序员节

博客介绍了Java开发中的注解使用，在类头添加@TableName注解指定表名并开启自动结果映射，在需加密字段添加@TableField注解并指定类型处理器。


/**
 * @author kxs
 * @email 1965654634@qq.com
 */
import com.chinaunicom.cpic.iot.dap.common.utils.AESUtils;
import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author xrp
 */
public class TypeHandler extends BaseTypeHandler<String> {
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, String parameter, JdbcType jdbcType) throws SQLException {
        ps.setString(i, AESUtils.encrypt(parameter,"1234567891234567"));
    }

    @Override
    public String getNullableResult(ResultSet rs, String columnName) throws SQLException {
        return AESUtils.decrypt(rs.getString(columnName),"1234567891234567");
    }

    @Override
    public String getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        return AESUtils.decrypt(rs.getString(columnIndex),"1234567891234567");
    }

    @Override
    public String getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        return AESUtils.decrypt(cs.getString(columnIndex),"1234567891234567");
    }
}

import lombok.extern.slf4j.Slf4j;
import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import java.security.Key;
import java.security.SecureRandom;

/**
 * @ClassName AESUtils
 * @Description AES工具类
 */
@Slf4j
public class AESUtils {

    // optional value AES/DES/DESede
    public static final String CIPHER_ALGORITHM = "AES/ECB/PKCS5Padding";

    public static Key getKey(String strKey) {
        try {
            if (strKey == null) {
                strKey = "";
            }
            KeyGenerator _generator = KeyGenerator.getInstance("AES");
            SecureRandom secureRandom = SecureRandom.getInstance("SHA1PRNG");
            secureRandom.setSeed(strKey.getBytes());
            _generator.init(128, secureRandom);
            return _generator.generateKey();
        } catch (Exception e) {
            throw new RuntimeException(" 密钥出现异常 ");
        }
    }

    public static String encrypt(String data, String key) {
        try {
            SecureRandom sr = new SecureRandom();
            Key secureKey = getKey(key);
            Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM);
            cipher.init(Cipher.ENCRYPT_MODE, secureKey, sr);
            byte[] bt = cipher.doFinal(data.getBytes());
            String strS = new BASE64Encoder().encode(bt);
            return strS;
        }catch (Exception e){
            log.error("加密失败,",e);
            throw new RuntimeException(802,"加密失败");
        }


    }

    public static String decrypt(String message,String key) {
        try {
            SecureRandom sr = new SecureRandom();
            Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM);
            Key secureKey = getKey(key);
            cipher.init(Cipher.DECRYPT_MODE, secureKey, sr);
            byte[] res = new BASE64Decoder().decodeBuffer(message);
            res = cipher.doFinal(res);
            return new String(res,"UTF-8");
        }catch (Exception e){
            log.error("解密失败,",e);
            throw new BusinessException(802,"解密失败");
        }

    }

}