lanyuanxiaoyao/leopard
1
0
Fork 0
Code Activity

fix: 启动时更新未完成的任务为失败

Browse Source
This commit is contained in:
兰缘小妖
2025-09-07 12:21:13 +08:00
parent 7f2aa07d18
commit 22b5456752
2 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
leopard-server/src/main/java/com/lanyuanxiaoyao/leopard/server/repository/TaskRepository.java
View File

@@ -2,6 +2,8 @@ package com.lanyuanxiaoyao.leopard.server.repository;
import com.lanyuanxiaoyao.leopard.server.entity.Task; import com.lanyuanxiaoyao.leopard.server.entity.Task;
import com.lanyuanxiaoyao.service.template.repository.SimpleRepository; import com.lanyuanxiaoyao.service.template.repository.SimpleRepository;
import org.springframework.data.jpa.repository.Modifying;
import org.springframework.data.jpa.repository.Query;
import org.springframework.stereotype.Repository; import org.springframework.stereotype.Repository;
/** /**
@@ -10,4 +12,7 @@ import org.springframework.stereotype.Repository;
*/ */
@Repository @Repository
public interface TaskRepository extends SimpleRepository<Task> { public interface TaskRepository extends SimpleRepository<Task> {
@Modifying
@Query("update Task task set task.status = com.lanyuanxiaoyao.leopard.server.entity.Task.Status.FAILURE where task.status = com.lanyuanxiaoyao.leopard.server.entity.Task.Status.RUNNING")
void updateAllRunningTaskToFailure();
} }

12
leopard-server/src/main/java/com/lanyuanxiaoyao/leopard/server/service/TaskService.java
View File

@@ -5,8 +5,11 @@ import com.lanyuanxiaoyao.leopard.server.repository.TaskRepository;
import com.lanyuanxiaoyao.leopard.server.service.task.TaskMonitorNodes; import com.lanyuanxiaoyao.leopard.server.service.task.TaskMonitorNodes;
import com.lanyuanxiaoyao.service.template.service.SimpleServiceSupport; import com.lanyuanxiaoyao.service.template.service.SimpleServiceSupport;
import com.yomahub.liteflow.core.FlowExecutor; import com.yomahub.liteflow.core.FlowExecutor;
import jakarta.transaction.Transactional;
import java.util.Map; import java.util.Map;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.event.EventListener;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
/** /**
@@ -16,15 +19,24 @@ import org.springframework.stereotype.Service;
@Slf4j @Slf4j
@Service @Service
public class TaskService extends SimpleServiceSupport<Task> { public class TaskService extends SimpleServiceSupport<Task> {
private final TaskRepository taskRepository;
private final TaskTemplateService taskTemplateService; private final TaskTemplateService taskTemplateService;
private final FlowExecutor flowExecutor; private final FlowExecutor flowExecutor;
public TaskService(TaskRepository repository, TaskTemplateService taskTemplateService, @SuppressWarnings("SpringJavaInjectionPointsAutowiringInspection") FlowExecutor flowExecutor) { public TaskService(TaskRepository repository, TaskTemplateService taskTemplateService, @SuppressWarnings("SpringJavaInjectionPointsAutowiringInspection") FlowExecutor flowExecutor) {
super(repository); super(repository);
this.taskRepository = repository;
this.taskTemplateService = taskTemplateService; this.taskTemplateService = taskTemplateService;
this.flowExecutor = flowExecutor; this.flowExecutor = flowExecutor;
} }
@Transactional(rollbackOn = Throwable.class)
@EventListener(ApplicationReadyEvent.class)
public void onApplicationReady() {
log.warn("更新所有未完成的任务状态为失败");
taskRepository.updateAllRunningTaskToFailure();
}
public void execute(Long templateId, Map<String, Object> params) { public void execute(Long templateId, Map<String, Object> params) {
var template = taskTemplateService.detail(templateId); var template = taskTemplateService.detail(templateId);
var context = new TaskMonitorNodes.TaskMonitorContext(template); var context = new TaskMonitorNodes.TaskMonitorContext(template);