mianbin/ops-pro
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SSE心跳+worker还有taskType定义

Browse Source
This commit is contained in:
even 2025-05-27 16:15:22 +08:00
parent 0734d01fb2
commit 43a950e816
3 changed files with 32 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/manager/impl/MemoryLogManager.java
View File

@ -16,10 +16,7 @@ import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.*;
/**
* 流水线运行时日志管理
@ -120,23 +117,26 @@ public class MemoryLogManager implements LoggerManager {
private void emitterInit(SseEmitter emitter,String taskId){
// 维持心跳
scheduler.scheduleWithFixedDelay(()-> {
ScheduledFuture<?> scheduledFuture = scheduler.scheduleWithFixedDelay(() -> {
try {
emitter.send("\n\n");
emitter.send(SseEmitter.event().comment("heartbeat"));
} catch (IOException e) {
log.error("", e);
}
}, 0, 30, TimeUnit.SECONDS);
emitter.onCompletion(()->{
taskIdSSEMap.remove(taskId);
scheduledFuture.cancel(true);
log.info("===================taskId:{}断开连接===============",taskId);
});
emitter.onError((e)->{
taskIdSSEMap.remove(taskId);
scheduledFuture.cancel(true);
log.error("===================错误，taskId:{}断开连接===============",taskId,e);
});
emitter.onTimeout(()->{
taskIdSSEMap.remove(taskId);
scheduledFuture.cancel(true);
log.error("===================超时，taskId:{}断开连接===============",taskId);
});
}

12
modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/worker/AFLWorker.java Normal file
View File

@ -0,0 +1,12 @@
package cd.casic.ci.process.engine.worker;
import cd.casic.ci.common.pipeline.annotation.Plugin;
import cd.casic.ci.process.engine.runContext.TaskRunContext;
@Plugin(taskType = "AFL")
public class AFLWorker extends SshWorker{
@Override
public void execute(TaskRunContext context) {
}
}

12
modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/worker/TestCaseGenerationWorker.java Normal file
View File

@ -0,0 +1,12 @@
package cd.casic.ci.process.engine.worker;
import cd.casic.ci.common.pipeline.annotation.Plugin;
import cd.casic.ci.process.engine.runContext.TaskRunContext;
@Plugin(taskType = "TEST_CASE_GENERATION")
public class TestCaseGenerationWorker extends SshWorker{
@Override
public void execute(TaskRunContext context) {
}
}