toolchaintechnologycenter/spdm-backend
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改：hpc定义接口实现提交

Browse Source
This commit is contained in:
yangyang01000846
2025-11-27 20:58:28 +08:00
parent 82d800cd12
commit 56aa5b14e5
12 changed files with 179 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
pbs/src/main/java/com/sdm/pbs/controller/TaskController.java
View File

@@ -1,7 +1,7 @@
package com.sdm.pbs.controller;
import com.sdm.common.common.SdmResponse;
import com.sdm.common.entity.req.pbs.SubmitHpcTaskReq;
import com.sdm.common.entity.req.pbs.SubmitHpcTaskRemoteReq;
import com.sdm.common.entity.req.pbs.hpc.*;
import com.sdm.common.entity.resp.pbs.hpc.*;
import com.sdm.common.entity.resp.pbs.hpc.listjobs.ListJobResp;
@@ -11,11 +11,14 @@ import com.sdm.common.entity.resp.pbs.hpc.nodelist.NodeListResp;
import com.sdm.common.feign.inter.pbs.ITaskFeignClient;
import com.sdm.common.utils.HpcCommandExcuteUtil;
import com.sdm.pbs.model.bo.HpcJobStatusInfo;
import com.sdm.pbs.model.bo.HpcResouceInfo;
import com.sdm.pbs.model.req.SubmitHpcTaskReq;
import com.sdm.pbs.service.HpcInstructionService;
import com.sdm.pbs.service.IPbsService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.http.ResponseEntity;
@@ -51,10 +54,24 @@ public class TaskController implements ITaskFeignClient {
return hpcCommandExcuteUtil.excuteCmd(command,"remote");
}
@GetMapping ("/queryHpcResource")
@Operation(summary = "HPC资源查询")
public SdmResponse<HpcResouceInfo> queryHpcResource() {
return pbsService.queryHpcResource();
}
@PostMapping("/submitHpcJob")
@Operation(summary = "作业提交")
public SdmResponse<String> submitHpcJob(@RequestBody SubmitHpcTaskReq req) {
return pbsService.submitHpcJob(req);
public SdmResponse<String> submitHpcJob(@RequestBody SubmitHpcTaskRemoteReq req) {
SubmitHpcTaskReq submitHpcTaskReq = new SubmitHpcTaskReq();
BeanUtils.copyProperties(req,submitHpcTaskReq);
return pbsService.submitHpcJob(submitHpcTaskReq);
}
@PostMapping("/stopHpcJob")
@Operation(summary = "作业停止")
public SdmResponse<Boolean> stopHpcJob(@RequestParam String jobId) {
return pbsService.stopHpcJob(jobId);
}
@PostMapping("/getJobStatus")

6
pbs/src/main/java/com/sdm/pbs/model/bo/HpcNodeInfo.java
View File

@@ -1,17 +1,19 @@
package com.sdm.pbs.model.bo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import java.util.ArrayList;
import java.util.List;
@Data
public class HpcNodeInfo {
@Schema(description = "计算节点名称")
public String nodeName;
@Schema(description = "计算节点状态 0：down 1：running 2：suspend")
public int nodeStatus;
@Schema(description = "计算节点状态 Online, Offline, Draining,Unreachable")
public String nodeStatus;
@Schema(description = "节点总核数")
public int totalCores;

2
pbs/src/main/java/com/sdm/pbs/model/bo/HpcResouceInfo.java
View File

@@ -1,10 +1,12 @@
package com.sdm.pbs.model.bo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import java.util.ArrayList;
import java.util.List;
@Data
public class HpcResouceInfo {
@Schema(description = "资源总核数")

50
pbs/src/main/java/com/sdm/pbs/model/req/SubmitHpcTaskReq.java Normal file
View File

@@ -0,0 +1,50 @@
package com.sdm.pbs.model.req;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import java.util.ArrayList;
import java.util.List;
@Data
public class SubmitHpcTaskReq {
@Schema(description = "计算任务名称")
public String jobName;
@Schema(description = "计算所需要核数")
public int coreNum;
@Schema(description = "计算软件")
public String software;
@Schema(description = "计算任务类型")
public String jobType;
@Schema(description = "计算任务是否独立存在 0：非独立任务 1：独立任务")
public int independence;
@Schema(description = "求解文件")
public List<String> inputFiles = new ArrayList<>();
@Schema(description = "计算主文件")
public String masterFile;
@Schema(description = "计算任务所属任务ID")
public String taskId;
@Schema(description = "计算任务所属任务名称")
public String taskName;
@Schema(description = "计算任务所属算力ID")
public String runId;
@Schema(description = "计算任务所属算力名称")
public String runName;
@Schema(description = "执行的命令")
public String command;
@Schema(description = "任务所属项目")
public String projectname;
}

2
pbs/src/main/java/com/sdm/pbs/service/IPbsService.java
View File

@@ -1,10 +1,10 @@
package com.sdm.pbs.service;
import com.sdm.common.common.SdmResponse;
import com.sdm.common.entity.req.pbs.SubmitHpcTaskReq;
import com.sdm.pbs.model.bo.FileBaseInfo;
import com.sdm.pbs.model.bo.HpcJobStatusInfo;
import com.sdm.pbs.model.bo.HpcResouceInfo;
import com.sdm.pbs.model.req.SubmitHpcTaskReq;
import java.util.List;

95
pbs/src/main/java/com/sdm/pbs/service/impl/IPbsHpcServiceImpl.java
View File

@@ -2,17 +2,25 @@ package com.sdm.pbs.service.impl;
import com.sdm.common.common.SdmResponse;
import com.sdm.common.entity.constants.HpcConstants;
import com.sdm.common.entity.req.pbs.SubmitHpcTaskReq;
import com.sdm.common.entity.constants.NumberConstants;
import com.sdm.common.entity.req.pbs.hpc.*;
import com.sdm.common.entity.resp.pbs.hpc.JobCancelResp;
import com.sdm.common.entity.resp.pbs.hpc.SubmitHpcJobResp;
import com.sdm.common.entity.resp.pbs.hpc.listjobs.ListJobResp;
import com.sdm.common.entity.resp.pbs.hpc.listjobs.ListJobs;
import com.sdm.common.entity.resp.pbs.hpc.listtasks.ListTasks;
import com.sdm.common.entity.resp.pbs.hpc.listtasks.ListTasksResp;
import com.sdm.common.entity.resp.pbs.hpc.nodelist.NodeList;
import com.sdm.common.entity.resp.pbs.hpc.nodelist.NodeListResp;
import com.sdm.common.log.CoreLogger;
import com.sdm.pbs.model.bo.FileBaseInfo;
import com.sdm.pbs.model.bo.HpcJobStatusInfo;
import com.sdm.pbs.model.bo.HpcNodeInfo;
import com.sdm.pbs.model.bo.HpcResouceInfo;
import com.sdm.pbs.model.req.SubmitHpcTaskReq;
import com.sdm.pbs.service.HpcInstructionService;
import com.sdm.pbs.service.IPbsService;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
@@ -21,6 +29,8 @@ import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.stereotype.Service;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
@@ -35,11 +45,15 @@ public class IPbsHpcServiceImpl implements IPbsService {
@Override
public SdmResponse<HpcResouceInfo> queryHpcResource() {
ListJobReq listJobReq = new ListJobReq();
listJobReq.setFormat(HpcConstants.FORMAT_LIST);
SdmResponse<ListJobResp> hpcResp = hpcInstructionService.jobList(listJobReq);
// todo
return null;
NodeListReq nodeReq = new NodeListReq();
nodeReq.setFormat(HpcConstants.FORMAT_LIST);
SdmResponse<NodeListResp> hpcResp = hpcInstructionService.nodeList(nodeReq);
if(!hpcResp.isSuccess()){
return SdmResponse.failed("HPC节点信息查询失败");
}
HpcResouceInfo resourceInfo =
buildHpcResourceInfo(hpcResp.getData().getNodes());
return SdmResponse.success(resourceInfo);
}
@Override
@@ -71,7 +85,14 @@ public class IPbsHpcServiceImpl implements IPbsService {
@Override
public SdmResponse<Boolean> stopHpcJob(String jobId) {
return null;
CancelJobReq req = new CancelJobReq();
req.setJobId(jobId);
req.setCancelWay(HpcConstants.CANCEL_JOB_FORCE);
SdmResponse<JobCancelResp> stopResp = hpcInstructionService.jobCancel(req);
if(!stopResp.isSuccess()){
return SdmResponse.failed(false);
}
return SdmResponse.success(stopResp.getData().getCanceled());
}
@Override
@@ -115,11 +136,61 @@ public class IPbsHpcServiceImpl implements IPbsService {
return null;
}
// @Value("${file.rootPath}")
// private String rootPath;
//
// @Resource
// private RedisUtil redisUtil;
private HpcResouceInfo buildHpcResourceInfo(List<NodeList> nodes) {
HpcResouceInfo result = new HpcResouceInfo();
result.setNodeList(Collections.emptyList());
if (nodes == null || nodes.isEmpty()) {
result.setFreeCores(0);
result.setTotalCores(0);
result.setUsedCores(0);
return result;
}
int totalCores = 0;
int usedCores = 0;
int freeCoresSum = 0;
List<HpcNodeInfo> nodeInfos = new ArrayList<>(nodes.size());
for (NodeList node : nodes) {
if (node == null) {
continue;
}
int max = safeParseInt(node.getMax());
int run = safeParseInt(node.getRun());
int free = Math.max(max - run, 0);
HpcNodeInfo info = new HpcNodeInfo();
info.setNodeName(node.getNodeName());
info.setTotalCores(max);
info.setUsedCores(run);
info.setFreeCores(free);
info.setNodeStatus(node.getState());
if (HpcConstants.NODE_STATE_ONLINE.equals(node.getState())) {
totalCores += max;
usedCores += run;
freeCoresSum += free;
}
nodeInfos.add(info);
}
result.setTotalCores(totalCores);
result.setUsedCores(usedCores);
result.setFreeCores(freeCoresSum);
result.setNodeList(nodeInfos);
return result;
}
private int safeParseInt(String value) {
if (value == null || value.trim().isEmpty()) {
return NumberConstants.ZERO;
}
try {
return Integer.parseInt(value.trim());
} catch (Exception e) {
CoreLogger.error("safeParseInt error:{},value:{}", e.getMessage(), value);
return NumberConstants.ZERO;
}
}

6
pbs/src/main/java/com/sdm/pbs/service/impl/PbsServiceDecorator.java
View File

@@ -1,10 +1,10 @@
package com.sdm.pbs.service.impl;
import com.sdm.common.common.SdmResponse;
import com.sdm.common.entity.req.pbs.SubmitHpcTaskReq;
import com.sdm.pbs.model.bo.FileBaseInfo;
import com.sdm.pbs.model.bo.HpcJobStatusInfo;
import com.sdm.pbs.model.bo.HpcResouceInfo;
import com.sdm.pbs.model.req.SubmitHpcTaskReq;
import com.sdm.pbs.service.IPbsService;
import com.sdm.pbs.service.IPbsServiceDecorator;
import org.springframework.beans.factory.annotation.Autowired;
@@ -34,7 +34,7 @@ public class PbsServiceDecorator implements IPbsServiceDecorator {
@Override
public SdmResponse<HpcResouceInfo> queryHpcResource() {
return null;
return pbsService.queryHpcResource();
}
@Override
@@ -45,7 +45,7 @@ public class PbsServiceDecorator implements IPbsServiceDecorator {
@Override
public SdmResponse<Boolean> stopHpcJob(String jobId) {
return null;
return pbsService.stopHpcJob(jobId);
}
@Override