获取状态接口修改

This commit is contained in:
even 2025-05-22 17:45:11 +08:00
parent 3cf62585ec
commit 5857f215a2
6 changed files with 12 additions and 6 deletions

View File

@ -61,6 +61,7 @@ public abstract class BaseWorker implements Runnable{
taskRunContext.getLocalVariables().put(EngineRuntimeConstant.LOG_KEY,pipTaskLog); taskRunContext.getLocalVariables().put(EngineRuntimeConstant.LOG_KEY,pipTaskLog);
execute(taskRunContext); execute(taskRunContext);
} catch (Exception e) { } catch (Exception e) {
log.error("================worker执行报错",e);
taskRunContext.changeContextState(ContextStateEnum.BAD_ENDING); taskRunContext.changeContextState(ContextStateEnum.BAD_ENDING);
return; return;
} }

View File

@ -26,7 +26,7 @@ import java.util.concurrent.atomic.AtomicInteger;
*/ */
@Slf4j @Slf4j
//@Plugin(taskType = "CUSTOM_IMAGE_EXECUTION_COMMAND") //@Plugin(taskType = "CUSTOM_IMAGE_EXECUTION_COMMAND")
@Plugin(taskType = "test") //@Plugin(taskType = "test")
public class DIYImageExecuteCommandWorker extends SshWorker { public class DIYImageExecuteCommandWorker extends SshWorker {
@Override @Override

View File

@ -39,7 +39,7 @@ import java.util.*;
* @Description: * @Description:
*/ */
@Slf4j @Slf4j
@Plugin(taskType = "scaSbom") @Plugin(taskType = "test")
public class ScaSbomWorker extends HttpWorker{ public class ScaSbomWorker extends HttpWorker{
private static final int POLLING_INTERVAL = 5000; // 轮询间隔单位毫秒 private static final int POLLING_INTERVAL = 5000; // 轮询间隔单位毫秒
@ -98,6 +98,7 @@ public class ScaSbomWorker extends HttpWorker{
pipTaskLog.append("==================SCA-SBOM节点执行失败================="); pipTaskLog.append("==================SCA-SBOM节点执行失败=================");
pipTaskLog.append("SCA-SBOM节点执行失败失败请检查当前节点配置!"); pipTaskLog.append("SCA-SBOM节点执行失败失败请检查当前节点配置!");
pipTaskLog.append(e.getMessage()); pipTaskLog.append(e.getMessage());
log.error("==================SCA-SBOM节点执行失败=================",e);
throw new ServiceException(GlobalErrorCodeConstants.INTERNAL_SERVER_ERROR.getCode(),"SCA-SBOM节点执行失败"); throw new ServiceException(GlobalErrorCodeConstants.INTERNAL_SERVER_ERROR.getCode(),"SCA-SBOM节点执行失败");
} }
} }
@ -135,6 +136,7 @@ public class ScaSbomWorker extends HttpWorker{
pollTaskStatus(restTemplate, oldScaTaskId); pollTaskStatus(restTemplate, oldScaTaskId);
} else { } else {
pipTaskLog.append("==================SCA接口异常调用失败================="); pipTaskLog.append("==================SCA接口异常调用失败=================");
log.error(message);
throw new ServiceException(GlobalErrorCodeConstants.INTERNAL_SERVER_ERROR.getCode(),"SCA-SBOM节点执行失败"); throw new ServiceException(GlobalErrorCodeConstants.INTERNAL_SERVER_ERROR.getCode(),"SCA-SBOM节点执行失败");
} }
} }

View File

