完善删除中的逻辑
- 前言
- 效果图
- 逻辑介绍
- 表结构
- 根据mybatisPlus接口规范创建实体类和service和mapper文件
- 1、实体类Dish和Setmeal
- 2、Mapper接口DishMapper和SetealMapper
- 3、Service接口DishService和setmealService
- 4、Service实现类DishServiceImpl和setmealServicelmpl
- 编写删除函数
- 1、将创建刚才两个表的实体类
- 2、获取表中categoryId和id相等的数据
- 3、判断获取到的数据有几条
- 4、如果大于0说明绑定了数据就抛出异常
- 5、如果都不大于0就通过super调用通过id删除的方法
- 自定义业务异常类
- 在全局异常处理器中捕获异常
前言
本项目gitee位置:添加链接描述
之前实现了直接删除分类功能,并没有实现分类删除中的逻辑编写,本篇文章主要是针对删除逻辑来进行讲解,
本篇文章需要使用到Dish(菜品)和Setmeal(套餐)表,需要根据mybatisPlus规范创建实体类和接口类
效果图
逻辑介绍
检测分类删除的时候是否关联了别的菜品或者套餐
表结构
根据mybatisPlus接口规范创建实体类和service和mapper文件
1、实体类Dish和Setmeal
2、Mapper接口DishMapper和SetealMapper
3、Service接口DishService和setmealService
4、Service实现类DishServiceImpl和setmealServicelmpl
切记加@Service注解否则报错
编写删除函数
1、将创建刚才两个表的实体类
2、获取表中categoryId和id相等的数据
3、判断获取到的数据有几条
4、如果大于0说明绑定了数据就抛出异常
5、如果都不大于0就通过super调用通过id删除的方法
package com.example.ruiji_demo.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.example.ruiji_demo.common.CustomException;
import com.example.ruiji_demo.entity.Category;
import com.example.ruiji_demo.entity.Dish;
import com.example.ruiji_demo.entity.Setmeal;
import com.example.ruiji_demo.mapper.CategoryMapper;
import com.example.ruiji_demo.service.CategoryService;
import com.example.ruiji_demo.service.DishService;
import com.example.ruiji_demo.service.SetmealService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
/**
* @author jitwxs
* @date 2024年04月01日 21:44
*/
@Service
public class CategoryServiceImpl extends ServiceImpl<CategoryMapper, Category> implements CategoryService {
@Autowired
private DishService dishService;
@Autowired
private SetmealService setmealService;
/**
* 根据id删除分类,删除之前进行判断
* @param id
*/
@Override
public void remove(Long id){
//检测分类删除的时候是否关联了别的菜品或者套餐
// 创建菜品的实例
LambdaQueryWrapper<Dish> dishLambdaQueryWrapper = new LambdaQueryWrapper<>();
// 获取表中categoryId和id相等的数据
dishLambdaQueryWrapper.eq(Dish::getCategoryId,id);
// 判断获取到的数据有几条
Long count1 = dishService.count(dishLambdaQueryWrapper);
// 如果大于0说明关联了菜品
if(count1>0){
throw new CustomException("当前分类下关联了菜品,不能删除");
}
// 创建套餐的实例
LambdaQueryWrapper<Setmeal> setmealLambdaQueryWrapper = new LambdaQueryWrapper<>();
setmealLambdaQueryWrapper.eq(Setmeal::getCategoryId,id);
Long count2 = setmealService.count(setmealLambdaQueryWrapper);
if(count2>0){
throw new CustomException("当前分类下关联了套餐,不能删除");
}
super.removeById(id);
}
}
自定义业务异常类
package com.example.ruiji_demo.common;
/**
* 自定义业务异常类
* @author jitwxs
* @date 2024年04月03日 16:52
*/
public class CustomException extends RuntimeException {
public CustomException(String message){
super(message);
}
}
在全局异常处理器中捕获异常
package com.example.ruiji_demo.common;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import java.sql.SQLIntegrityConstraintViolationException;
/**
* @author jitwxs
* @date 2024年03月30日 22:24
*/
@ControllerAdvice(annotations = {RestController.class, Controller.class})
@ResponseBody
@Slf4j
public class GlobalExceptionHandler {
/**
* 异常处理方法
* @param ex
* @return
*/
@ExceptionHandler(SQLIntegrityConstraintViolationException.class)
public R<String> exceptionHandler(SQLIntegrityConstraintViolationException ex){
log.error(ex.getMessage());
// 获取到是否是含有Duplicate entry的错误,如果是就提取出来名称,然后添加到报错信息
if(ex.getMessage().contains("Duplicate entry")){
String[] split = ex.getMessage().split(" ");
String msg = split[2] + "已存在";
return R.error(msg);
}
return R.error("添加失败");
}
@ExceptionHandler(CustomException.class)
public R<String> exceptionHandler(CustomException ex){
return R.error(ex.getMessage());
}
}