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) {
        // 可以在这里设置拦截器的属性,如果需要的话
    }
}
  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ThinkLess404

有问题可以私信交流

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值