扩展tk.mybatis的批量更新的功能

tk.mybatis没有带批量更新的功能,批量更新却是经常使用的,所以需要自己实现。

批量更新网上主要有2种方式:case when方式、foreach方式

但是foreachzhe这种方式效率非常低下,不知道为何那多么帖子在流传,请看我另一个文章。

扩展tk.mybatis的批量更新,采用case when方式,源码干货如下:

  1. 首先定义下mapper接口
import org.apache.ibatis.annotations.UpdateProvider;

import java.util.List;

/**
 * 批量update
 *
 * @param <T> 不能为空
 */
@tk.mybatis.mapper.annotation.RegisterMapper
public interface UpdateBatchByPrimaryKeySelectiveMapper<T> {

    /**
     * 根据Example条件批量更新实体`record`包含的不是null的属性值
     *
     * @return
     */
    @UpdateProvider(type = BatchExampleProvider.class, method = "dynamicSQL")
    int updateBatchByPrimaryKeySelective(List<? extends T> recordList);

}
  1. 实现UpdatePriUpdateProvider,使用case when方式拼写mapper动态语句,必须id为主键
import org.apache.ibatis.mapping.MappedStatement;
import tk.mybatis.mapper.entity.EntityColumn;
import tk.mybatis.mapper.mapperhelper.EntityHelper;
import tk.mybatis.mapper.mapperhelper.MapperHelper;
import tk.mybatis.mapper.mapperhelper.SqlHelper;
import tk.mybatis.mapper.provider.ExampleProvider;

import java.util.Set;

/**
 * 批量更新的SqlProvider
 * @author sunchangtan
 */

public class BatchExampleProvider extends ExampleProvider {

    public BatchExampleProvider(Class<?> mapperClass, MapperHelper mapperHelper) {
        super(mapperClass, mapperHelper);
    }


    /**
     * 拼update sql, 使用case when方式,id为主键
     *
     * @param ms
     * @return
     */
    public String updateBatchByPrimaryKeySelective(MappedStatement ms) {
        final Class<?> entityClass = getEntityClass(ms);
        //开始拼sql
        StringBuilder sql = new StringBuilder();
        sql.append(SqlHelper.updateTable(entityClass, tableName(entityClass)));
        sql.append("<trim prefix=\"set\" suffixOverrides=\",\">");

        //获取全部列
        Set<EntityColumn> columnList = EntityHelper.getColumns(entityClass);
        for (EntityColumn column : columnList) {
            if (!column.isId() && column.isUpdatable()) {
                sql.append("  <trim prefix=\""+column.getColumn()+" =case\" suffix=\"end,\">");
                sql.append("    <foreach collection=\"list\" item=\"i\" index=\"index\">");
                sql.append("      <if test=\"i."+column.getEntityField().getName()+"!=null\">");
                sql.append("         when id=#{i.id} then #{i."+column.getEntityField().getName()+"}");
                sql.append("      </if>");
                sql.append("    </foreach>");
                sql.append("  </trim>");
            }
        }

        sql.append("</trim>");
        sql.append("WHERE");
        sql.append(" id IN ");
        sql.append("<trim prefix=\"(\" suffix=\")\">");
        sql.append("<foreach collection=\"list\" separator=\", \" item=\"i\" index=\"index\" >");
        sql.append("#{i.id}");
        sql.append("</foreach>");
        sql.append("</trim>");

        return sql.toString();
    }
}

  1. 统一定义批量操作接口
/**
 * 批量操作接口
 *
 * @param <T>
 * @author sunchangtan
 */
@tk.mybatis.mapper.annotation.RegisterMapper
public interface BatchMapper<T> extends
        UpdateBatchByPrimaryKeySelectiveMapper<T> {
}
  1. 使用例子:
List<List<BatchTest>> splitList = ListUtils.splitList(list, 100);
        splitList.forEach(data -> {
            batchTestMapper.updateBatchByPrimaryKeySelective(data);
        });
你可以使用Tk.mybatis的`@UpdateProvider`注解来实现批量更新。 首先,创建一个Mapper接口,使用`@UpdateProvider`注解来指定更新操作的提供者类和方法。在提供者类中编写批量更新的SQL语句。 ```java @Mapper public interface MyMapper { @UpdateProvider(type = MyBatchUpdateProvider.class, method = "dynamicSQL") void batchUpdate(List<MyEntity> entityList); } ``` 然后,创建一个提供者类,并实现相应的方法来生成动态SQL语句。 ```java public class MyBatchUpdateProvider { public String batchUpdate(Map<String, Object> map) { List<MyEntity> entityList = (List<MyEntity>) map.get("list"); StringBuilder sb = new StringBuilder(); sb.append("UPDATE my_table SET "); sb.append("<foreach collection='list' item='entity' separator=','>"); sb.append("column1 = #{entity.column1},"); sb.append("column2 = #{entity.column2}"); sb.append("</foreach>"); sb.append("WHERE id IN "); sb.append("<foreach collection='list' item='entity' open='(' close=')' separator=','>"); sb.append("#{entity.id}"); sb.append("</foreach>"); return sb.toString(); } } ``` 在上述示例中,我们使用了`<foreach>`标签来循环遍历`entityList`,生成批量更新的SQL语句。 最后,你可以在需要进行批量更新操作的地方注入`MyMapper`接口,并调用`batchUpdate`方法即可。 ```java @Autowired private MyMapper myMapper; public void updateBatch(List<MyEntity> entityList) { myMapper.batchUpdate(entityList); } ``` 这样,你就可以使用Tk.mybatis实现批量更新操作了。
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值