详解Mybatis拦截器安全加解密MySQL数据实战

背景

在实际开发中，我们往往需要对敏感数据进行加解密，以保证系统的安全性。Mybatis作为一个流行的ORM框架，提供了很好的拦截器功能，我们可以使用拦截器对Mybatis执行的SQL进行修改，以实现对敏感数据的安全加解密。本文将详细讲解如何使用Mybatis的拦截器实现对MySQL敏感数据的安全加解密。

拦截器实现

1. 编写拦截器

首先，我们需要编写一个拦截器，用于对Mybatis的执行进行拦截。拦截器的核心代码如下：

@Intercepts({@Signature(type = StatementHandler.class, method = "update", args = {Statement.class})})
public class MyInterceptor implements Interceptor {

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        StatementHandler statementHandler = (StatementHandler)invocation.getTarget();
        ParameterHandler parameterHandler = statementHandler.getParameterHandler();
        //TODO: 对敏感数据进行加解密
        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }

    @Override
    public void setProperties(Properties properties) {
    }
}

2. 配置拦截器

接下来，我们需要在Mybatis的配置文件中配置该拦截器。配置文件的核心代码如下：

<plugins>
    <plugin interceptor="com.example.MyInterceptor"/>
</plugins>

加解密实现

1. 加密解密算法

在本文中，我们使用AES算法进行加解密。AES算法是一种高级加密标准，安全性较高，被广泛应用于各种领域。

2. 数据库字段加密

在数据库中，我们需要对敏感数据字段进行加密，以保证数据安全。在本文中，我们以用户信息表“user”为例，对“user”表中的“password”字段进行加密。加密的核心代码如下：

public static String AES_encrypt(String content, String key) throws Exception {
    SecretKeySpec skey = new SecretKeySpec(key.getBytes(), "AES");
    Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
    cipher.init(Cipher.ENCRYPT_MODE, skey);
    byte[] encrypted = cipher.doFinal(content.getBytes());
    return new BigInteger(1, encrypted).toString(16);
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    StatementHandler statementHandler = (StatementHandler)invocation.getTarget();
    ParameterHandler parameterHandler = statementHandler.getParameterHandler();
    Object parameterObject = parameterHandler.getParameterObject();

    if (statementHandler instanceof PreparedStatementHandler) {
        PreparedStatementHandler preparedStatementHandler = (PreparedStatementHandler)statementHandler;
        Field field = preparedStatementHandler.getClass().getDeclaredField("delegate");
        field.setAccessible(true);
        PreparedStatement delegate = (PreparedStatement)field.get(preparedStatementHandler);
        if (parameterObject != null) {
            List<ParameterMapping> parameterMappings = preparedStatementHandler.getBoundSql().getParameterMappings();
            for (int i = 0; i < parameterMappings.size(); i++) {
                ParameterMapping parameterMapping = parameterMappings.get(i);
                if (parameterMapping.getJavaType().equals(String.class) && "password".equals(parameterMapping.getProperty())) {
                    Object value = parameterHandler.getParameterValue(i + 1);

                    if (value != null) {
                        String encryptValue = AES_encrypt(value.toString(), encryptionKey);
                        delegate.setString(i + 1, encryptValue);
                    }
                }
            }
        }
    }
    return invocation.proceed();
}

3. 数据库字段解密

在应用程序中，我们需要对加密后的敏感数据字段进行解密才能正确地使用数据。在本文中，我们以“password”字段为例，对数据库中的密码字段进行解密。解密的核心代码如下：

public static String AES_decrypt(String content, String key) throws Exception {
    SecretKeySpec skey = new SecretKeySpec(key.getBytes(), "AES");
    Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
    cipher.init(Cipher.DECRYPT_MODE, skey);
    byte[] original = cipher.doFinal(new BigInteger(content, 16).toByteArray());
    return new String(original);
}

@Override
public void handleResultSets(Statement statement) throws SQLException {
    if (statement instanceof PreparedStatement) {
        PreparedStatement preparedStatement = (PreparedStatement) statement;
        preparedStatement.setFetchSize(fetchSize);
        ResultSet rs = preparedStatement.getResultSet();
        ResultSetMetaData metaData = rs.getMetaData();
        while (rs.next()) {
            for (int i = 1; i <= metaData.getColumnCount(); i++) {
                String columnName = metaData.getColumnName(i);
                if (metaData.getColumnType(i) == Types.VARCHAR && "password".equals(columnName)) {
                    String encryptValue = rs.getString(i);
                    if (encryptValue != null) {
                        String decryptValue = AES_decrypt(encryptValue, encryptionKey);
                        Method method = rs.getClass().getMethod("updateString", int.class, String.class);
                        method.invoke(rs, i, decryptValue);
                    }
                }
            }
        }
    }
}

示例说明

示例1

在代码中访问数据库，查询用户信息。查询的SQL如下：

SELECT * FROM user WHERE username = ?

其中，“user”表中的“password”字段存储的是加密后的密码。

通过使用Mybatis拦截器，在执行SQL之前，对“password”字段进行解密，以获取正确的用户密码。解密的过程在上述“数据库字段解密”中已经演示。

示例2

在代码中向数据库中插入用户信息。插入的SQL如下：

INSERT INTO user(username, password) VALUES (?, ?)

其中，“password”字段需要存储加密后的用户密码。

通过使用Mybatis拦截器，在执行SQL之前，对“password”字段进行加密，以保证数据安全。加密的过程在上述“数据库字段加密”中已经演示。

总结

通过使用Mybatis拦截器，我们可以很容易地实现对MySQL敏感数据的安全加解密。当我们需要对敏感数据进行加解密时，可以采用该方法来保证数据安全。