a-tuoheng-task/src/main/java/com/ruoyi/task/domain/convert/TaskConvert.java

81 lines
2.7 KiB
Java

package com.ruoyi.task.domain.convert;
import com.ruoyi.task.api.enums.ExecuteTypeEnum;
import com.ruoyi.task.api.enums.StatusEnum;
import com.ruoyi.task.domain.model.Task;
import com.ruoyi.task.mapper.entity.TaskInfoEntity;
/**
* 任务转换类
*
* @author ruoyi
* @date 2026-03-05
*/
public class TaskConvert {
/**
* 将实体转换为领域模型
*
* @param entity 任务实体
* @return 任务领域模型
*/
public static Task toModel(TaskInfoEntity entity) {
if (entity == null) {
return null;
}
Task model = new Task();
model.setId(entity.getId());
model.setPlanId(entity.getPlanId());
model.setTaskName(entity.getTaskName());
model.setTaskCategory(entity.getTaskCategory());
model.setTaskType(entity.getTaskType());
model.setExecuteType(ExecuteTypeEnum.getByCode(entity.getExecuteType()));
model.setRouteId(entity.getRouteId());
model.setUavId(entity.getUavId());
model.setStatus(StatusEnum.getByCode(entity.getStatus()));
model.setStartTime(entity.getStartTime());
model.setEndTime(entity.getEndTime());
model.setDescription(entity.getDescription());
model.setCreateBy(entity.getCreateBy());
model.setCreateTime(entity.getCreateTime());
model.setUpdateBy(entity.getUpdateBy());
model.setUpdateTime(entity.getUpdateTime());
model.setRemark(entity.getRemark());
return model;
}
/**
* 将领域模型转换为实体
*
* @param model 任务领域模型
* @return 任务实体
*/
public static TaskInfoEntity toEntity(Task model) {
if (model == null) {
return null;
}
TaskInfoEntity entity = new TaskInfoEntity();
entity.setId(model.getId());
entity.setPlanId(model.getPlanId());
entity.setTaskName(model.getTaskName());
entity.setTaskCategory(model.getTaskCategory());
entity.setTaskType(model.getTaskType());
entity.setExecuteType(model.getExecuteType() != null ? model.getExecuteType().getCode() : null);
entity.setRouteId(model.getRouteId());
entity.setUavId(model.getUavId());
entity.setStatus(model.getStatus() != null ? model.getStatus().getCode() : null);
entity.setStartTime(model.getStartTime());
entity.setEndTime(model.getEndTime());
entity.setDescription(model.getDescription());
entity.setCreateBy(model.getCreateBy());
entity.setCreateTime(model.getCreateTime());
entity.setUpdateBy(model.getUpdateBy());
entity.setUpdateTime(model.getUpdateTime());
entity.setRemark(model.getRemark());
return entity;
}
}