a-tuoheng-task/src/main/java/com/ruoyi/task/controller/TaskController.java

134 lines
4.1 KiB
Java

package com.ruoyi.task.controller;
import com.alibaba.fastjson.JSON;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.web.controller.BaseController;
import com.ruoyi.task.api.domain.TaskStatQueryVO;
import com.ruoyi.task.api.enums.TaskCategoryEnum;
import com.ruoyi.task.api.enums.TaskTypeEnum;
import com.ruoyi.task.service.dto.TaskQueryDTO;
import com.ruoyi.task.api.enums.StatusEnum;
import com.ruoyi.task.service.api.ITaskService;
import com.ruoyi.task.controller.convert.TaskControllerConvert;
import com.ruoyi.task.controller.convert.TaskStatControllerConvert;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import java.util.List;
/**
* 任务Controller
*
* @author ruoyi
* @date 2026-01-17
*/
@RestController
@RequestMapping("/task")
@Slf4j
public class TaskController extends BaseController
{
@Autowired
private ITaskService taskService;
/**
* 创建普通-立即任务
*/
@PostMapping
public R<Long> createTaskWithoutPlan(@RequestBody com.ruoyi.task.api.domain.TaskDTO taskDTO)
{
taskDTO.setTaskType(TaskTypeEnum.IMMEDIATELY);
taskDTO.setTaskCategory(TaskCategoryEnum.NORMAL);
log.info("createTaskWithoutPlan {}", JSON.toJSONString(taskDTO));
Long taskId = taskService.createTaskWithoutPlan(TaskControllerConvert.to(taskDTO));
return R.ok(taskId);
}
/**
* 获取任务详细信息
*/
@GetMapping(value = "/{taskId}")
public R<com.ruoyi.task.api.domain.TaskDTO> getTaskById(@PathVariable("taskId") Long taskId)
{
return R.ok(TaskControllerConvert.from(taskService.getTaskById(taskId)));
}
/**
* 复杂条件查询任务列表
*/
@PostMapping("/list")
public R<List<com.ruoyi.task.api.domain.TaskDTO>> list(@RequestBody TaskQueryDTO queryDTO)
{
return R.ok(TaskControllerConvert.fromList(taskService.getTaskList(queryDTO)));
}
/**
* 更新任务
*/
@PutMapping
public R<Boolean> updateTask(@RequestBody com.ruoyi.task.api.domain.TaskDTO taskDTO)
{
log.info("updateTask {}", JSON.toJSONString(taskDTO));
boolean result = taskService.updateTask(TaskControllerConvert.to(taskDTO));
return R.ok(result);
}
/**
* 删除任务
*/
@DeleteMapping("/{taskId}")
public R<Boolean> deleteTask(@PathVariable("taskId") Long taskId)
{
log.info("deleteTask {}", taskId);
boolean result = taskService.deleteTask(taskId);
return R.ok(result);
}
/**
* 根据无人机ID查询任务列表
*/
@GetMapping("/uav/{uavId}")
public R<List<com.ruoyi.task.api.domain.TaskDTO>> getTaskByUavId(@PathVariable("uavId") String uavId)
{
return R.ok(TaskControllerConvert.fromList(taskService.getTaskByUavId(uavId)));
}
/**
* 根据无人机ID获取最新的一条任务
*/
@GetMapping("/uav/current/{uavId}")
public R<com.ruoyi.task.api.domain.TaskDTO> getCurrentTaskByUavId(@PathVariable("uavId") String uavId)
{
return R.ok(TaskControllerConvert.from(taskService.getCurrentTaskByUavId(uavId)));
}
/**
* 修改执行状态
*/
@PutMapping("/status/{taskId}")
public R<Boolean> updateTaskStatus(@PathVariable("taskId") Long taskId, @RequestParam("status") StatusEnum status)
{
boolean result = taskService.updateTaskStatus(taskId, status);
return R.ok(result);
}
/**
* 按年统计任务
*/
@PostMapping("/stat/year")
public R<com.ruoyi.task.api.domain.TaskStatByYearDTO> getTaskStatByYear(@RequestBody TaskStatQueryVO queryDTO)
{
return R.ok(TaskStatControllerConvert.fromYear(taskService.getTaskStatByYear(TaskStatControllerConvert.toQuery(queryDTO))));
}
/**
* 按年月统计任务
*/
@PostMapping("/stat/month")
public R<com.ruoyi.task.api.domain.TaskStatByMonthDTO> getTaskStatByMonth(@RequestBody TaskStatQueryVO queryDTO)
{
return R.ok(TaskStatControllerConvert.fromMonth(taskService.getTaskStatByMonth(TaskStatControllerConvert.toQuery(queryDTO))));
}
}