Browse Source

修复创建数据集模型名称重复bug

dev-complex-computation
chenzhihang 1 year ago
parent
commit
bf0c565255
3 changed files with 15 additions and 9 deletions
  1. +3
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/constant/Constant.java
  2. +6
    -5
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java
  3. +6
    -4
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java

+ 3
- 0
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/constant/Constant.java View File

@@ -22,6 +22,9 @@ public class Constant {

public final static int Git_Category_Id = 39;

public final static String Topic_Dataset = "ci4s-dataset";
public final static String Topic_model = "ci4s-model";

public final static String Item_Public = "public";

public final static String Source_Auto_Export = "auto_export";


+ 6
- 5
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java View File

@@ -578,7 +578,7 @@ public class ModelsServiceImpl implements ModelsService {
// 拼接project
GitProjectVo gitProjectVo = new GitProjectVo();
gitProjectVo.setRepositoryName(repositoryName);
gitProjectVo.setName("model" + modelsVo.getName());
gitProjectVo.setName(Constant.Topic_model + "-" + modelsVo.getName());
gitProjectVo.setDescription(modelsVo.getDescription());
gitProjectVo.setPrivate(!modelsVo.getIsPublic());
gitProjectVo.setUserId(userId);
@@ -604,7 +604,7 @@ public class ModelsServiceImpl implements ModelsService {
// 创建分支
gitService.createBranch(token, (String) userInfo.get("login"), repositoryName, branchName, "master");
// 定义标签 标签1:ci4s-model 标签2:ModelTag 标签3:ModelType
gitService.createTopic(token, gitlinIid, "ci4s-model");
gitService.createTopic(token, gitlinIid, Constant.Topic_model);
if (StringUtils.isNotEmpty(modelsVo.getModelTag())) {
gitService.createTopic(token, gitlinIid, "modeltag-" + modelsVo.getModelTag());
}
@@ -943,7 +943,7 @@ public class ModelsServiceImpl implements ModelsService {
String req = httpUtils.sendGetWithToken(url, null, token);
Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req);
List<Map<String, Object>> projects = (List<Map<String, Object>>) stringObjectMap.get("projects");
List<ModelsVo> collect = convert(projects, "ci4s-model", modelTagName, modelTypeName);
List<ModelsVo> collect = convert(projects, Constant.Topic_model, modelTagName, modelTypeName);
List<ModelsVo> result = collect.stream().skip((pageRequest.getPageNumber()) * pageRequest.getPageSize()).limit(pageRequest.getPageSize()).
collect(Collectors.toList());
return new PageImpl<>(result, pageRequest, collect.size());
@@ -957,7 +957,6 @@ public class ModelsServiceImpl implements ModelsService {
String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();

Map<String, Object> userInfo = getUserInfo(ci4sUsername, gitLinkUsername, gitLinkPassword);
Integer userId = (Integer) userInfo.get("user_id");
String token = (String) userInfo.get("token");

//拼接查询url
@@ -974,7 +973,7 @@ public class ModelsServiceImpl implements ModelsService {
String req = httpUtils.sendGetWithToken(url, null, token);
Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req);
List<Map<String, Object>> projects = (List<Map<String, Object>>) stringObjectMap.get("projects");
List<ModelsVo> collect = convert(projects, "ci4s-model", modelTagName, modelTypeName);
List<ModelsVo> collect = convert(projects, Constant.Topic_model, modelTagName, modelTypeName);
List<ModelsVo> result = collect.stream().skip((pageRequest.getPageNumber()) * pageRequest.getPageSize()).limit(pageRequest.getPageSize()).
collect(Collectors.toList());
return new PageImpl<>(result, pageRequest, collect.size());
@@ -1219,6 +1218,8 @@ public class ModelsServiceImpl implements ModelsService {
Map<String, Object> author = (Map<String, Object>) map.get("author");
newModelVo.setCreateBy((String) author.get("name"));
newModelVo.setOwner((String) author.get("login"));
String name = (String) map.get("name");
newModelVo.setName(name.substring((modelTopic + "-").length()));
result.add(newModelVo);
}
}


+ 6
- 4
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java View File

@@ -94,7 +94,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
String repositoryName = ci4sUsername + "_dataset_" + DateUtils.dateTimeNow();
GitProjectVo gitProjectVo = new GitProjectVo();
gitProjectVo.setRepositoryName(repositoryName);
gitProjectVo.setName("dataset-" + datasetVo.getName());
gitProjectVo.setName(Constant.Topic_Dataset + "-" + datasetVo.getName());
gitProjectVo.setDescription(datasetVo.getDescription());
gitProjectVo.setPrivate(!datasetVo.getIsPublic());
gitProjectVo.setUserId(userId);
@@ -111,7 +111,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {

gitService.createBranch(token, (String) userInfo.get("login"), repositoryName, branchName, "master");
// 定义标签 标签1:ci4s-dataset 标签2:DataTag 标签3:DataType
gitService.createTopic(token, gitlinIid, "ci4s-dataset");
gitService.createTopic(token, gitlinIid, Constant.Topic_Dataset);
if (StringUtils.isNotEmpty(datasetVo.getDataTag())) {
gitService.createTopic(token, gitlinIid, "datatag-" + datasetVo.getDataTag());
}
@@ -320,7 +320,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req);
List<Map<String, Object>> projects = (List<Map<String, Object>>) stringObjectMap.get("projects");

List<NewDatasetVo> collect = convert(projects, "ci4s-dataset", dataset.getDataTag(), dataset.getDataType());
List<NewDatasetVo> collect = convert(projects, Constant.Topic_Dataset, dataset.getDataTag(), dataset.getDataType());
List<NewDatasetVo> result = collect.stream().skip((pageRequest.getPageNumber()) * pageRequest.getPageSize()).limit(pageRequest.getPageSize()).
collect(Collectors.toList());
return new PageImpl<>(result, pageRequest, collect.size());
@@ -348,7 +348,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
if (projects == null) {
return new PageImpl<>(new ArrayList<>(), pageRequest, 0);
}
List<NewDatasetVo> collect = convert(projects, "ci4s-dataset", dataset.getDataTag(), dataset.getDataType());
List<NewDatasetVo> collect = convert(projects, Constant.Topic_Dataset, dataset.getDataTag(), dataset.getDataType());
List<NewDatasetVo> result = collect.stream().skip((pageRequest.getPageNumber()) * pageRequest.getPageSize()).limit(pageRequest.getPageSize()).
collect(Collectors.toList());
return new PageImpl<>(result, pageRequest, collect.size());
@@ -545,6 +545,8 @@ public class NewDatasetServiceImpl implements NewDatasetService {
Map<String, Object> author = (Map<String, Object>) map.get("author");
newDatasetVo.setCreateBy((String) author.get("name"));
newDatasetVo.setOwner((String) author.get("login"));
String name = (String)map.get("name");
newDatasetVo.setName(name.substring((datasetTopic +"-").length()));
result.add(newDatasetVo);
}
}


Loading…
Cancel
Save