首页接口

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));
+    }
 }