test-version22
/
MeterSphere
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix(性能测试): 修改性能测试执行时检查是否含有jmx文件的校验方法

This commit is contained in:
song-tianyang 2023-08-02 15:23:17 +08:00 committed by 刘瑞斌
parent eac325046c
commit 0ba6e94d06
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
framework/sdk-parent/sdk/src/main/java/io/metersphere/metadata/service/FileMetadataService.java
View File

@ -250,6 +250,10 @@ public class FileMetadataService {
return null;
}
public FileMetadataWithBLOBs selectById(String id) {
return fileMetadataMapper.selectByPrimaryKey(id);
}
public byte[] loadFileAsBytes(String id) {
FileMetadataWithBLOBs fileMetadata = fileMetadataMapper.selectByPrimaryKey(id);
if (fileMetadata == null) {
@ -258,7 +262,7 @@ public class FileMetadataService {
return this.loadFileAsBytes(fileMetadata);
}
private byte[] loadFileAsBytes(FileMetadataWithBLOBs fileMetadata) {
public byte[] loadFileAsBytes(FileMetadataWithBLOBs fileMetadata) {
byte[] bytes = new byte[0];
// 兼容历史数据
if (StringUtils.isEmpty(fileMetadata.getStorage()) && StringUtils.isEmpty(fileMetadata.getResourceType())) {

10
performance-test/backend/src/main/java/io/metersphere/service/PerformanceTestService.java
View File

@ -387,12 +387,14 @@ public class PerformanceTestService {
loadTestFileExample.createCriteria().andTestIdEqualTo(id);
List<LoadTestFile> loadTestFileList = loadTestFileMapper.selectByExample(loadTestFileExample);
List<String> fileIdList = loadTestFileList.stream().map(LoadTestFile::getFileId).collect(Collectors.toList());
List<FileInfoDTO> fileInfoDTOList = fileMetadataService.downloadFileByIds(fileIdList);
boolean hasJmx = false;
for (FileInfoDTO fileInfoDTO : fileInfoDTOList) {
if (StringUtils.equalsIgnoreCase(fileInfoDTO.getType(), "jmx")) {
for (String fileId : fileIdList) {
FileMetadataWithBLOBs fileMetadata = fileMetadataService.selectById(fileId);
if (fileMetadata != null && StringUtils.equalsIgnoreCase(fileMetadata.getType(), "jmx")) {
hasJmx = true;
if (!JmxParseUtil.isJmxFile(fileInfoDTO.getFileByte())) {
byte[] bytes = fileMetadataService.loadFileAsBytes(fileMetadata);
if (bytes == null || bytes.length == 0) {
MSException.throwException(Translator.get("load_test_file_is_not_jmx"));
}
}