遇到的问题

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

解决思路

参考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只能获取第一个

发现上述步骤的Jdbc3KeyGenerator是有接口的KeyGenerator,查看后发现有3个实现类了解到SelectKeyGenerator通过selectKeyt标签可以返回id；
查看SelectKeyGenerator源码了解到只支持返回单个id；
通过百度了解到SqlServer通过output inserted.id可以输出id；
是否可以自己实现KeyGenerator来解决批量返回id的方法呢?
最后我没找到配置自定义的实现类的方式,决定通过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 {@Overridepublic 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]);// 封装idfor (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;}@Overridepublic Object plugin(Object target) {return null;}@Overridepublic 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.idvalues<foreach collection="list" separator="," item="item">(#{item.name}, #{itme.password})</foreach><selectKey>
</insert>

PS:需要注意的是id对象参数必须放在第一位,拦截器的代码写的比较粗糙,给各位提供思路.有更好方案的可以留言.

mybatis-sqlserver批量新增返回id相关推荐

Mybatis Mysql 批量插入返回id
1. 目标: 使用Mybatis 批量插入数据返回自增的id 2. 具体实现注意: MyBatis版本3.3.1或者以上 <insert id="save" useGene ...
Mybatis执行批量插入返回数据库主键列表
原文地址 [推荐阅读]微服务还能火多久?>>> 我们都知道Mybatis在插入单条数据的时候有两种方式返回自增主键: 1.对于支持生成自增主键的数据库:增加 useGenerateK ...
orcal 批量新增，批量更新多条件
批量新增 <insert id="savePatientEmr" parameterType="java.util.List" useGeneratedK ...
Mybatis新增返回主键ID
在做项目的过程中有时候难免会运用到新增一条数据并获取到新增的这条数据的主键ID,这样的一个功能,新增一条订单数据,并返回新增订单的ID.这样做的前提是数据的ID必须为自增. 我是用ssm框架做的项目, ...
mybatis插入数据后返回自增主键ID详解
1.场景介绍: 开发过程中我们经常性的会用到许多的中间表,用于数据之间的对应和关联.这个时候我们关联最多的就是ID,我们在一张表中插入数据后级联增加到关联表中.我们熟知的mybatis在插入数据后 ...
mybatis多表新增如何获取主键ID
记录一下碰到的问题,就当做是笔记吧,防止忘记.在做ssm多表新增的时候,碰到了一个小问题,就是在新增成功一个表的时候我要获取它的ID,然后用它的ID新增下一个表. 但是,我要怎么做呢?新增成功,再查询 ...
mybatis高级查询，批量新增
review sql脚本实体类 sql watch out mapper mapper test 之前的比较分散,自己用... sql脚本 -- auto-generated definition ...
mybatis批量新增和批量更新的效率对比
今天,为了更多了解下,mybatis批量新增和批量更新在simple/batch模式 + MySQL的rewriteBatchedStatements下效率有什么区别,做了一次实验. 实验结果,让人意 ...
基于ruoyi+vue+elementUI实现列表，新增，附件上传，tab+springBoot+mybatis+oracle序列+批量新增
基于ruoyi+vue+elementUI实现列表,新增,附件上传,tab+springBoot+mybatis+oracle序列+批量新增页面效果列表页面新增页面详情页面代码实现列表+新 ...

mybatis-sqlserver批量新增返回id

mybatis-SqlServer批量新增返回id

遇到的问题

解决思路

代码

mybatis-sqlserver批量新增返回id相关推荐

最新文章

热门文章