Mybatis-plus解决兼容oracle批量插入的示例详解

目录

• 1. 自定义SQL注入器
• 2. 将SQL注入器交给Spring容器
• 3. 配置EasyBaseMapper继承BaseMapper
• 4.自定义MyBATis拦截器oracleSqlInterceptor

1. 自定义SQL注入器

新建一个名为EasySqlInjector的类，继承DefaultSqlInjector。

public class EasySqlInjector extends DefaultSqlInjector {
 
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        // 注意：此SQL注入器继承了DefaultSqlInjector(默认注入器)，调用了DefaultSqlInjector的getMethodList方法，保留了mybatis-plus的自带方法
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        methodList.add(new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));
        return methodList;
    }
 
}

2. 将SQL注入器交给Spring容器

在MybatisPlusConfig类中，将刚才创建的SQL注入器EasySqlInjector,注册为一个bean。

@EnableTransactionManagement(proxyTargetClass = true)
@Configuration
public class MybatisPlusConfig
{
    @Bean
    public PaginationInterceptor paginationInterceptor() {
        PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
        // 设置请求的页面大于最大页后操作， true调回到首页，false 继续请求 默认false
        // paginationInterceptor.setOverflow(false);
        // 设置最大单页限制数量，默认 500 条，-1 不受限制
        // paginationInterceptor.setLimit(500);
        // 开启 count 的 join 优化,只针对部分 left join
        paginationInterceptor.setLimit(-1);
//        paginationInterceptor.setCountSqlParser(new jsqlParserCountOptimize(true));
        return paginationInterceptor;
    }
 
    /**
     * 分页插件，自动识别数据库类型 https://baomidou.com/guide/interceptor-pagination.html
     */
    public PaginationInnerInterceptor paginationInnerInterceptor()
    {
        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
        // 设置数据库类型为mysql
        paginationInnerInterceptor.setDbType(DbType.ORACLE);
        // 设置最大单页限制数量，默认 500 条，-1 不受限制
        paginationInnerInterceptor.setMaxLimit(-1L);
        return paginationInnerInterceptor;
    }
 
    /**
     * 乐观锁插件 https://baomidou.com/guide/interceptor-optimistic-locker.html
     */
    public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor()
    {
        return new OptimisticLockerInnerInterceptor();
    }
 
    /**
     * 如果是对全表的删除或更新操作，就会终止该操作 https://baomidou.com/guide/interceptor-block-attack.html
     */
    public BlockAttackInnerInterceptor blockAttackInnerInterceptor()
    {
        return new BlockAttackInnerInterceptor();
    }
 
    @Bean
    public EasySqlInjector sqlInjector() {
        return new EasySqlInjector();
    }
 
}

3. 配置EasyBaseMapper继承BaseMapper

新建EasyBaseMapper类，继承BaseMapper，并在此类中配置insertBatchSomeColumn()方法。

public interface  EasyBaseMapper<T> extends BaseMapper<T> {
    /**
     * @param entityList 实体列表
     */
    void insertBatchSomeColumn(Collection<T> entityList);
}

4.自定义Mybatis拦截器OracleSqlInterceptor

这个地方要注意,表的主键我用触发器已经自动填入，所以keyGenerator设置为NoKeyGenerator.INSTANCE,这个地方有个坑，不那么设置，SQL一直报错，折腾了两个小时，实际上拼接的SQL没问题

@Component
@Slf4j
@Order(1)
@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class OracleSqlInterceptor implements Interceptor {
 
 
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        //当前业务，兼容pg 和 oracle，需要兼容oracle的批量插入语句
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
 
        BoundSql boundSql = statementHandler.getBoundSql();
        String sql = boundSql.getSql();
        StatementHandler delegate = (StatementHandler) ReflectUtil.getFieldValue(statementHandler, "delegate");
        MappedStatement mappedStatement = (MappedStatement) ReflectUtil.getFieldValue(delegate, "mappedStatement");
        ReflectUtil.setFieldValue(mappedStatement,"keyGenerator", NoKeyGenerator.INSTANCE);
        String mName  = mappedStatement.getId().substring(mappedStatement.getId().lastIndexOf(".") + 1);
        if("insertBatchSomeColumn".equals(mName)){
            //开始兼容批量插入语句,并设置boundSql
            Field declaredField = boundSql.getClass().getDeclaredField("sql");
            declaredField.setAccessible(true);
            declaredField.set(boundSql, convertOracleInsertSql(sql));
            log.info("---转换后的sql为：{}", boundSql.getSql());
        }
        return invocation.proceed();
    }
 
    /**
     * Oraclpythone Insert语句转化
     *
     * @param sql 传入的pg的sql
     * @return 转化后的sql
     */
    public String convertOracleInsertSql(String sql) {
        //用oracle中的批量语句代替
        //查找values的位置，将后面全部括号里的东西取出，然后再用对应的数据进行封装
        //获取前面的sql，这段sql与Oracle的相同
        String prefix = sql.substring(0, getKeywordValueIndex(sql));
        //排除table中的括号，取后面的括号
        String subSql = sql.substring(getKeywordValueIndex(sql));
        String valueSql = subSql.substring(subSql.indexOf("("));
    www.devze.com    List<String> valueList = getValues(valueSql);
 
        //拼接sql
        StringBuilder sqlBuilder = new StringBuilder().append(prefix);
        //sqlBuilder.append("SELECT A.* FROM (");
        String selectValue = "SELECT ";
        String endValue = " FROM DUAL ";
        String unionValue = "UNION ALL ";
        boolean start = true;
        for (String value : valueList) {
            if (!start) {
                sqlBuilder.append(unionValue);
            }
            else {
                start = false;
            }
            sqlBuilder.append(selectValue).append(value).append(endValue);
        }
        //sqlBuilder.append(") A");
        return sqlBuilder.toString();
    }
 
    /**
     * 使用栈实现获取value中括号的值
     *
     */
    public List<String> getValues(String sql) {
        List<String> values = new ArrayList<>();
        Stack<Character> brackets = new Stack<>();
        StringBuilder splitValue = new StringBuilder();
        for (Character c : sql.toCharArray()) {
            if ('('www.devze.com == c) {
                //左括号进栈
                brackets.push(c);
            }
            else if (')' == c) {
                //右括号则将左括号出栈,清空builder
                brackets.pop();
                values.add(splitValue.toString());
                splitValue.delete(0, splitValue.length());
       javascript     }
            else if (!brackets.empty()) {
                //只有进入括号中才将值放入，排除括号外的逗号
                splitValue.append(c);
            }
        }
        return values;
 
    }
 
    /**
     * 查找关键字value的位置
     */
    public  int getKeywordValueIndex(String sql) {
        //先找values，再找value
        if (sql.contains("values")) {
            return sql.indexOf("values");
        }
        else if (sql.contains("VALUES")) {
            return sql.indexOf("VALUES");
        }
        else if (sql.contains("value")) {
            return sql.indexOf("value");
        }
        else {
            return sql.indexOf("VALUE");
        }
    }
 
}

然后，用业务Mapper继承EasyBaseMapper就可以调用insertBatchSomeColumn()方法了。

到此这篇关于Mybatis-plus解决兼容oracle批量插入的示例详解的文章就介绍到这了http://www.devze.com,更多相关Mybatis-plus oracle批量插入内容请搜索编程客栈(www.devze.com)以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程客栈(www.devze.com)！