1、完善超分功能
2、完善积分系统 3、新增任务列表
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
package com.ai.da.controller;
|
||||
|
||||
import com.ai.da.common.context.UserContext;
|
||||
import com.ai.da.common.response.PageBaseResponse;
|
||||
import com.ai.da.common.response.Response;
|
||||
import com.ai.da.mapper.primary.entity.CreditsDetail;
|
||||
@@ -25,14 +26,14 @@ public class CreditsController {
|
||||
|
||||
@ApiOperation("获取当前积分")
|
||||
@GetMapping("/getCredits")
|
||||
public Response<String> getCredits(){
|
||||
String credits = creditsService.getCredits();
|
||||
public Response<String> getCredits() {
|
||||
String credits = creditsService.getCredits(UserContext.getUserHolder().getId());
|
||||
return Response.success(credits);
|
||||
}
|
||||
|
||||
@ApiOperation("获取积分详细")
|
||||
@PostMapping("/getCreditsDetail")
|
||||
public Response<PageBaseResponse<CreditsDetail>> getCreditsDetail(@Valid @RequestBody QueryIncomeOrExpenditureDTO queryPageByTimeDTO){
|
||||
public Response<PageBaseResponse<CreditsDetail>> getCreditsDetail(@Valid @RequestBody QueryIncomeOrExpenditureDTO queryPageByTimeDTO) {
|
||||
PageBaseResponse<CreditsDetail> credits = creditsService.queryCreditsDetailsPage(queryPageByTimeDTO);
|
||||
return Response.success(credits);
|
||||
}
|
||||
|
||||
@@ -115,7 +115,7 @@ public class PythonController {
|
||||
|
||||
@ApiOperation(value = "超分辨率")
|
||||
@PostMapping("/prepareForSR")
|
||||
public Response<String> superResolution(@RequestBody SuperResolutionDTO superResolutionDTO){
|
||||
public Response<List<String>> superResolution(@RequestBody List<SuperResolutionDTO> superResolutionDTO) {
|
||||
return Response.success(superResolutionService.prepareForSR(superResolutionDTO));
|
||||
}
|
||||
|
||||
|
||||
40
src/main/java/com/ai/da/controller/TaskListController.java
Normal file
40
src/main/java/com/ai/da/controller/TaskListController.java
Normal file
@@ -0,0 +1,40 @@
|
||||
package com.ai.da.controller;
|
||||
|
||||
import com.ai.da.common.response.PageBaseResponse;
|
||||
import com.ai.da.common.response.Response;
|
||||
import com.ai.da.model.dto.QueryTaskHistoryDTO;
|
||||
import com.ai.da.model.dto.SuperResolutionDTO;
|
||||
import com.ai.da.model.dto.TaskDTO;
|
||||
import com.ai.da.model.vo.TaskVO;
|
||||
import com.ai.da.service.TaskListService;
|
||||
import io.swagger.annotations.Api;
|
||||
import io.swagger.annotations.ApiOperation;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import javax.validation.Valid;
|
||||
import java.util.List;
|
||||
|
||||
@Api(tags = "任务列表模块")
|
||||
@Slf4j
|
||||
@RestController
|
||||
@RequestMapping("/api/tasks")
|
||||
public class TaskListController {
|
||||
|
||||
@Resource
|
||||
private TaskListService taskListService;
|
||||
|
||||
@PostMapping("/getList")
|
||||
@ApiOperation("获取未执行完的任务")
|
||||
public Response<List<TaskDTO<SuperResolutionDTO>>> getTaskList(@Valid @RequestBody List<String> taskIdList) {
|
||||
return Response.success(taskListService.getExecTask(taskIdList));
|
||||
}
|
||||
|
||||
@PostMapping("/getAllTask")
|
||||
@ApiOperation("获取所有任务")
|
||||
public Response<PageBaseResponse<TaskVO>> getAllTask(@Valid @RequestBody QueryTaskHistoryDTO queryTaskHistoryDTO) {
|
||||
return Response.success(taskListService.getAllTask(queryTaskHistoryDTO));
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user