1、完善超分功能

2、完善积分系统
3、新增任务列表
This commit is contained in:
2024-03-26 14:58:43 +08:00
parent 305324fe1a
commit d411b428f8
26 changed files with 807 additions and 328 deletions

View File

@@ -0,0 +1,18 @@
package com.ai.da.model.dto;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
import javax.validation.constraints.NotEmpty;
@EqualsAndHashCode(callSuper = true)
@Data
@ApiModel("按分类分页查询历史任务")
public class QueryTaskHistoryDTO extends QueryPageByTimeDTO {
@NotEmpty(message = "type cannot be empty")
@ApiModelProperty("可选类型 SR")
private String type;
}

View File

@@ -6,7 +6,6 @@ import lombok.Data;
import lombok.NoArgsConstructor;
import javax.validation.constraints.NotBlank;
import java.util.List;
@Data
@NoArgsConstructor
@@ -22,6 +21,5 @@ public class SuperResolutionDTO {
private Integer scale;
@ApiModelProperty("唯一id用于保持消息唯一性")
String uniqueId;
private String uniqueId;
}

View File

@@ -0,0 +1,46 @@
package com.ai.da.model.dto;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@NoArgsConstructor
@AllArgsConstructor
public class TaskDTO<T> {
private String taskId;
// 可选type : SR GENERATE
private String type;
// 输入的图片名
private String imageName;
private T inputParam;
private String outputImage;
// 任务状态,暂定状态:排队中、执行中、成功/失败
private String status;
// 当前任务的创建时间
private String createDate;
public TaskDTO(String taskId, String type, String imageName, T inputParam, String status, String createDate) {
this.taskId = taskId;
this.type = type;
this.imageName = imageName;
this.inputParam = inputParam;
this.status = status;
this.createDate = createDate;
}
public TaskDTO(String taskId, String type, T inputParam, String status, String createDate) {
this.taskId = taskId;
this.type = type;
this.inputParam = inputParam;
this.status = status;
this.createDate = createDate;
}
}

View File

@@ -0,0 +1,22 @@
package com.ai.da.model.vo;
import lombok.Data;
@Data
public class TaskVO {
// 图片名
private String imageName;
private String inputImage;
private String outputImage;
private String otherInput;
private String status;
private String taskId;
private String createDate;
}