批量插入,可以把Mybatis-Plus看作是Mybatis加强版;故Mybatis中的相关操作都可以在Mybatis-Plus中使用;在mysql数据库中支持批量插入,所以只要配置useGeneratedKeys和keyProperty就可以批量插入并返回主键了。
下面是批量插入的Dao层接口
一注解方式:
直接撸代码:
package org.jd.data.netty.big.window.chat.frame.ui.room.mapper;import com.baomidou.mybatisplus.core.mapper.BaseMapper; import org.apache.ibatis.annotations.Insert; import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Options; import org.apache.ibatis.annotations.Param; import org.jd.data.netty.big.window.chat.entity.SysRandomColor;import java.util.List;/*** <p>@Description:Mapper随机颜色服务接口</p>* <p>@Copyright: Copyright (c)2,023</p>* <p>@Company: 北京惠农科技有限公司</p>* <P>@Created Date :2023年03月19日</P>* <P>@version 1.0* <p>@author yh19166</p>*/ @Mapper public interface SysRandomColorMapper extends BaseMapper<SysRandomColor> { /*** 批量插入** @param sysRandomColorList 集合参数* @return {@link Integer} 返回插入的总条数*/@Insert("<script>" +" insert into sys_random_color (color_zh_name,color_cn_name,color_hexadecimal,color_rgb_value,create_time,update_time) values " +" <foreach" +" collection='aList' item='c' separator=','>(#{c.colorZhName},#{c.colorCnName},#{c.colorHexadecimal},#{c.colorRgbValue},#{c.createTime},#{c.updateTime})" +" </foreach>" +"</script>")@Options(useGeneratedKeys = true, keyProperty = "colorId")int batchSave(@Param("aList") List<SysRandomColor> sysRandomColorList);}
第二种: XML方式
<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> <mapper namespace="org.jd.data.netty.big.window.chat.frame.ui.room.mapper.SysRandomColorMapper"><insert id="batchSave" useGeneratedKeys="true" keyProperty="colorId">insert into sys_random_color (color_zh_name,color_cn_name,color_hexadecimal,color_rgb_value,create_time,update_time)values<foreach collection="aList" item="c" separator=",">(#{c.colorZhName},#{c.colorCnName},#{c.colorHexadecimal},#{c.colorRgbValue},#{c.createTime},#{c.updateTime})</foreach></insert> </mapper>
测试实例:
package org.jd.data.netty.big.window.chat.entity;import com.baomidou.mybatisplus.annotation.IdType; import com.baomidou.mybatisplus.annotation.TableField; import com.baomidou.mybatisplus.annotation.TableId; import com.baomidou.mybatisplus.annotation.TableName; import lombok.*;import java.io.Serializable; import java.util.Date;/*** <p>@Description: 顾客入住信息领域模型</p>* <p>@Copyright: Copyright (c)2,023</p>* <p>@Company: 北京惠农科技有限公司</p>* <P>@Created Date :2023年03月19日</P>* <P>@version 1.0* <p>@author yh19166</p>*/ @TableName("sys_random_color") @Data @NoArgsConstructor public class SysRandomColor implements Serializable { /*** =========================================================================================* 注解名称 功能* @Setter 自动添加类中所有属性相关的 set 方法* @Getter 自动添加类中所有属性相关的 get 方法* @Builder 使得该类可以通过 builder (建造者模式)构建对象* @RequiredArgsConstructor 生成一个该类的构造方法,禁止无参构造* @ToString 重写该类的toString()方法* @EqualsAndHashCode 重写该类的equals()和hashCode()方法* @Data 等价于上面的@Setter、@Getter、@RequiredArgsConstructor、@ToString、@EqualsAndHashCode* ===========================================================================================*/@TableField(exist = false)// 不需要为该字段生成get方法@Getter(AccessLevel.NONE)@Setter(AccessLevel.NONE)private static final long serialVersionUID = 1L;/*** 颜色ID*/@TableId(value = "color_id", type = IdType.AUTO)private Long colorId;/*** 中文名称*/@TableField(value = "color_zh_name")private String colorZhName;/*** 英文名称*/@TableField(value = "color_cn_name")private String colorCnName;/*** 16进制*/@TableField(value = "color_hexadecimal")private String colorHexadecimal;/*** 颜色RGB值*/@TableField(value = "color_rgb_value")private String colorRgbValue;/*** 创建时间*/private Date createTime;/*** 更新时间*/private Date updateTime;@Overridepublic String toString() {return "SysRandomColor{" +", colorId = " + colorId +", colorZhName = " + colorZhName +", colorCnName = " + colorCnName +", colorHexadecimal = " + colorHexadecimal +", colorRgbValue = " + colorRgbValue +", createTime = " + createTime +", updateTime = " + updateTime +"}";} }
测试类:
package org.jd.data.frame.big.window.service;import cn.hutool.core.convert.Convert; import org.jd.data.netty.big.window.chat.entity.SysRandomColor; import org.jd.data.netty.big.window.chat.frame.ui.room.provider.SysRandomColorProvider; import org.jd.data.netty.big.window.chat.util.FileReaderLineUtil; import org.jd.data.netty.big.window.chat.util.handler.BeanInstanceFactoryHandler; import org.junit.Before; import org.junit.Test;import java.util.ArrayList; import java.util.List; import java.util.Optional;public class SysRandomColorProviderTest {private Optional<SysRandomColorProvider> sysRandomColorProviderInstance;@Beforepublic void init() {sysRandomColorProviderInstance = BeanInstanceFactoryHandler.getBeanInstance(SysRandomColorProvider.class);} @Testpublic void saveSysRandomColor() {if (sysRandomColorProviderInstance.isPresent()) {SysRandomColorProvider sysRandomColorProvider = sysRandomColorProviderInstance.get();Optional<List<FileReaderLineUtil.RandomColor>> result = FileReaderLineUtil.productColor();if (result.isPresent()) {List<SysRandomColor> targetList = new ArrayList<>();result.get().stream().forEach(e -> {// 相同字段属性间的拷贝SysRandomColor sysRandomColor = Convert.convert(SysRandomColor.class, e);targetList.add(sysRandomColor);});int n = sysRandomColorProvider.batchSaveRandomColor(targetList);System.out.println("返回插入的总条数: " + n);}}}@Testpublic void list() {if (sysRandomColorProviderInstance.isPresent()) {SysRandomColorProvider sysRandomColorProvider = sysRandomColorProviderInstance.get();List<SysRandomColor> resultList = sysRandomColorProvider.list(null, null);System.out.println("resultList: "+resultList);}} }
数据提供者:
package org.jd.data.netty.big.window.chat.frame.ui.room.provider;import org.jd.data.netty.big.window.chat.entity.SysRandomColor; import org.jd.data.netty.big.window.chat.frame.ui.room.service.SysRandomColorService; import org.jd.data.netty.big.window.chat.frame.ui.room.service.impl.SysRandomColorServiceImpl; import org.jd.data.netty.big.window.chat.util.instance.annotion.AutoInstanceServiceBean;import java.util.List;public class SysRandomColorProvider {private SysRandomColorService sysRandomColorService;@AutoInstanceServiceBean(value = SysRandomColorServiceImpl.class)public void setSysRandomColorService(SysRandomColorService sysRandomColorService) {this.sysRandomColorService = sysRandomColorService;}public int saveRandomColor(SysRandomColor sysRandomColor){return sysRandomColorService.saveSysRandomColor(sysRandomColor);}public int batchSaveRandomColor(List<SysRandomColor> randomColorList){return sysRandomColorService.batchSaveRandomColor(randomColorList);}public List<SysRandomColor> list(String colorZhName,String colorCnName){return sysRandomColorService.list(colorZhName,colorCnName);} }
服务接口及实现类:
package org.jd.data.netty.big.window.chat.frame.ui.room.service;import org.jd.data.netty.big.window.chat.entity.SysRandomColor; import java.util.List;/*** <p>@Description: 随机颜色服务接口</p>* <p>@Copyright: Copyright (c)2,023</p>* <p>@Company: 北京惠农科技有限公司</p>* <P>@Created Date :2023年03月19日</P>* <P>@version 1.0* <p>@author yh19166</p>*/ public interface SysRandomColorService{int saveSysRandomColor(SysRandomColor sysRandomColor);int batchSaveRandomColor(List<SysRandomColor> sysRandomColorList);/*** 获取所有随机颜色记录* @param colorZhName* @param colorCnName* @return {@link List<SysRandomColor>}*/List<SysRandomColor> list(String colorZhName,String colorCnName);}
package org.jd.data.netty.big.window.chat.frame.ui.room.service.impl;import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import org.apache.ibatis.session.SqlSession; import org.jd.data.netty.big.window.chat.entity.SysRandomColor; import org.jd.data.netty.big.window.chat.frame.ui.contant.SsyConstant; import org.jd.data.netty.big.window.chat.frame.ui.room.mapper.SysRandomColorMapper; import org.jd.data.netty.big.window.chat.frame.ui.room.service.SysRandomColorService; import org.jd.data.netty.big.window.chat.util.ProxyDruidPoolUtil;import java.util.List;/*** <p>@Description: 房间类型信息服务实现类</p>* <p>@Copyright: Copyright (c)2,023</p>* <p>@Company: 北京惠农科技有限公司</p>* <P>@Created Date :2023年03月19日</P>* <P>@version 1.0* <p>@author yh19166</p>*/ public class SysRandomColorServiceImpl implements SysRandomColorService {private SysRandomColorMapper sysRandomColorMapper;@Overridepublic int saveSysRandomColor(SysRandomColor sysRandomColor) {SqlSession session = ProxyDruidPoolUtil.getSession();int result;try {sysRandomColorMapper = session.getMapper(SysRandomColorMapper.class);result = sysRandomColorMapper.insert(sysRandomColor);session.commit();} catch (Exception e) {e.printStackTrace();result = SsyConstant.INSERT_DATABASE_ERROR;session.rollback();} finally {sysRandomColorMapper = null; // 帮助垃圾回收session.close();}return result;} /*** 返回总条数** @param sysRandomColorList* @return*/@Overridepublic int batchSaveRandomColor(List<SysRandomColor> sysRandomColorList) {SqlSession session = ProxyDruidPoolUtil.getSession();int result;try {sysRandomColorMapper = session.getMapper(SysRandomColorMapper.class);result = sysRandomColorMapper.batchSave(sysRandomColorList);session.commit();} catch (Exception e) {session.rollback();result = SsyConstant.INSERT_DATABASE_ERROR;} finally {sysRandomColorMapper = null; // 帮助垃圾回收session.close();}return result;} @Overridepublic List<SysRandomColor> list(String colorZhName, String colorCnName) {SqlSession session = ProxyDruidPoolUtil.getSession();try {QueryWrapper<SysRandomColor> queryWrapper = new QueryWrapper<>();queryWrapper.lambda().eq(StrUtil.isNotEmpty(colorCnName), SysRandomColor::getColorCnName, colorCnName).eq(StrUtil.isNotEmpty(colorZhName), SysRandomColor::getColorZhName, colorZhName).orderByAsc(SysRandomColor::getCreateTime);sysRandomColorMapper = session.getMapper(SysRandomColorMapper.class);return sysRandomColorMapper.selectList(queryWrapper);} catch (Exception e) {return null;} finally {sysRandomColorMapper = null;session.close();}}}