服务器之家：专注于服务器技术及软件下载分享
分类导航

PHP教程|ASP.NET教程|Java教程|ASP教程|编程技术|正则表达式|C/C++|IOS|C#|Swift|Android|VB|R语言|JavaScript|易语言|vb.net|

服务器之家 - 编程语言 - Java教程 - mybatisplus添加真正的批量新增、批量更新的实现

mybatisplus添加真正的批量新增、批量更新的实现

2021-08-29 12:43@lehao Java教程

这篇文章主要介绍了mybatisplus添加真正的批量新增、批量更新的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

使用mybatis-plus来进行批量新增和更新时，你会发现其实是一条条sql执行,下面进行优化。

1.添加InsertBatchMethod和UpdateBatchMethod类

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.executor.keygen.NoKeyGenerator;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;
 
/**
 * 批量插入方法实现
 */
@Slf4j
public class InsertBatchMethod extends AbstractMethod {
  @Override
  public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
    final String sql = "<script>insert into %s %s values %s</script>";
    final String fieldSql = prepareFieldSql(tableInfo);
    final String valueSql = prepareValuesSql(tableInfo);
    final String sqlResult = String.format(sql, tableInfo.getTableName(), fieldSql, valueSql);
    //log.debug("sqlResult----->{}", sqlResult);
    SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
    return this.addInsertMappedStatement(mapperClass, modelClass, "insertBatch", sqlSource, new NoKeyGenerator(), null, null);
  }
 
  private String prepareFieldSql(TableInfo tableInfo) {
    StringBuilder fieldSql = new StringBuilder();
    fieldSql.append(tableInfo.getKeyColumn()).append(",");
    tableInfo.getFieldList().forEach(x -> fieldSql.append(x.getColumn()).append(","));
    fieldSql.delete(fieldSql.length() - 1, fieldSql.length());
    fieldSql.insert(0, "(");
    fieldSql.append(")");
    return fieldSql.toString();
  }
 
  private String prepareValuesSql(TableInfo tableInfo) {
    final StringBuilder valueSql = new StringBuilder();
    valueSql.append("<foreach collection=\"list\" item=\"item\" index=\"index\" open=\"(\" separator=\"),(\" close=\")\">");
    valueSql.append("#{item.").append(tableInfo.getKeyProperty()).append("},");
    tableInfo.getFieldList().forEach(x -> valueSql.append("#{item.").append(x.getProperty()).append("},"));
    valueSql.delete(valueSql.length() - 1, valueSql.length());
    valueSql.append("</foreach>");
    return valueSql.toString();
  }
}
?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;
 
/**
 * 批量更新方法实现，条件为主键，选择性更新
 */
@Slf4j
public class UpdateBatchMethod extends AbstractMethod {
  @Override
  public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
    String sql = "<script>\n<foreach collection=\"list\" item=\"item\" separator=\";\">\nupdate %s %s where %s=#{%s} %s\n</foreach>\n</script>";
    String additional = tableInfo.isWithVersion() ? tableInfo.getVersionFieldInfo().getVersionOli("item", "item.") : "" + tableInfo.getLogicDeleteSql(true, true);
    String setSql = sqlSet(tableInfo.isWithLogicDelete(), false, tableInfo, false, "item", "item.");
    String sqlResult = String.format(sql, tableInfo.getTableName(), setSql, tableInfo.getKeyColumn(), "item." + tableInfo.getKeyProperty(), additional);
    //log.debug("sqlResult----->{}", sqlResult);
    SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
    // 第三个参数必须和RootMapper的自定义方法名一致
    return this.addUpdateMappedStatement(mapperClass, modelClass, "updateBatch", sqlSource);
  }
}

2.添加自定义方法SQL注入器

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
 
import java.util.List;
 
public class CustomizedSqlInjector extends DefaultSqlInjector {
  /**
   * 如果只需增加方法，保留mybatis plus自带方法，
   * 可以先获取super.getMethodList()，再添加add
   */
  @Override
  public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
    List<AbstractMethod> methodList = super.getMethodList(mapperClass);
    methodList.add(new InsertBatchMethod());
    methodList.add(new UpdateBatchMethod());
    return methodList;
  }
}

3.注入配置

?
1
2
3
4
5
6
7
8
@MapperScan("com.xxx.mapper")
@Configuration
public class MyBatisPlusConfig {
  @Bean
  public CustomizedSqlInjector customizedSqlInjector() {
    return new CustomizedSqlInjector();
  }
}

4.添加通用mapper

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.Param;
 
import java.util.List;
 
/**
 * 根Mapper，给表Mapper继承用的，可以自定义通用方法
 * {@link com.baomidou.mybatisplus.core.mapper.BaseMapper}
 * {@link com.baomidou.mybatisplus.extension.service.IService}
 * {@link com.baomidou.mybatisplus.extension.service.impl.ServiceImpl}
 */
