首先在启动类似添加注解
@EnableTransactionManagement
然后在service的实现方法上添加注解
@Transactional(rollbackFor = CustomException.class)
这里使用自定义异常进行处理,抛自定义异常进行数据的回滚
package com.ruoyi.common.exception;
/**
* 自定义异常
*
* @author ruoyi
*/
public class CustomException extends RuntimeException {
private static final long serialVersionUID = 1L;
private Integer code;
private String message;
public CustomException(String message) {
this.message = message;
}
public CustomException(String message, Integer code) {
this.message = message;
this.code = code;
}
public CustomException(String message, Throwable e) {
super(message, e);
this.message = message;
}
@Override
public String getMessage() {
return message;
}
public Integer getCode() {
return code;
}
}
具体的service方法
/**
* 添加流程
* @param flowVo
* @return
*/
@Override
@Transactional(rollbackFor = CustomException.class)
public AjaxResult saveFlow(FlowVo flowVo) {
// 流程表中插入数据
Flow flow = new Flow();
if (!StringUtils.isNull(flowVo)) {
// 流程名字唯一
if (!getFlowName(flowVo.getFlowName())) {
flow.setFlowName(flowVo.getFlowName());
// 判断流程次数
if (flowVo.getCyclicTimes() == null) {
flow.setCyclicTimes(1L);
} else {
if (flowVo.getCyclicTimes().toString().trim().length() == 0) {
flow.setCyclicTimes(1L);
} else {
flow.setCyclicTimes(flowVo.getCyclicTimes());
}
}
save(flow);
} else {
throw new CustomException("流程名已存在");
}
// 循环遍历任务列表
List<FlowTask> taskList = flowVo.getTaskList();
if (StringUtils.isNotEmpty(taskList)) {
for (int i = 0; i < taskList.size(); i++) {
FlowTask flowTask = taskList.get(i);
// 验证
if (StringUtils.isEmpty(flowTask.getTaskId())) {
throw new CustomException("任务id不能为空或空格");
} else {
// 判断任务id是否存在
if (!taskService.checkTaskId(flowTask.getTaskId())) {
throw new CustomException("任务id不存在");
}
}
if (StringUtils.isEmpty(flowTask.getTaskName())) {
throw new CustomException("任务名称不能为空或空格");
} else {
// 判断任务名称是否存在
if (!taskService.checkTaskName(flowTask.getTaskName())) {
throw new CustomException("任务名称不存在");
} else {
// 判断任务名称和任务id是否匹配
if (!taskService.checkTaskIdAndName(flowTask.getTaskId(), flowTask.getTaskName())) {
throw new CustomException("任务id和任务名称不匹配");
}
}
}
if (StringUtils.isEmpty(flowTask.getDeviceType())) {
throw new CustomException("设备类型不能为空或空格");
} else {
if (("1").equals(flowTask.getDeviceType()) || ("0").equals(flowTask.getDeviceType())) {
// 任务id和任务名称以及和类型都应该保持一致
if (!taskService.checkTaskIdAndNameAndType(flowTask.getTaskId(), flowTask.getTaskName(), flowTask.getDeviceType())) {
throw new CustomException("设备类型与任务名称不匹配");
}
} else {
throw new CustomException("设备类型错误");
}
}
if (flowTask.getParam1().length() > 50 || flowTask.getParam2().length() > 50) {
throw new CustomException("参数超长错误");
}
if (StringUtils.isNull(flowTask.getTaskStart())) {
throw new CustomException("启动次数为空");
}
if (flowTask.getTaskStart() < 1 || flowTask.getTaskStart() > 999999) {
throw new CustomException("启动次数错误");
}
if (StringUtils.isEmpty(flowTask.getIsCyclic())) {
throw new CustomException("循环不能为空或空格");
} else {
if (("1").equals(flowTask.getIsCyclic()) || ("0").equals(flowTask.getIsCyclic())) {
} else {
throw new CustomException("循环类型错误");
}
}
flowTask.setFlowId(flow.getFlowId());
flowTask.setTaskSort(i);
flowTaskService.save(flowTask);
}
}
}
return null;
}
在controller中进行调用即可,并且捕获自定义的异常,将信息给前端展示
@PostMapping("/add")
public AjaxResult add(@Validated @RequestBody FlowVo flowVo) {
try {
flowService.saveFlow(flowVo);
}catch (CustomException e){
return AjaxResult.error(e.getMessage());
}
return AjaxResult.success("添加成功");
}