From f390e2b7973575a96f5039f5692e8e4f2c67408d Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Wed, 23 Jul 2025 10:54:47 +0800
Subject: [PATCH] =?UTF-8?q?=E9=A6=96=E9=A1=B5=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../workspace/WorkspaceController.java        | 62 ++++++++++++++++++-
 1 file changed, 60 insertions(+), 2 deletions(-)

diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/workspace/WorkspaceController.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/workspace/WorkspaceController.java
index fe0d04a4..0e75d0ee 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/workspace/WorkspaceController.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/workspace/WorkspaceController.java
@@ -3,10 +3,17 @@ package com.ruoyi.platform.controller.workspace;
 import com.ruoyi.common.core.web.controller.BaseController;
 import com.ruoyi.common.core.web.domain.AjaxResult;
 import com.ruoyi.common.core.web.domain.GenericsAjaxResult;
-import com.ruoyi.platform.service.JupyterService;
-import com.ruoyi.platform.service.WorkspaceService;
+import com.ruoyi.platform.domain.CodeConfig;
+import com.ruoyi.platform.domain.Dataset;
+import com.ruoyi.platform.domain.Image;
+import com.ruoyi.platform.domain.Service;
+import com.ruoyi.platform.service.*;
+import com.ruoyi.platform.vo.ModelsVo;
+import com.ruoyi.system.api.constant.Constant;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageRequest;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
@@ -19,6 +26,16 @@ public class WorkspaceController extends BaseController {
 
     @Resource
     private WorkspaceService workspaceService;
+    @Resource
+    private NewDatasetService newDatasetService;
+    @Resource
+    private ModelsService modelsService;
+    @Resource
+    private CodeConfigService codeConfigService;
+    @Resource
+    private ServiceService serviceService;
+    @Resource
+    private ImageService imageService;
 
     @GetMapping("/overview")
     @ApiOperation("得到运行概览")
@@ -38,4 +55,45 @@ public class WorkspaceController extends BaseController {
     public GenericsAjaxResult<Map<String, Integer>> getAssetPublicCount() throws Exception {
         return genericsSuccess(this.workspaceService.getAssetPublicCount());
     }
+
+    @GetMapping("/getPublicDatasets")
+    @ApiOperation("前6个公开数据集")
+    public AjaxResult getPublicDatasets() throws Exception {
+        PageRequest pageRequest = PageRequest.of(0, 6);
+        Dataset dataset = new Dataset();
+        return AjaxResult.success(this.newDatasetService.newPubilcQueryByPage(dataset, pageRequest, null));
+    }
+
+    @GetMapping("/getPublicModels")
+    @ApiOperation("前6个公开模型")
+    public AjaxResult getPublicModels() throws Exception {
+        PageRequest pageRequest = PageRequest.of(0, 6);
+        ModelsVo modelsVo = new ModelsVo();
+        return AjaxResult.success(this.modelsService.newPubilcQueryByPage(modelsVo, pageRequest, null));
+    }
+
+    @GetMapping("/getPublicCodeConfigs")
+    @ApiOperation("前6个代码配置")
+    public GenericsAjaxResult<Page<CodeConfig>> getPublicCodeConfigs() {
+        PageRequest pageRequest = PageRequest.of(0, 6);
+        CodeConfig codeConfig = new CodeConfig();
+        return genericsSuccess(this.codeConfigService.queryByPage(codeConfig, pageRequest));
+    }
+
+    @GetMapping("/getPublicServices")
+    @ApiOperation("前6个服务")
+    public GenericsAjaxResult<Page<Service>> getPublicServices() throws Exception {
+        PageRequest pageRequest = PageRequest.of(0, 6);
+        Service service = new Service();
+        return genericsSuccess(serviceService.queryByPageService(service, pageRequest));
+    }
+
+    @GetMapping("/getPublicImages")
+    @ApiOperation("前6个镜像")
+    public GenericsAjaxResult<Page<Image>> getPublicImages() throws Exception {
+        PageRequest pageRequest = PageRequest.of(0, 6);
+        Image image = new Image();
+        image.setImageType(Constant.Image_Type_Pub);
+        return genericsSuccess(this.imageService.queryByPage(image, pageRequest));
+    }
 }