mybatis-sqlserver批量新增返回id方式

 更新时间:2024年05月18日 09:51:52   作者:rong0016  
这篇文章主要介绍了mybatis-sqlserver批量新增返回id方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

遇到的问题

公司最近接到项目需要使用SqlServer,在做SQL兼容的时候遇到问题,批量新增数据时只返回的第一条记录的id

解决思路

1.参考mysql

<insert id="batchInsert" userGeneratedKeys="true" keyProperty="id">
   insert into public_user (name,password)
   values
   <foreach collection="list" separator="," item="item">
   (#{item.name}, #{itme.password})
   </foreach>
</insert>

具体mybatis封装id的地方在org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator

主要逻辑是通过Statement.getGeneratedKeys()获取id结果集.

SqlServer只能获取第一个

2.发现上述步骤的Jdbc3KeyGenerator是有接口的KeyGenerator,查看后发现有3个实现类了解到SelectKeyGenerator通过selectKeyt标签可以返回id;

3.查看SelectKeyGenerator源码了解到只支持返回单个id;

4.通过百度了解到SqlServer通过output inserted.id可以输出id;

5.是否可以自己实现KeyGenerator来解决批量返回id的方法呢?

6.最后我没找到配置自定义的实现类的方式,决定通过mybatis的拦截器解决

代码

mybatis拦截器

@Component
@Intercepts(@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}))
public class SqlServerKeyGeneratorInterceptor implements Interceptor {
   @Override
   public Object intercept(Invocation invocation) throws Throwable {
       // 正常执行代码 获取返回结构集
       List<Object> values = (List)invocation.proceed();
       MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
       // 判断是否selectKey查询语句
       if (ms.getId().endsWith(SelectKeyGenerator.SELECT_KEY_SUFFIX)) {
           Configuration configuration = ms.getConfiguration();
           // 处理入参对象
           Collection<Object> parameters = getParameters(invocation.getArgs()[1]);
           // 封装id
           for (int i = 0; i < parameters.size(); i++) {
               MetaObject metaObject = configuration.newMetaObject(parameters.toArray()[i]);
               metaObject.setValue("id", Long.valueOf((Integer)(((Map)values.get(i)).get("id"))));
           }
           // 返回假数据防止异常
           List<Long> ids = new ArrayList<>();
           ids.add(1L);
           return ids;
       }
       return values;
   }

   @Override
   public Object plugin(Object target) {
       return null;
   }

   @Override
   public void setProperties(Properties properties) {

   }

   /** 来源 Jdbc3KeyGenerator */
   private Collection<Object> getParameters(Object parameter) {
       Collection<Object> parameters = null;
       if (parameter instanceof Collection) {
           parameters = (Collection) parameter;
       } else if (parameter instanceof Map) {
           Map parameterMap = (Map) parameter;
           if (parameterMap.containsKey("collection")) {
               parameters = (Collection) parameterMap.get("collection");
           } else if (parameterMap.containsKey("list")) {
               parameters = (List) parameterMap.get("list");
           } else if (parameterMap.containsKey("array")) {
               parameters = Arrays.asList((Object[]) parameterMap.get("array"));
           }
       }
       if (parameters == null) {
           parameters = new ArrayList<Object>();
           parameters.add(parameter);
       }
       return parameters;
   }
}

mapper

<insert id="batchInsert">
	<selectKey keyColumn="id" keyProperty="id" order="AFTER" resultType="java.util.Map">
		insert into public_user (name,password)
		output inserted.id
		values
		<foreach collection="list" separator="," item="item">
		(#{item.name}, #{itme.password})
		</foreach>
	<selectKey>
</insert>

PS:

需要注意的是id对象参数必须放在第一位,拦截器的代码写的比较粗糙,给各位提供思路。

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 使用Java将Excel转换为Text的实现方法

    使用Java将Excel转换为Text的实现方法

    在日常的数据处理和系统集成中,我们经常会遇到需要将Excel文件中的数据提取出来,并以纯文本格式进行处理的场景,将Excel转换为Text是一项基础而重要的任务,本文将向您介绍如何利用强大的 Spire.XLS for Java 库,在Java应用程序中轻松实现Excel到Text的转换
    2025-10-10
  • 详解Java 中的UnitTest 和 PowerMock

    详解Java 中的UnitTest 和 PowerMock

    这篇文章主要介绍了Java中的 UnitTest 和 PowerMock,文中讲解非常详细,对大家学习有很大的帮助,感兴趣的朋友可以了解下
    2020-06-06
  • java批量下载生成zip压缩包的思路详解

    java批量下载生成zip压缩包的思路详解

    这篇文章主要介绍了java批量下载生成zip压缩包的思路详解,设计思路大概是本地先创建一个zip文件,将批量下载的文件依次放入zip文件中,将zip文件返回给前端,本文结合实例代码讲解的非常详细,需要的朋友可以参考下
    2024-01-01
  • JavaEE的进程,线程和创建线程的5种方式详解

    JavaEE的进程,线程和创建线程的5种方式详解

    这篇文章主要为大家详细介绍了JavaEE的进程,线程和创建线程的5种方式,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助
    2022-03-03
  • 解决idea中maven项目无端显示404错误的方法

    解决idea中maven项目无端显示404错误的方法

    这篇文章主要介绍了解决idea中maven项目无端显示404错误的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • SpringBoot两种方式接入DeepSeek的实现

    SpringBoot两种方式接入DeepSeek的实现

    本文主要介绍了SpringBoot两种方式接入DeepSeek的实现,包括HttpClient方式和基于spring-ai-openai的方式,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2025-03-03
  • Java8中利用stream对map集合进行过滤的方法

    Java8中利用stream对map集合进行过滤的方法

    这篇文章主要给大家介绍了关于Java8中利用stream对map集合进行过滤的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2018-07-07
  • SpringBoot项目如何打可执行war包

    SpringBoot项目如何打可执行war包

    最近小编做了一个springboot项目,最后需要打成war包在容器中部署,下面小编给大家分享下SpringBoot项目如何打可执行war包,感兴趣的朋友一起看看吧
    2020-04-04
  • 基于ElasticSearch Analyzer的使用规则详解

    基于ElasticSearch Analyzer的使用规则详解

    这篇文章主要介绍了基于ElasticSearch Analyzer的使用规则,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2021-07-07
  • SpringBoot中的异步任务解析

    SpringBoot中的异步任务解析

    这篇文章主要介绍了SpringBoot中的异步任务解析,SpringBoot 异步任务是一种在SpringBoot框架中使用的异步处理机制,可以提高系统的并发能力和响应速度,需要的朋友可以参考下
    2023-10-10

最新评论