From f1d079f9c0b3d384090477ca54e0548373d62b57 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Thu, 16 Jan 2025 11:56:39 +0800
Subject: [PATCH] 配置修改

---
 src/main/java/com/genersoft/iot/vmp/vmanager/user/RoleController.java |   78 ++++++++++++++-------------------------
 1 files changed, 28 insertions(+), 50 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/vmanager/user/RoleController.java b/src/main/java/com/genersoft/iot/vmp/vmanager/user/RoleController.java
index 6a75e1d..769dca0 100644
--- a/src/main/java/com/genersoft/iot/vmp/vmanager/user/RoleController.java
+++ b/src/main/java/com/genersoft/iot/vmp/vmanager/user/RoleController.java
@@ -1,27 +1,24 @@
 package com.genersoft.iot.vmp.vmanager.user;
 
+import com.genersoft.iot.vmp.conf.exception.ControllerException;
 import com.genersoft.iot.vmp.conf.security.SecurityUtils;
 import com.genersoft.iot.vmp.service.IRoleService;
-import com.genersoft.iot.vmp.service.IUserService;
 import com.genersoft.iot.vmp.storager.dao.dto.Role;
-import com.genersoft.iot.vmp.storager.dao.dto.User;
+import com.genersoft.iot.vmp.utils.DateUtil;
+import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
 import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
-import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiImplicitParam;
-import io.swagger.annotations.ApiImplicitParams;
-import io.swagger.annotations.ApiOperation;
+
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.tags.Tag;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
-import org.springframework.security.authentication.AuthenticationManager;
-import org.springframework.util.DigestUtils;
-import org.springframework.util.StringUtils;
 import org.springframework.web.bind.annotation.*;
 
-import java.text.SimpleDateFormat;
 import java.util.List;
 
-@Api(tags = "角色管理")
+@Tag(name  = "角色管理")
 @CrossOrigin
 @RestController
 @RequestMapping("/api/role")
@@ -30,72 +27,53 @@
     @Autowired
     private IRoleService roleService;
 
-    private final SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-
-    @ApiOperation("添加角色")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "name", required = true, value = "角色名", dataTypeClass = String.class),
-            @ApiImplicitParam(name = "authority", required = true, value = "权限(自行定义内容,目前未使用)", dataTypeClass = String.class),
-    })
     @PostMapping("/add")
-    public ResponseEntity<WVPResult<Integer>> add(@RequestParam String name,
+    @Operation(summary = "添加角色")
+    @Parameter(name = "name", description = "角色名", required = true)
+    @Parameter(name = "authority", description = "权限(自行定义内容,目前未使用)", required = true)
+    public void add(@RequestParam String name,
                                                   @RequestParam(required = false) String authority){
-        WVPResult<Integer> result = new WVPResult<>();
         // 获取当前登录用户id
         int currenRoleId = SecurityUtils.getUserInfo().getRole().getId();
         if (currenRoleId != 1) {
             // 只用角色id为1才可以删除和添加用户
-            result.setCode(-1);
-            result.setMsg("用户无权限");
-            return new ResponseEntity<>(result, HttpStatus.FORBIDDEN);
+            throw new ControllerException(ErrorCode.ERROR403);
         }
 
         Role role = new Role();
         role.setName(name);
         role.setAuthority(authority);
-        role.setCreateTime(format.format(System.currentTimeMillis()));
-        role.setUpdateTime(format.format(System.currentTimeMillis()));
+        role.setCreateTime(DateUtil.getNow());
+        role.setUpdateTime(DateUtil.getNow());
 
         int addResult = roleService.add(role);
-
-        result.setCode(addResult > 0 ? 0 : -1);
-        result.setMsg(addResult > 0 ? "success" : "fail");
-        result.setData(addResult);
-        return new ResponseEntity<>(result, HttpStatus.OK);
+        if (addResult <= 0) {
+            throw new ControllerException(ErrorCode.ERROR100);
+        }
     }
 
-    @ApiOperation("删除角色")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "id", required = true, value = "用户Id", dataTypeClass = Integer.class),
-    })
     @DeleteMapping("/delete")
-    public ResponseEntity<WVPResult<String>> delete(@RequestParam Integer id){
+    @Operation(summary = "删除角色")
+    @Parameter(name = "id", description = "用户Id", required = true)
+    public void delete(@RequestParam Integer id){
         // 获取当前登录用户id
         int currenRoleId = SecurityUtils.getUserInfo().getRole().getId();
-        WVPResult<String> result = new WVPResult<>();
         if (currenRoleId != 1) {
             // 只用角色id为0才可以删除和添加用户
-            result.setCode(-1);
-            result.setMsg("用户无权限");
-            return new ResponseEntity<>(result, HttpStatus.FORBIDDEN);
+            throw new ControllerException(ErrorCode.ERROR403);
         }
         int deleteResult = roleService.delete(id);
 
-        result.setCode(deleteResult>0? 0 : -1);
-        result.setMsg(deleteResult>0? "success" : "fail");
-        return new ResponseEntity<>(result, HttpStatus.OK);
+        if (deleteResult <= 0) {
+            throw new ControllerException(ErrorCode.ERROR100);
+        }
     }
 
-    @ApiOperation("查询角色")
-    @ApiImplicitParams({})
     @GetMapping("/all")
-    public ResponseEntity<WVPResult<List<Role>>> all(){
+    @Operation(summary = "查询角色")
+    public List<Role> all(){
         // 获取当前登录用户id
         List<Role> allRoles = roleService.getAll();
-        WVPResult<List<Role>> result = new WVPResult<>();
-        result.setCode(0);
-        result.setMsg("success");
-        result.setData(allRoles);
-        return new ResponseEntity<>(result, HttpStatus.OK);
+        return roleService.getAll();
     }
 }

--
Gitblit v1.9.3