dromara/dbswitch
1
0
Fork 0
mirror of https://gitee.com/dromara/dbswitch.git synced 2025-10-13 21:30:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

修复issue反馈的问题

Browse Source
This commit is contained in:
inrgihc
2024-12-05 21:40:07 +08:00
parent 7f0139adad
commit 34274ec1ab
3 changed files with 57 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
dbswitch-admin/src/main/java/com/gitee/dbswitch/admin/service/ScheduleService.java
View File

@@ -17,6 +17,7 @@ import com.gitee.dbswitch.admin.execution.ExecuteJobTaskRunnable;
import com.gitee.dbswitch.admin.type.JobStatusEnum;
import com.gitee.dbswitch.admin.type.ScheduleModeEnum;
import com.gitee.dbswitch.common.event.EventSubscriber;
import com.gitee.dbswitch.common.event.ExceptionHandler;
import com.gitee.dbswitch.common.event.ListenedEvent;
import com.gitee.dbswitch.common.event.TaskEventHub;
import com.gitee.dbswitch.common.util.UuidUtils;
@@ -44,7 +45,7 @@ import org.springframework.stereotype.Service;
@Slf4j
@Service
public class ScheduleService implements InitializingBean {
public class ScheduleService implements InitializingBean, ExceptionHandler {
/**
* @Bean是一个方法级别上的注解，Bean的ID为方法名字。
@@ -60,14 +61,20 @@ public class ScheduleService implements InitializingBean {
@Resource
private AssignmentJobDAO assignmentJobDAO;
private TaskEventHub taskEventBus = new TaskEventHub("manualRun", 5);
private TaskEventHub taskEventBus = new TaskEventHub("manual-run", 5, this);
private Map<String, ExecuteJobTaskRunnable> taskRunnableMap = new ConcurrentHashMap<>();
@Override
public void afterPropertiesSet() throws Exception {
taskEventBus.registerSubscriber(new EventSubscriber(this::manualRunTask));
}
@Override
public void handleException(ListenedEvent event, Throwable throwable) {
log.warn("Failed to handle event: {}", event, throwable);
}
private void manualRunTask(ListenedEvent event) {
event.checkArgs(Long.class, String.class);
Long taskId = (Long) event.getArgs()[0];