From b27e55f4de22d27588033ef369048ed2e23f93b5 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期一, 06 六月 2022 15:16:20 +0800
Subject: [PATCH] add 增加 Excel 模板导出 测试类

---
 ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestTreeController.java |   44 +++++++++++++++++++++++---------------------
 1 files changed, 23 insertions(+), 21 deletions(-)

diff --git a/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestTreeController.java b/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestTreeController.java
index c34c77c..64f36c9 100644
--- a/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestTreeController.java
+++ b/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestTreeController.java
@@ -1,9 +1,10 @@
 package com.ruoyi.demo.controller;
 
+import cn.dev33.satoken.annotation.SaCheckPermission;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.annotation.RepeatSubmit;
 import com.ruoyi.common.core.controller.BaseController;
-import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.validate.AddGroup;
 import com.ruoyi.common.core.validate.EditGroup;
 import com.ruoyi.common.core.validate.QueryGroup;
@@ -14,9 +15,8 @@
 import com.ruoyi.demo.service.ITestTreeService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import io.swagger.annotations.ApiParam;
 import lombok.RequiredArgsConstructor;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 
@@ -34,7 +34,7 @@
  */
 @Validated
 @Api(value = "娴嬭瘯鏍戣〃鎺у埗鍣�", tags = {"娴嬭瘯鏍戣〃绠$悊"})
-@RequiredArgsConstructor(onConstructor_ = @Autowired)
+@RequiredArgsConstructor
 @RestController
 @RequestMapping("/demo/tree")
 public class TestTreeController extends BaseController {
@@ -45,45 +45,46 @@
      * 鏌ヨ娴嬭瘯鏍戣〃鍒楄〃
      */
     @ApiOperation("鏌ヨ娴嬭瘯鏍戣〃鍒楄〃")
-    @PreAuthorize("@ss.hasPermi('demo:tree:list')")
+    @SaCheckPermission("demo:tree:list")
     @GetMapping("/list")
-    public AjaxResult<List<TestTreeVo>> list(@Validated(QueryGroup.class) TestTreeBo bo) {
+    public R<List<TestTreeVo>> list(@Validated(QueryGroup.class) TestTreeBo bo) {
         List<TestTreeVo> list = iTestTreeService.queryList(bo);
-        return AjaxResult.success(list);
+        return R.ok(list);
     }
 
     /**
      * 瀵煎嚭娴嬭瘯鏍戣〃鍒楄〃
      */
     @ApiOperation("瀵煎嚭娴嬭瘯鏍戣〃鍒楄〃")
-    @PreAuthorize("@ss.hasPermi('demo:tree:export')")
+    @SaCheckPermission("demo:tree:export")
     @Log(title = "娴嬭瘯鏍戣〃", businessType = BusinessType.EXPORT)
     @GetMapping("/export")
     public void export(@Validated TestTreeBo bo, HttpServletResponse response) {
         List<TestTreeVo> list = iTestTreeService.queryList(bo);
-		ExcelUtil.exportExcel(list, "娴嬭瘯鏍戣〃", TestTreeVo.class, response);
+        ExcelUtil.exportExcel(list, "娴嬭瘯鏍戣〃", TestTreeVo.class, response);
     }
 
     /**
      * 鑾峰彇娴嬭瘯鏍戣〃璇︾粏淇℃伅
      */
     @ApiOperation("鑾峰彇娴嬭瘯鏍戣〃璇︾粏淇℃伅")
-    @PreAuthorize("@ss.hasPermi('demo:tree:query')")
+    @SaCheckPermission("demo:tree:query")
     @GetMapping("/{id}")
-    public AjaxResult<TestTreeVo> getInfo(@NotNull(message = "涓婚敭涓嶈兘涓虹┖")
-                                                  @PathVariable("id") Long id) {
-        return AjaxResult.success(iTestTreeService.queryById(id));
+    public R<TestTreeVo> getInfo(@ApiParam("娴嬭瘯鏍慖D")
+                                          @NotNull(message = "涓婚敭涓嶈兘涓虹┖")
+                                          @PathVariable("id") Long id) {
+        return R.ok(iTestTreeService.queryById(id));
     }
 
     /**
      * 鏂板娴嬭瘯鏍戣〃
      */
     @ApiOperation("鏂板娴嬭瘯鏍戣〃")
-    @PreAuthorize("@ss.hasPermi('demo:tree:add')")
+    @SaCheckPermission("demo:tree:add")
     @Log(title = "娴嬭瘯鏍戣〃", businessType = BusinessType.INSERT)
     @RepeatSubmit
     @PostMapping()
-    public AjaxResult<Void> add(@Validated(AddGroup.class) @RequestBody TestTreeBo bo) {
+    public R<Void> add(@Validated(AddGroup.class) @RequestBody TestTreeBo bo) {
         return toAjax(iTestTreeService.insertByBo(bo) ? 1 : 0);
     }
 
@@ -91,11 +92,11 @@
      * 淇敼娴嬭瘯鏍戣〃
      */
     @ApiOperation("淇敼娴嬭瘯鏍戣〃")
-    @PreAuthorize("@ss.hasPermi('demo:tree:edit')")
+    @SaCheckPermission("demo:tree:edit")
     @Log(title = "娴嬭瘯鏍戣〃", businessType = BusinessType.UPDATE)
     @RepeatSubmit
     @PutMapping()
-    public AjaxResult<Void> edit(@Validated(EditGroup.class) @RequestBody TestTreeBo bo) {
+    public R<Void> edit(@Validated(EditGroup.class) @RequestBody TestTreeBo bo) {
         return toAjax(iTestTreeService.updateByBo(bo) ? 1 : 0);
     }
 
@@ -103,11 +104,12 @@
      * 鍒犻櫎娴嬭瘯鏍戣〃
      */
     @ApiOperation("鍒犻櫎娴嬭瘯鏍戣〃")
-    @PreAuthorize("@ss.hasPermi('demo:tree:remove')")
-    @Log(title = "娴嬭瘯鏍戣〃" , businessType = BusinessType.DELETE)
+    @SaCheckPermission("demo:tree:remove")
+    @Log(title = "娴嬭瘯鏍戣〃", businessType = BusinessType.DELETE)
     @DeleteMapping("/{ids}")
-    public AjaxResult<Void> remove(@NotEmpty(message = "涓婚敭涓嶈兘涓虹┖")
-                                       @PathVariable Long[] ids) {
+    public R<Void> remove(@ApiParam("娴嬭瘯鏍慖D涓�")
+                                   @NotEmpty(message = "涓婚敭涓嶈兘涓虹┖")
+                                   @PathVariable Long[] ids) {
         return toAjax(iTestTreeService.deleteWithValidByIds(Arrays.asList(ids), true) ? 1 : 0);
     }
 }

--
Gitblit v1.9.3