问题
这个问题是本人在开发一个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) {
// 可以在这里设置拦截器的属性,如果需要的话
}
}