Mybatis执行非标准的SQL语句报错解决方法

目录

• 问题
• 解决方法

问题

这个问题是本人在开发一个GIS项目后端时，使用了Postgis数据库，我在程序中需要管理多个物视图数据的更新。需要通过下面的SQL语句执行更新

REFRESH MATERIALIZED VIEW

然而通过mybatis执行这条语句是却报语法错误，没有办法执行。

解决方法

可以通过mybatis的拦截器功能，判断如果是上面的语句，我们直接放行或通过原生连接执行，这样我们绕过校验就可以了。
下面是我们添加的拦截器

@Component
@Intercepts({
        @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})
})
public class DDLWhiteListSqlInterceptor implements Interceptor {

    private List<String> whiteList = Arrays.asList("REFRESH MATERIALIZED VIEW","DROP MATERIALIZED VIEW",
            "CREATE MATERIALIZED VIEW");

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        // 获取参数
        MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
        Object parameter = invocation.getArgs()[1];

        // 获取绑定的 SQL 语句
        String sql = mappedStatement.getBoundSql(parameter).getSql();

        // 检查 SQL 是否是 REFRESH MATERIALIZED VIEW
        if (this.isWhiteList(sql)) {
            // 获取数据库连接
            Executor target = (Executor) invocation.getTarget();
            Executor plugin = (Executor) Plugin.wrap(target, this);
            Connection connection = plugin.getTransaction().getConnection();

            // 直接执行 SQL
            try (PreparedStatement preparedStatement = connection.prepareStatement(sql)) {
                return preparedStatement.executeUpdate();
            }
        }

        // 如果不是 REFRESH MATERIALIZED VIEW 语句，则继续执行原方法
        return invocation.proceed();
    }

    private boolean isWhiteList(String sql){
        return whiteList.stream().anyMatch(sqlWhite->sql.toUpperCase().startsWith(sqlWhite));
    }


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

    @Override
    public void setProperties(Properties properties) {
        // 可以在这里设置拦截器的属性，如果需要的话
    }
}