public interface RootMapper<T> extends BaseMapper<T> {
 
  /**
   * 自定义批量插入
   * 如果要自动填充，@Param(xx) xx参数名必须是 list/collection/array 3个的其中之一
   */
  int insertBatch(@Param("list") List<T> list);
 
  /**
   * 自定义批量更新，条件为主键
   * 如果要自动填充，@Param(xx) xx参数名必须是 list/collection/array 3个的其中之一
   */
  int updateBatch(@Param("list") List<T> list);
}

5.如何使用

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
@Repository
public interface UserInfoMapper extends RootMapper<UserInfo> {
}
 
public interface UserInfoService extends IService<UserInfo> {
  int saveAll();
  int updateAll();
}
 
@Service
public class UserInfoServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo> implements UserInfoService{
 
  @Override
  public int saveAll() {
    List<UserInfo> list = new ArrayList<>();
    for (int i = 0; i < 20; i++) {
      UserInfo userInfo = new UserInfo();
      userInfo.setUserName("厉害" + i);
      userInfo.setSalt(RandomStringUtils.randomAlphabetic(6));
      userInfo.setPassword(SecureUtil.sha256("123456" + userInfo.getSalt()));
      userInfo.setSex(0);
      userInfo.setAvatar(LoginServiceImpl.AVATAR);
      list.add(userInfo);
    }
    return baseMapper.insertBatch(list);
  }
 
  @Override
  public int updateAll() {
    List<UserInfo> userInfos = baseMapper.selectList(Wrappers.<UserInfo>lambdaQuery().between(BaseEntity::getId, 43, 62));
    userInfos.forEach(userInfo -> {
      userInfo.setUserName("更新了" + IdUtil.simpleUUID());
    });
    return baseMapper.updateBatch(userInfos);
  }
}

到此这篇关于mybatisplus添加真正的批量新增、批量更新的实现的文章就介绍到这了,更多相关mybatisplus批量新增、批量更新内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！

原文链接：https://blog.csdn.net/qq_18432653/article/details/110875608

  • MybatisPlus
  • 批量更新
  • 批量新增

    • 延伸 · 阅读

    精彩推荐
    • Java教程xml与Java对象的转换详解

      xml与Java对象的转换详解

      这篇文章主要介绍了xml与Java对象的转换详解的相关资料,需要的朋友可以参考下...

      Java教程网2942020-09-17
    • Java教程20个非常实用的Java程序代码片段

      20个非常实用的Java程序代码片段

      这篇文章主要为大家分享了20个非常实用的Java程序片段，对java开发项目有所帮助，感兴趣的小伙伴们可以参考一下 ...

      lijiao5352020-04-06
    • Java教程Java8中Stream使用的一个注意事项

      Java8中Stream使用的一个注意事项

      最近在工作中发现了对于集合操作转换的神器，java8新特性 stream，但在使用中遇到了一个非常重要的注意点，所以这篇文章主要给大家介绍了关于Java8中S...

      阿杜7472021-02-04
    • Java教程Java使用SAX解析xml的示例

      Java使用SAX解析xml的示例

      这篇文章主要介绍了Java使用SAX解析xml的示例，帮助大家更好的理解和学习使用Java，感兴趣的朋友可以了解下...

      大行者10067412021-08-30
    • Java教程升级IDEA后Lombok不能使用的解决方法

      升级IDEA后Lombok不能使用的解决方法

      最近看到提示IDEA提示升级，寻思已经有好久没有升过级了。升级完毕重启之后，突然发现好多错误，本文就来介绍一下如何解决，感兴趣的可以了解一下...

      程序猿DD9332021-10-08
    • Java教程Java BufferWriter写文件写不进去或缺失数据的解决

      Java BufferWriter写文件写不进去或缺失数据的解决

      这篇文章主要介绍了Java BufferWriter写文件写不进去或缺失数据的解决方案，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望...

      spcoder14552021-10-18
    • Java教程小米推送Java代码

      小米推送Java代码

      今天小编就为大家分享一篇关于小米推送Java代码，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧...

      富贵稳中求8032021-07-12
    • Java教程Java实现抢红包功能

      Java实现抢红包功能

      这篇文章主要为大家详细介绍了Java实现抢红包功能，采用多线程模拟多人同时抢红包，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙...

      littleschemer13532021-05-16
    最近更新
    编辑推荐
    阅读排行
    热门标签
    501 　 406 　 415 　 1433 　 2000 　 401 　 330 　 GOF23 　 堆栈类 　 微信公众号 　 注册服务 　 嵌套循环 　 critical 　 consul 　 spring 　 集合 　 volatile 　 Hibernate 　 Java 　 new 　 double 　 泛型 　 Java编程 　 遍历 　 设计模式 　 supplyAsync 　 制表符 　 回车 　 写入文件 　 JAVA8 　
    © 2019-2024 服务器之家(www.zzvips.com) 版权所有关于我们联系我们版权申明网站地图