@ -5,6 +5,7 @@ import cd.casic.ci.common.pipeline.req.pipeline.PipelineQueryReq;
import cd.casic.ci.common.pipeline.req.pipeline.PipelineReq; import cd.casic.ci.common.pipeline.req.pipeline.PipelineReq;
import cd.casic.ci.common.pipeline.req.pipeline.PipelineUpdateReq; import cd.casic.ci.common.pipeline.req.pipeline.PipelineUpdateReq;
import cd.casic.ci.common.pipeline.resp.context.SingletonRunContextResp; import cd.casic.ci.common.pipeline.resp.context.SingletonRunContextResp;
import cd.casic.ci.common.pipeline.resp.context.TreeRunContextResp;
import cd.casic.ci.common.pipeline.resp.pipeline.PipelineFindResp; import cd.casic.ci.common.pipeline.resp.pipeline.PipelineFindResp;
import cd.casic.ci.common.pipeline.utils.PageResult; import cd.casic.ci.common.pipeline.utils.PageResult;
import cd.casic.ci.process.process.dataObject.pipeline.PipPipeline; import cd.casic.ci.process.process.dataObject.pipeline.PipPipeline;
@ -36,5 +37,5 @@ public interface PipelineService extends IService<PipPipeline> {
PipelineFindResp findPipelineById(@Valid PipelineQueryReq pipelineQueryReq); PipelineFindResp findPipelineById(@Valid PipelineQueryReq pipelineQueryReq);
Map<String, SingletonRunContextResp> getPipelineRunState(String pipelineId); TreeRunContextResp getPipelineRunState(String pipelineId);
} }

View File

@ -526,7 +526,7 @@ public class PipelineServiceImpl extends ServiceImpl<PipelineDao, PipPipeline> i
} }
@Override @Override
public Map<String, SingletonRunContextResp> getPipelineRunState(String pipelineId) { public TreeRunContextResp getPipelineRunState(String pipelineId) {
BaseRunContext context = contextManager.getContext(pipelineId); BaseRunContext context = contextManager.getContext(pipelineId);
if (context instanceof PipelineRunContext pipelineRunContext) { if (context instanceof PipelineRunContext pipelineRunContext) {
TreeRunContextResp pipeline = new TreeRunContextResp(); TreeRunContextResp pipeline = new TreeRunContextResp();
@ -554,8 +554,9 @@ public class PipelineServiceImpl extends ServiceImpl<PipelineDao, PipPipeline> i
taskStateMap.put(taskId,taskState); taskStateMap.put(taskId,taskState);
} }
} }
return pipeline;
} }
return new HashMap<>(); return new TreeRunContextResp();
} }
} }

View File

@ -5,6 +5,7 @@ import cd.casic.ci.common.pipeline.req.pipeline.PipelineQueryReq;
import cd.casic.ci.common.pipeline.req.pipeline.PipelineReq; import cd.casic.ci.common.pipeline.req.pipeline.PipelineReq;
import cd.casic.ci.common.pipeline.req.pipeline.PipelineUpdateReq; import cd.casic.ci.common.pipeline.req.pipeline.PipelineUpdateReq;
import cd.casic.ci.common.pipeline.resp.context.SingletonRunContextResp; import cd.casic.ci.common.pipeline.resp.context.SingletonRunContextResp;
import cd.casic.ci.common.pipeline.resp.context.TreeRunContextResp;
import cd.casic.ci.common.pipeline.resp.pipeline.PipelineFindResp; import cd.casic.ci.common.pipeline.resp.pipeline.PipelineFindResp;
import cd.casic.ci.common.pipeline.utils.PageResult; import cd.casic.ci.common.pipeline.utils.PageResult;
import cd.casic.ci.process.engine.executor.PipelineExecutor; import cd.casic.ci.process.engine.executor.PipelineExecutor;
@ -105,7 +106,7 @@ public class PipelineController {
return CommonResult.success(execute); return CommonResult.success(execute);
} }
@PostMapping("/getStageRunState/{pipelineId}") @PostMapping("/getStageRunState/{pipelineId}")
public CommonResult<Map<String, SingletonRunContextResp>> getPipelineRunState(@PathVariable String pipelineId){ public CommonResult<TreeRunContextResp> getPipelineRunState(@PathVariable String pipelineId){
return CommonResult.success(pipelineService.getPipelineRunState(pipelineId)); return CommonResult.success(pipelineService.getPipelineRunState(pipelineId));
} }
} }