Browse Source

模型元数据添加创建时间,模型大小字段

dev-DXTZYK
chenzhihang 1 year ago
parent
commit
f29bb734a2
3 changed files with 30 additions and 0 deletions
  1. +8
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java
  2. +16
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java
  3. +6
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java

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

@@ -601,6 +601,7 @@ public class ModelsServiceImpl implements ModelsService {
BeanUtils.copyProperties(modelsVo, modelMetaVo); BeanUtils.copyProperties(modelsVo, modelMetaVo);
modelMetaVo.setId(gitlinIid); modelMetaVo.setId(gitlinIid);
modelMetaVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login"))); modelMetaVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login")));
modelMetaVo.setCreateTime(DateUtils.getTime());
modelMetaVo.setUpdateTime(DateUtils.getTime()); modelMetaVo.setUpdateTime(DateUtils.getTime());
modelMetaVo.setUsage("<pre><code>" + modelMetaVo.setUsage("<pre><code>" +
"# 克隆模型配置文件与存储参数到本地\n" + "# 克隆模型配置文件与存储参数到本地\n" +
@@ -612,6 +613,9 @@ public class ModelsServiceImpl implements ModelsService {
modelMetaVo.setOwner(owner); modelMetaVo.setOwner(owner);
modelMetaVo.setVersionDesc(modelMetaVo.getDescription()); modelMetaVo.setVersionDesc(modelMetaVo.getDescription());
modelMetaVo.setRelativePaths(relatePath); modelMetaVo.setRelativePaths(relatePath);
File folder = new File(modelPath);
long folderSize = FileUtil.getFolderSize(folder);
modelMetaVo.setModelSize(FileUtil.formatFileSize(folderSize));


Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo); Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo);
YamlUtils.generateYamlFile(metaMap, metaPath, "metadata"); YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");
@@ -695,6 +699,7 @@ public class ModelsServiceImpl implements ModelsService {
ModelMetaVo modelMetaVo = new ModelMetaVo(); ModelMetaVo modelMetaVo = new ModelMetaVo();
BeanUtils.copyProperties(modelsVo, modelMetaVo); BeanUtils.copyProperties(modelsVo, modelMetaVo);
modelMetaVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login"))); modelMetaVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login")));
modelMetaVo.setCreateTime(DateUtils.getTime());
modelMetaVo.setUpdateTime(DateUtils.getTime()); modelMetaVo.setUpdateTime(DateUtils.getTime());
modelMetaVo.setUsage("<pre><code>" + modelMetaVo.setUsage("<pre><code>" +
"# 克隆模型配置文件与存储参数到本地\n" + "# 克隆模型配置文件与存储参数到本地\n" +
@@ -708,6 +713,9 @@ public class ModelsServiceImpl implements ModelsService {
modelMetaVo.setModelTag(oldModelVo.getModelTag()); modelMetaVo.setModelTag(oldModelVo.getModelTag());
modelMetaVo.setModelType(oldModelVo.getModelType()); modelMetaVo.setModelType(oldModelVo.getModelType());
modelMetaVo.setRelativePaths(relatePath); modelMetaVo.setRelativePaths(relatePath);
File folder = new File(modelPath);
long folderSize = FileUtil.getFolderSize(folder);
modelMetaVo.setModelSize(FileUtil.formatFileSize(folderSize));


Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo); Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo);
YamlUtils.generateYamlFile(metaMap, metaPath, "metadata"); YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");


+ 16
- 0
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java View File

@@ -44,5 +44,21 @@ public class FileUtil {
} }




public static long getFolderSize(File folder) {
long size = 0;
File[] files = folder.listFiles(); // 获取文件夹下所有文件和子目录


if (files != null) {
for (File file : files) {
if (file.isFile()) {
// 如果是文件,则累加其大小
size += file.length();
} else if (file.isDirectory()) {
// 如果是子目录,则递归计算其大小
size += getFolderSize(file);
}
}
}
return size;
}
} }

+ 6
- 0
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java View File

@@ -29,9 +29,15 @@ public class ModelMetaVo implements Serializable {
@ApiModelProperty(value = "创建人") @ApiModelProperty(value = "创建人")
private String createBy; private String createBy;


@ApiModelProperty(value = "创建时间")
private String createTime;

@ApiModelProperty(value = "更新时间") @ApiModelProperty(value = "更新时间")
private String updateTime; private String updateTime;


@ApiModelProperty(value = "模型大小")
private String modelSize;

@ApiModelProperty(value = "训练镜像") @ApiModelProperty(value = "训练镜像")
private String image; private String image;




Loading…
Cancel
Save