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 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 Item_Public = "public";


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


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


//拼接查询url //拼接查询url
@@ -974,7 +973,7 @@ public class ModelsServiceImpl implements ModelsService {
String req = httpUtils.sendGetWithToken(url, null, token); String req = httpUtils.sendGetWithToken(url, null, token);
Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req); Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req);
List<Map<String, Object>> projects = (List<Map<String, Object>>) stringObjectMap.get("projects"); 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()). List<ModelsVo> result = collect.stream().skip((pageRequest.getPageNumber()) * pageRequest.getPageSize()).limit(pageRequest.getPageSize()).
collect(Collectors.toList()); collect(Collectors.toList());
return new PageImpl<>(result, pageRequest, collect.size()); 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"); Map<String, Object> author = (Map<String, Object>) map.get("author");
newModelVo.setCreateBy((String) author.get("name")); newModelVo.setCreateBy((String) author.get("name"));
newModelVo.setOwner((String) author.get("login")); newModelVo.setOwner((String) author.get("login"));
String name = (String) map.get("name");
newModelVo.setName(name.substring((modelTopic + "-").length()));
result.add(newModelVo); 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(); String repositoryName = ci4sUsername + "_dataset_" + DateUtils.dateTimeNow();
GitProjectVo gitProjectVo = new GitProjectVo(); GitProjectVo gitProjectVo = new GitProjectVo();
gitProjectVo.setRepositoryName(repositoryName); gitProjectVo.setRepositoryName(repositoryName);
gitProjectVo.setName("dataset-" + datasetVo.getName());
gitProjectVo.setName(Constant.Topic_Dataset + "-" + datasetVo.getName());
gitProjectVo.setDescription(datasetVo.getDescription()); gitProjectVo.setDescription(datasetVo.getDescription());
gitProjectVo.setPrivate(!datasetVo.getIsPublic()); gitProjectVo.setPrivate(!datasetVo.getIsPublic());
gitProjectVo.setUserId(userId); gitProjectVo.setUserId(userId);
@@ -111,7 +111,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {


gitService.createBranch(token, (String) userInfo.get("login"), repositoryName, branchName, "master"); gitService.createBranch(token, (String) userInfo.get("login"), repositoryName, branchName, "master");
// 定义标签 标签1:ci4s-dataset 标签2:DataTag 标签3:DataType // 定义标签 标签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())) { if (StringUtils.isNotEmpty(datasetVo.getDataTag())) {
gitService.createTopic(token, gitlinIid, "datatag-" + 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); Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req);
List<Map<String, Object>> projects = (List<Map<String, Object>>) stringObjectMap.get("projects"); 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()). List<NewDatasetVo> result = collect.stream().skip((pageRequest.getPageNumber()) * pageRequest.getPageSize()).limit(pageRequest.getPageSize()).
collect(Collectors.toList()); collect(Collectors.toList());
return new PageImpl<>(result, pageRequest, collect.size()); return new PageImpl<>(result, pageRequest, collect.size());
@@ -348,7 +348,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
if (projects == null) { if (projects == null) {
return new PageImpl<>(new ArrayList<>(), pageRequest, 0); 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()). List<NewDatasetVo> result = collect.stream().skip((pageRequest.getPageNumber()) * pageRequest.getPageSize()).limit(pageRequest.getPageSize()).
collect(Collectors.toList()); collect(Collectors.toList());
return new PageImpl<>(result, pageRequest, collect.size()); 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"); Map<String, Object> author = (Map<String, Object>) map.get("author");
newDatasetVo.setCreateBy((String) author.get("name")); newDatasetVo.setCreateBy((String) author.get("name"));
newDatasetVo.setOwner((String) author.get("login")); newDatasetVo.setOwner((String) author.get("login"));
String name = (String)map.get("name");
newDatasetVo.setName(name.substring((datasetTopic +"-").length()));
result.add(newDatasetVo); result.add(newDatasetVo);
} }
} }


Loading…
Cancel
Save