idea抽取独立方法快捷键:ctrl+alt+m
TaskDto.java
java
package com.dkd.manage.service.impl;
import java.time.Duration;
import java.util.List;
import java.util.stream.Collectors;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.dkd.common.constant.DkdContants;
import com.dkd.common.exception.ServiceException;
import com.dkd.common.utils.DateUtils;
import com.dkd.common.utils.bean.BeanUtils;
import com.dkd.manage.domain.Emp;
import com.dkd.manage.domain.TaskDetails;
import com.dkd.manage.domain.VendingMachine;
import com.dkd.manage.domain.dto.TaskDetailDto;
import com.dkd.manage.domain.dto.TaskDto;
import com.dkd.manage.domain.vo.TaskVo;
import com.dkd.manage.service.IEmpService;
import com.dkd.manage.service.ITaskDetailsService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import com.dkd.manage.mapper.TaskMapper;
import com.dkd.manage.domain.Task;
import com.dkd.manage.service.ITaskService;
/**
* 工单管理Service业务层处理
*
* @author zzq
* @date 2024-08-01
*/
@Service
public class TaskServiceImpl implements ITaskService
{
@Autowired
private TaskMapper taskMapper;
@Autowired
private VendingMachineServiceImpl vendingMachineService;
@Autowired
private IEmpService empService;
@Autowired
private RedisTemplate redisTemplate;
@Autowired
private ITaskDetailsService taskDetailsService;
/**
* 查询工单管理
*
* @param taskId 工单管理主键
* @return 工单管理
*/
@Override
public Task selectTaskByTaskId(Long taskId)
{
return taskMapper.selectTaskByTaskId(taskId);
}
/**
* 查询工单管理列表
*
* @param task 工单管理
* @return 工单管理
*/
@Override
public List<Task> selectTaskList(Task task)
{
return taskMapper.selectTaskList(task);
}
/**
* 新增工单管理
*
* @param task 工单管理
* @return 结果
*/
@Override
public int insertTask(Task task)
{
task.setCreateTime(DateUtils.getNowDate());
return taskMapper.insertTask(task);
}
/**
* 修改工单管理
*
* @param task 工单管理
* @return 结果
*/
@Override
public int updateTask(Task task)
{
task.setUpdateTime(DateUtils.getNowDate());
return taskMapper.updateTask(task);
}
/**
* 批量删除工单管理
*
* @param taskIds 需要删除的工单管理主键
* @return 结果
*/
@Override
public int deleteTaskByTaskIds(Long[] taskIds)
{
return taskMapper.deleteTaskByTaskIds(taskIds);
}
/**
* 删除工单管理信息
*
* @param taskId 工单管理主键
* @return 结果
*/
@Override
public int deleteTaskByTaskId(Long taskId)
{
return taskMapper.deleteTaskByTaskId(taskId);
}
/**
* 查询工单类表
* @param taskVo
* * @return TaskVo集合
*/
@Override
public List<TaskVo> selectTaskVoList(Task taskVo) {
return taskMapper.selectTaskVoList(taskVo);
}
/**
*新增运营、运维工单
* @param taskDto
* @return 结果
*/
@Override
public int insertTaskDto(TaskDto taskDto) {
//1、查询售货机是否存在
VendingMachine vm = vendingMachineService.selectVendingMachineByInnerCode(taskDto.getInnerCode());
if(vm==null){
throw new ServiceException("设备不存在");
}
//2、校验售货机状态与工单类型是否相符
checkCreateTask(vm.getVmStatus(), taskDto.getProductTypeId());
//3、检查设备是否有未完成的同类型工单
hasTask(taskDto);
//4、查询并校验员工是否存在
Emp emp = empService.selectEmpById(taskDto.getUserId());
if(emp==null){
throw new ServiceException("员工不存在");
}
//5、校验员工区域是否匹配
if(!emp.getRegionId().equals(vm.getRegionId())){
throw new ServiceException("员工区域不匹配");
}
//6、将Dto转为po并补充属性,保存工单
Task task= BeanUtil.copyProperties(taskDto,Task.class);//属性复制
task.setTaskStatus(DkdContants.TASK_STATUS_CREATE);//创建工单
task.setUserName(emp.getUserName());//执行人名称
task.setRegionId(vm.getRegionId());//所属区域id
task.setAddr(vm.getAddr());//地址
task.setCreateTime(DateUtils.getNowDate());//创建时间
task.setTaskCode(generateTaskCode());//工单编号
int taskResult = taskMapper.insertTask(task);
//7、判断是否为补货工单
if(taskDto.getProductTypeId().equals(DkdContants.TASK_TYPE_SUPPLY)){
//8、保存工单详情
List<TaskDetailDto>details=taskDto.getDetails();
if(CollUtil.isEmpty(details)){
throw new ServiceException("补货工单详情不能为空");
}
//将dto转为po补充属性
List<TaskDetails>taskDetailsList=details.stream().map(detail -> {
TaskDetails taskDetails=BeanUtil.copyProperties(detail,TaskDetails.class);
taskDetails.setTaskId(task.getTaskId());
return taskDetails;
}).collect(Collectors.toList());
//批量新增
taskDetailsService.insertTaskDetailsBatch(taskDetailsList);
}
return taskResult;
}
//生成并获取当天工单编号(唯一标识)
private String generateTaskCode(){
//获取当前日期并格式化为"yyyymmdd"
String dateStr=DateUtils.getDate().replaceAll("-", "");
//根据日期生成redis的键
String key="dkd.task.code."+dateStr;
//判断key是否存在
if(!redisTemplate.hasKey(key)){
//如果key不存在,设置初始值为1,并指定过期时间为1天
redisTemplate.opsForValue().set(key,1, Duration.ofDays(1));
//返回工单编号(日期+0001)
return dateStr+"0001";
}
//如果key存在,计数器+1(0001),确保字符串长度为4位
return StrUtil.padPre(redisTemplate.opsForValue().increment(key).toString(),4,"0");
}
//检查是否有未完成的同类型的工单
private int hasTask(TaskDto taskDto) {
//创建task条件对象,并设置编号和工单类型,以及工单状态为进行中
Task taskParam = new Task();
taskParam.setInnerCode(taskDto.getInnerCode());
taskParam.setProductTypeId(taskDto.getProductTypeId());
taskParam.setTaskStatus(DkdContants.TASK_STATUS_PROGRESS);
//调用taskMapper查询数据库查看是否有符合条件的工单列表
List<Task> taskList = taskMapper.selectTaskList(taskParam);
//如果存在未完成的同类型工单,抛出异常
if(taskList!=null&&taskList.size()>0){
throw new ServiceException("设备有未完成的工单,不能重复创建");
}
return 0;
}
// 校验售货机状态与工单类型是否相符
private void checkCreateTask(Long vmStatus,Long productTypeId){
//如果是投放工单,设备在运行中,抛出异常
if(productTypeId ==DkdContants.TASK_TYPE_DEPLOY&&vmStatus==DkdContants.VM_STATUS_RUNNING){
throw new ServiceException("设备不在运行中,无法进行投放");
}
//如果是投放工单,设备不在运行中,抛出异常
if(productTypeId ==DkdContants.TASK_TYPE_SUPPLY&&vmStatus!=DkdContants.VM_STATUS_RUNNING){
throw new ServiceException("设备不在运行中,无法进行补货");
}
//如果是补货工单,设备不在运行中,抛出异常
if(productTypeId ==DkdContants.TASK_TYPE_REPAIR&&vmStatus!=DkdContants.VM_STATUS_RUNNING){
throw new ServiceException("设备不在运行中,无法进行补货");
}
//如果的撤机工单,设备不在运行中,抛出异常
if(productTypeId ==DkdContants.TASK_TYPE_REVOKE&&vmStatus!=DkdContants.VM_STATUS_RUNNING){
throw new ServiceException("设备不在运行中,无法进行撤机");
}
}
}
TaskController.java
java
package com.dkd.manage.controller;
import java.util.List;
import javax.servlet.http.HttpServletResponse;
import com.dkd.manage.domain.dto.TaskDto;
import com.dkd.manage.domain.vo.TaskVo;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.dkd.common.annotation.Log;
import com.dkd.common.core.controller.BaseController;
import com.dkd.common.core.domain.AjaxResult;
import com.dkd.common.enums.BusinessType;
import com.dkd.manage.domain.Task;
import com.dkd.manage.service.ITaskService;
import com.dkd.common.utils.poi.ExcelUtil;
import com.dkd.common.core.page.TableDataInfo;
/**
* 工单管理Controller
*
* @author zzq
* @date 2024-08-01
*/
@RestController
@RequestMapping("/manage/task")
public class TaskController extends BaseController
{
@Autowired
private ITaskService taskService;
/**
* 查询工单管理列表
*/
@PreAuthorize("@ss.hasPermi('manage:task:list')")
@GetMapping("/list")
public TableDataInfo list(Task task)
{
startPage();
List<TaskVo> voList = taskService.selectTaskVoList(task);
return getDataTable(voList);
}
/**
* 导出工单管理列表
*/
@PreAuthorize("@ss.hasPermi('manage:task:export')")
@Log(title = "工单管理", businessType = BusinessType.EXPORT)
@PostMapping("/export")
public void export(HttpServletResponse response, Task task)
{
List<Task> list = taskService.selectTaskList(task);
ExcelUtil<Task> util = new ExcelUtil<Task>(Task.class);
util.exportExcel(response, list, "工单管理数据");
}
/**
* 获取工单管理详细信息
*/
@PreAuthorize("@ss.hasPermi('manage:task:query')")
@GetMapping(value = "/{taskId}")
public AjaxResult getInfo(@PathVariable("taskId") Long taskId)
{
return success(taskService.selectTaskByTaskId(taskId));
}
/**
* 新增工单管理
*/
@PreAuthorize("@ss.hasPermi('manage:task:add')")
@Log(title = "工单管理", businessType = BusinessType.INSERT)
@PostMapping
public AjaxResult add(@RequestBody TaskDto taskDto)
{
//设置指派人id(登录用户)
taskDto.setAssigneeId(getUserId());
return toAjax(taskService.insertTaskDto(taskDto));
}
/**
* 修改工单管理
*/
@PreAuthorize("@ss.hasPermi('manage:task:edit')")
@Log(title = "工单管理", businessType = BusinessType.UPDATE)
@PutMapping
public AjaxResult edit(@RequestBody Task task)
{
return toAjax(taskService.updateTask(task));
}
/**
* 删除工单管理
*/
@PreAuthorize("@ss.hasPermi('manage:task:remove')")
@Log(title = "工单管理", businessType = BusinessType.DELETE)
@DeleteMapping("/{taskIds}")
public AjaxResult remove(@PathVariable Long[] taskIds)
{
return toAjax(taskService.deleteTaskByTaskIds(taskIds));
}
}