diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java index e05e1086..281d9943 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java @@ -601,6 +601,7 @@ public class ModelsServiceImpl implements ModelsService { BeanUtils.copyProperties(modelsVo, modelMetaVo); modelMetaVo.setId(gitlinIid); 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.setUsage("
" +
                     "# 克隆模型配置文件与存储参数到本地\n" +
@@ -612,6 +613,9 @@ public class ModelsServiceImpl implements ModelsService {
             modelMetaVo.setOwner(owner);
             modelMetaVo.setVersionDesc(modelMetaVo.getDescription());
             modelMetaVo.setRelativePaths(relatePath);
+            File folder = new File(modelPath);
+            long folderSize = FileUtil.getFolderSize(folder);
+            modelMetaVo.setModelSize(FileUtil.formatFileSize(folderSize));
 
             Map metaMap = JsonUtils.objectToMap(modelMetaVo);
             YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");
@@ -695,6 +699,7 @@ public class ModelsServiceImpl implements ModelsService {
             ModelMetaVo modelMetaVo = new ModelMetaVo();
             BeanUtils.copyProperties(modelsVo, modelMetaVo);
             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.setUsage("
" +
                     "# 克隆模型配置文件与存储参数到本地\n" +
@@ -708,6 +713,9 @@ public class ModelsServiceImpl implements ModelsService {
             modelMetaVo.setModelTag(oldModelVo.getModelTag());
             modelMetaVo.setModelType(oldModelVo.getModelType());
             modelMetaVo.setRelativePaths(relatePath);
+            File folder = new File(modelPath);
+            long folderSize = FileUtil.getFolderSize(folder);
+            modelMetaVo.setModelSize(FileUtil.formatFileSize(folderSize));
 
             Map metaMap = JsonUtils.objectToMap(modelMetaVo);
             YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");
diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java
index c70b223a..58e7d84b 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java
@@ -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;
+    }
 }
diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java
index f39e79e2..589255f7 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java
@@ -29,9 +29,15 @@ public class ModelMetaVo implements Serializable {
     @ApiModelProperty(value = "创建人")
     private String createBy;
 
+    @ApiModelProperty(value = "创建时间")
+    private String createTime;
+
     @ApiModelProperty(value = "更新时间")
     private String updateTime;
 
+    @ApiModelProperty(value = "模型大小")
+    private String modelSize;
+
     @ApiModelProperty(value = "训练镜像")
     private String image;