MPG代码生成器生成的Service接口继承了MP的IService接口,该接口除了基本的增删改查外,还包含了一些批量操作方法,该接口中定义的方法除了default方法外都在ServiceImpl类中实现了
public interface ITeslaService extends IService<Tesla> {
}
// 插入一条记录(选择字段,策略插入), 直接调用的BaseMapper的insert()方法
default boolean save(T entity) {
// 插入(批量)
default boolean saveBatch(Collection<T> entityList)
// 插入(批量)
boolean saveBatch(Collection<T> entityList, int batchSize);
// 批量修改插入
default boolean saveOrUpdateBatch(Collection<T> entityList) {
// 批量修改插入
boolean saveOrUpdateBatch(Collection<T> entityList, int batchSize);
在ITeslaServiceTest测试类中增加相应的测试方法
@Test
public void saveBatch(){
List<Tesla> teslaList = new ArrayList<>();
for (int i = 0; i < 5; i++) {
Tesla tesla = new Tesla();
tesla.setName("Cyber Truck 202" + i);
tesla.setFactory("得克萨斯州特斯拉超级工厂");
tesla.setVehicleType("皮卡");
tesla.setPrice(300000.00 + 1000 * i);
teslaList.add(tesla);
}
boolean b = teslaService.saveBatch(teslaList);
System.out.println("是否保存成功:" + b);
}
执行该测试方法
保存成功,控制台只执行了一次SQL将数据全部保存到表中
@Test
public void saveBatchByBatchSize(){
List<Tesla> teslaList = new ArrayList<>();
for (int i = 0; i < 5; i++) {
Tesla tesla = new Tesla();
tesla.setName("Cyber Truck 202" + i);
tesla.setFactory("上海特斯拉超级工厂");
tesla.setVehicleType("皮卡");
tesla.setPrice(300000.00 + 1000 * i);
teslaList.add(tesla);
}
boolean b = teslaService.saveBatch(teslaList, 2);
System.out.println("是否保存成功:" + b);
}
执行该测试方法
批量保存时设置类batchSize,既每次执行批量插入时只插入两条数据,因此控制台执行了3次SQL语句
@Test
public void saveOrUpdateBatch(){
List<Tesla> updateList = new ArrayList<>();
for (int i = 1166057549; i < 1166057551; i++) {
Tesla tesla = new Tesla();
tesla.setId(i);
tesla.setName("Semi Truck 202" + i);
tesla.setFactory("弗拉蒙特特斯拉超级工厂");
updateList.add(tesla);
}
List<Tesla> saveList = new ArrayList<>();
for (int i = 0; i < 2; i++) {
Tesla tesla = new Tesla();
tesla.setName("Semi Truck 202" + i);
tesla.setFactory("柏林特斯拉超级工厂");
saveList.add(tesla);
}
List<Tesla> saveOrUpdateList = new ArrayList<>();
saveOrUpdateList.addAll(updateList);
saveOrUpdateList.addAll(saveList);
boolean b = teslaService.saveOrUpdateBatch(saveOrUpdateList);
System.out.println("是否更新或者保存成功:" + b);
}
执行该测试方法
对于设置了id的对象会限制性SELECT语句再执行UPDATE语句,并且只会更新给出的字段,对于没有设置id的对象会执行INSERT语句
// 根据 ID 删除, 直接调用BaseMapper的deleteById()方法
default boolean removeById(Serializable id)
// 根据 columnMap 条件,删除记录,直接调用BaseMapper的deleteByMap()方法
default boolean removeByMap(Map<String, Object> columnMap)
// 根据 entity 条件,删除记录,直接调用BaseMapper的remove()方法
default boolean remove(Wrapper<T> queryWrapper)
// 删除(根据ID 批量删除)
default boolean removeByIds(Collection<?> list)
// 批量删除(jdbc批量提交)
default boolean removeBatchByIds(Collection<?> list)
// 批量删除(jdbc批量提交)
default boolean removeBatchByIds(Collection<?> list, int batchSize)
在ITeslaServiceTest测试类中增加remove的测试方法
@Test
public void removeByMap(){
Map<String, Object> map = new HashMap<>();
map.put("id", 1166057551);
map.put("name", "Cyber Truck 2029");
boolean b = teslaService.removeByMap(map);
System.out.println("是否删除成功:" + b);
}
根据Map组成删除的Where子句执行删除操作
@Test
public void removeByIds(){
List<Integer> idList = new ArrayList<>();
idList.add(1166057563);
idList.add(1166057562);
idList.add(1166057561);
boolean b = teslaService.removeByIds(idList);
System.out.println("是否批量删除成功:" + b);
}
批量删除成功
@Test
public void removeBatchByIds(){
List<Integer> idList = new ArrayList<>();
idList.add(1166057560);
idList.add(1166057559);
idList.add(1166057558);
boolean b = teslaService.removeBatchByIds(idList);
System.out.println("是否批量删除成功:" + b);
}
与removeByIds()方法所调用删除语句不同
///根据 ID 选择修改,直接调用BaseMapper的updateById()方法
default boolean updateById(T entity);
// 根据 whereEntity 条件,更新记录,直接调用BaseMapper的update()方法,传入实体类
default boolean update(T entity, Wrapper<T> updateWrapper);
// 根据ID 批量更新
default boolean updateBatchById(Collection<T> entityList);
// TableId 注解存在更新记录,否插入一条记录
boolean saveOrUpdate(T entity);
在ITeslaServiceTest测试类中增加update的测试方法
@Test
public void updateBatchById(){
List<Tesla> updateList = new ArrayList<>();
for (int i = 1166057556; i < 1166057560; i++) {
Tesla tesla = new Tesla();
tesla.setId(i);
tesla.setFactory("柏林特斯拉超级工厂");
updateList.add(tesla);
}
boolean b = teslaService.updateBatchById(updateList);
System.out.println("是否更新成功:" + b);
}
@Test
public void saveOrUpdate(){
Tesla tesla = new Tesla();
tesla.setName("Model S");
tesla.setFactory("得克萨斯州特斯拉超级工厂");
tesla.setPrice(880000.00);
boolean b = teslaService.saveOrUpdate(tesla);
System.out.println("保存或者更新成功:" + b);
}
MPG 根据模板生成 service 和 controller 代码, MPG的代码模板在 generator包下的templates文件夹下
默认提供的模板只能够生成XxxController类,并不包含任何方法,如果需要在XxxController类中生成增删改查方法就需要自定义controller模板。
使用 Spring 全家桶之 Spring Boot 2.6.4(四)- Data Access(Part D MyBatis Plus) 中的 spring-boot-mybatisplus项目,将controller代码模板拷贝至自己项目中的templates文件夹下,项目中使用的是freemarker模板引擎,所以拷贝ftl结尾的controller模板
在application.yml中增加freemarker模板引擎的配置
spring:
freemarker:
template-loader-path: classpath:/resources/templates
suffix: .ftl
package ${package.Controller};
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.web.bind.annotation.RequestMapping;
import javax.annotation.Resource;
import java.util.List;
import org.springframework.web.bind.annotation.RestController;
import ${package.Entity}.${entity};
/**
* @author ${author}
* @since ${date}
*/
@RestController
@RequestMapping("/${table.entityPath}")
public class ${table.controllerName} {
@Resource
private ${table.serviceName} ${table.entityPath}Service;
// 查询列表
@GetMapping
public List<${entity}> list(){
return ${table.entityPath}Service.list();
}
// 根据id查询
@GetMapping("/{id}")
public ${entity} findOne(@PathVariable("id") Integer id){
return ${table.entityPath}Service.getById(id);
}
// 更新或保存
@PostMapping
public Boolean save(@RequestBody ${entity} ${table.entityPath}){
return ${table.entityPath}Service.saveOrUpdate(${table.entityPath});
}
// 根据id删除
@DeleteMapping("/{id}")
public Boolean removeById(@PathVariable("id") Integer id){
return ${table.entityPath}Service.removeById(id);
}
// 查询总记录数
@GetMapping("/count")
public Long count(){
return ${table.entityPath}Service.count();
}
// 批量删除
@PostMapping("/delete")
public Boolean batchDelete(@RequestBody List<Integer> ids){
return ${table.entityPath}Service.removeBatchByIds(ids);
}
// 批量更新或者保存
@PostMapping("/batch")
public Boolean saveOrUpdateBatch(List<${entity}> ${table.entityPath}List){
return ${table.entityPath}Service.saveOrUpdateBatch(${table.entityPath}List);
}
// 分页查询
@GetMapping("/page")
public Page<${entity}> findPage(@RequestParam("pageNum") Integer pageNum,
@RequestParam("pageSize") Integer pageSize){
Page<${entity}> page = new Page<>(pageNum, pageSize);
QueryWrapper<${entity}> queryWrapper = new QueryWrapper<>();
userService.page(page, queryWrapper);
return page;
}
}
代码模板中
将mybatis-plus-mpg中的代码生成器GeneratorApp拷贝至spring-boot-mybatisplus的test目录下,运行代码生成器
@RestController
@RequestMapping("/user")
public class UserController {
@Resource
private IUserService userService;
// 查询列表
@GetMapping
public List<User> list(){
return userService.list();
}
// 根据id查询
@GetMapping("/{id}")
public User findOne(@PathVariable("id") Integer id){
return userService.getById(id);
}
// 更新或保存
@PostMapping
public Boolean save(@RequestBody User user){
return userService.saveOrUpdate(user);
}
// 根据id删除
@DeleteMapping("/{id}")
public Boolean removeById(@PathVariable("id") Integer id){
return userService.removeById(id);
}
// 查询总记录数
@GetMapping("/count")
public Long count(){
return userService.count();
}
// 批量删除
@PostMapping("/delete")
public Boolean batchDelete(@RequestBody List<Integer> ids){
return userService.removeBatchByIds(ids);
}
// 批量更新或者保存
@PostMapping("/batch")
public Boolean saveOrUpdateBatch(List<User> userList){
return userService.saveOrUpdateBatch(userList);
}
// 分页查询
@GetMapping("/page")
public Page<User> findPage(@RequestParam("pageNum") Integer pageNum,
@RequestParam("pageSize") Integer pageSize){
Page<User> page = new Page<>(pageNum, pageSize);
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
userService.page(page, queryWrapper);
return page;
}
}
启动SpringBoot项目,测试生成的代码
查询用户列表 /user
根据id查询用户 /user/1
更新或者保存用户 /user
查询用户总数 /user/count
根据id删除用户 /user/1
分页查询用户 /user/page?pageNum=2&pageSize=3
批量删除 /user/delete