编写一个mybatis批处理工具类 用来执行业务的批量操作 而不用到xml里面写循环

 

不太科学  待后续优化  在事务情况下无法实现acid 

原理就是 使用工具类的时候单独得到项目环境的 SqlSessionFactory  然后用它开一个BATCH的session  然后进行操作 

 

1 定义一个枚举类 表示我们的批处理是什么操作

public enum OpreationType {

    INSERT,UPDATE,DELETE

}

 

2 写工具类

 

@Component
public class MyBatisBatchUtil {

    @Autowired
    private List<SqlSessionFactory> sqlSessionFactoryList;


    public List<?> executeBatch(OpreationType opreationType,List<?> param,Class mapper,String methodName){

        int size = sqlSessionFactoryList.size();
        if(size>1){
            throw  new RuntimeException("当前项目为多数据环境 请指明具体数据库");
        }
        String name = mapper.getName();
        String id=name+"."+methodName;
        SqlSessionFactory sqlSessionFactory = sqlSessionFactoryList.get(0);
        return  doOpreation(sqlSessionFactory,param,id,opreationType);

    }
    public List<?> executeBatch(OpreationType opreationType,List<?> param,Class mapper,String methodName,String sqlSessionFactoryId){
        
        String name = mapper.getName();
        String id=name+"."+methodName;
        SqlSessionFactory sqlSessionFactory = sqlSessionFactoryList.stream().filter(p->{
            return p.getConfiguration().getEnvironment().getId().equals(sqlSessionFactoryId);
        }).findFirst().orElseThrow(()->new RuntimeException("找不到对应的sqlSessionFactory"));
        return  doOpreation(sqlSessionFactory,param,id,opreationType);

    }


   private List<?> doOpreation( SqlSessionFactory sqlSessionFactory,List<?> params,String mapperId,OpreationType opreationType){
       List<Object> sucessfulResult=new ArrayList<>();
       SqlSession sqlSession =new SqlSessionTemplate(sqlSessionFactory,ExecutorType.BATCH);

       try {
           String mapperLocation=mapperId;
           switch (opreationType){

               case INSERT:
                   params.forEach(p->{
                       sqlSession.insert(mapperLocation, p);
                   });
                   break;
               case DELETE:
                   params.forEach(p->{
                       sqlSession.delete(mapperLocation,p);
                   });
                   break;
               case UPDATE:
                   params.forEach(p->{
                       sqlSession.update(mapperLocation,p);
                   });
                   break;
           }
           List<BatchResult> batchResults = sqlSession.flushStatements();
           batchResults.forEach(p->{
               int[] updateCounts = p.getUpdateCounts();
               sucessfulResult.addAll(Arrays.asList(updateCounts));
           });
           sqlSession.commit();
           sqlSession.close();
       } catch (Exception e){
           sqlSession.rollback();
           sqlSession.close();

           throw  new RuntimeException(e);
       }

       return sucessfulResult;



   }

}

 

使用方法：如下

 List<Shop> shops=new ArrayList<>();
 for (int i = 1; i <=30 ; i++) {
     Shop shop=new Shop();
     shop.setId(i);
     shop.setName("order"+i);
     shops.add(shop);
 }

List<?> result = myBatisBatchUtil.executeBatch(OpreationType.INSERT, shops, TestMapper2.class, "insert");

我的TestMapper2如下所示：

public interface TestMapper2 {
    int insert(Shop shop);
    int delete(Integer id);
}

 

请注意此种方式也需要我们的mapper被mybatis扫描到！