Commit 493690d0 authored by 黎博's avatar 黎博

接口加上权限限制

parent 74be2f7d
package cn.qg.holmes.controller.auth; package cn.qg.holmes.controller.auth;
import cn.qg.holmes.common.JsonResult; import cn.qg.holmes.common.JsonResult;
import cn.qg.holmes.config.annotation.RequiresPermissions;
import cn.qg.holmes.entity.auth.Role; import cn.qg.holmes.entity.auth.Role;
import cn.qg.holmes.entity.auth.RolePermission; import cn.qg.holmes.entity.auth.RolePermission;
import cn.qg.holmes.entity.auth.RolePermissionVo; import cn.qg.holmes.entity.auth.RolePermissionVo;
...@@ -29,6 +30,7 @@ public class RolePermissionController { ...@@ -29,6 +30,7 @@ public class RolePermissionController {
* 新增角色权限 * 新增角色权限
* @return * @return
*/ */
@RequiresPermissions("role:permission:add")
@PostMapping("/add") @PostMapping("/add")
public JsonResult addRolePermission(@RequestBody RolePermissionVo rolePermissionVo) { public JsonResult addRolePermission(@RequestBody RolePermissionVo rolePermissionVo) {
String roleName = rolePermissionVo.getRoleName(); String roleName = rolePermissionVo.getRoleName();
...@@ -61,6 +63,7 @@ public class RolePermissionController { ...@@ -61,6 +63,7 @@ public class RolePermissionController {
* 编辑角色权限 * 编辑角色权限
* @return * @return
*/ */
@RequiresPermissions("role:permission:edit")
@PostMapping("/edit") @PostMapping("/edit")
public JsonResult editRolePermission(@RequestBody RolePermissionVo rolePermissionVo) { public JsonResult editRolePermission(@RequestBody RolePermissionVo rolePermissionVo) {
Integer roleId = rolePermissionVo.getRoleId(); Integer roleId = rolePermissionVo.getRoleId();
...@@ -114,6 +117,7 @@ public class RolePermissionController { ...@@ -114,6 +117,7 @@ public class RolePermissionController {
* @param roleId 角色id * @param roleId 角色id
* @return * @return
*/ */
@RequiresPermissions("role:permission:del")
@GetMapping("/del") @GetMapping("/del")
public JsonResult delRolePermission(Integer roleId) { public JsonResult delRolePermission(Integer roleId) {
try { try {
......
package cn.qg.holmes.controller.auth; package cn.qg.holmes.controller.auth;
import cn.qg.holmes.common.JsonResult; import cn.qg.holmes.common.JsonResult;
import cn.qg.holmes.config.annotation.RequiresPermissions;
import cn.qg.holmes.entity.auth.UserRole; import cn.qg.holmes.entity.auth.UserRole;
import cn.qg.holmes.service.auth.UserRoleService; import cn.qg.holmes.service.auth.UserRoleService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
...@@ -24,6 +25,7 @@ public class UserRoleController { ...@@ -24,6 +25,7 @@ public class UserRoleController {
* @param roleId 角色id * @param roleId 角色id
* @return * @return
*/ */
@RequiresPermissions("user:role:add")
@PostMapping("/add") @PostMapping("/add")
public JsonResult addUserRole(Integer userId, Integer roleId) { public JsonResult addUserRole(Integer userId, Integer roleId) {
QueryWrapper<UserRole> userRoleQueryWrapper = new QueryWrapper<>(); QueryWrapper<UserRole> userRoleQueryWrapper = new QueryWrapper<>();
...@@ -51,6 +53,7 @@ public class UserRoleController { ...@@ -51,6 +53,7 @@ public class UserRoleController {
* @param userRole 用户角色实体 * @param userRole 用户角色实体
* @return * @return
*/ */
@RequiresPermissions("user:role:edit")
@PostMapping("/edit") @PostMapping("/edit")
public JsonResult editUserRole(UserRole userRole) { public JsonResult editUserRole(UserRole userRole) {
return JsonResult.buildSuccessResult(userRoleService.updateById(userRole)); return JsonResult.buildSuccessResult(userRoleService.updateById(userRole));
......
package cn.qg.holmes.controller.auto; package cn.qg.holmes.controller.auto;
import cn.qg.holmes.common.JsonResult; import cn.qg.holmes.common.JsonResult;
import cn.qg.holmes.config.annotation.RequiresPermissions;
import cn.qg.holmes.entity.auto.Interface; import cn.qg.holmes.entity.auto.Interface;
import cn.qg.holmes.service.auto.AutoModuleService; import cn.qg.holmes.service.auto.AutoModuleService;
import cn.qg.holmes.service.auto.InterfaceService; import cn.qg.holmes.service.auto.InterfaceService;
...@@ -70,6 +71,7 @@ public class InterfaceController { ...@@ -70,6 +71,7 @@ public class InterfaceController {
* @param interfaceEntity 接口类实体 * @param interfaceEntity 接口类实体
* @return * @return
*/ */
@RequiresPermissions("auto:interface:add")
@PostMapping("/add") @PostMapping("/add")
public JsonResult addInterface(@RequestBody Interface interfaceEntity) { public JsonResult addInterface(@RequestBody Interface interfaceEntity) {
QueryWrapper<Interface> queryWrapper = new QueryWrapper<>(); QueryWrapper<Interface> queryWrapper = new QueryWrapper<>();
...@@ -90,6 +92,7 @@ public class InterfaceController { ...@@ -90,6 +92,7 @@ public class InterfaceController {
* @param interfaceEntity 接口实体类 * @param interfaceEntity 接口实体类
* @return * @return
*/ */
@RequiresPermissions("auto:interface:edit")
@PostMapping("/modify") @PostMapping("/modify")
public JsonResult modifyInterface(@RequestBody Interface interfaceEntity) { public JsonResult modifyInterface(@RequestBody Interface interfaceEntity) {
if (interfaceService.getById(interfaceEntity.getId()) == null) { if (interfaceService.getById(interfaceEntity.getId()) == null) {
...@@ -103,6 +106,7 @@ public class InterfaceController { ...@@ -103,6 +106,7 @@ public class InterfaceController {
* @param interfaceId 接口id * @param interfaceId 接口id
* @return * @return
*/ */
@RequiresPermissions("auto:interface:del")
@GetMapping("/delete") @GetMapping("/delete")
public JsonResult deleteInterface(@RequestParam Integer interfaceId) { public JsonResult deleteInterface(@RequestParam Integer interfaceId) {
if (interfaceService.getById(interfaceId) == null) { if (interfaceService.getById(interfaceId) == null) {
......
package cn.qg.holmes.controller.auto; package cn.qg.holmes.controller.auto;
import cn.qg.holmes.common.JsonResult; import cn.qg.holmes.common.JsonResult;
import cn.qg.holmes.config.annotation.RequiresPermissions;
import cn.qg.holmes.entity.auto.Scene; import cn.qg.holmes.entity.auto.Scene;
import cn.qg.holmes.service.auto.AutoProjectService; import cn.qg.holmes.service.auto.AutoProjectService;
import cn.qg.holmes.service.auto.SceneService; import cn.qg.holmes.service.auto.SceneService;
...@@ -52,6 +53,7 @@ public class SceneController { ...@@ -52,6 +53,7 @@ public class SceneController {
* @param scene 场景实体 * @param scene 场景实体
* @return * @return
*/ */
@RequiresPermissions("auto:scene:add")
@PostMapping("/add") @PostMapping("/add")
public JsonResult addScene(@RequestBody Scene scene) { public JsonResult addScene(@RequestBody Scene scene) {
if (scene.getProjectId() == null) { if (scene.getProjectId() == null) {
...@@ -71,6 +73,7 @@ public class SceneController { ...@@ -71,6 +73,7 @@ public class SceneController {
* @param scene 场景实体 * @param scene 场景实体
* @return * @return
*/ */
@RequiresPermissions("auto:scene:edit")
@PostMapping("/modify") @PostMapping("/modify")
public JsonResult modifyScene(@RequestBody Scene scene) { public JsonResult modifyScene(@RequestBody Scene scene) {
if (scene.getId() == null) { if (scene.getId() == null) {
...@@ -87,6 +90,7 @@ public class SceneController { ...@@ -87,6 +90,7 @@ public class SceneController {
* @param sceneId 场景id * @param sceneId 场景id
* @return * @return
*/ */
@RequiresPermissions("auto:scene:del")
@GetMapping("/delete") @GetMapping("/delete")
public JsonResult deleteScene(Integer sceneId) { public JsonResult deleteScene(Integer sceneId) {
if (sceneService.getById(sceneId) == null) { if (sceneService.getById(sceneId) == null) {
......
package cn.qg.holmes.controller.auto; package cn.qg.holmes.controller.auto;
import cn.qg.holmes.common.JsonResult; import cn.qg.holmes.common.JsonResult;
import cn.qg.holmes.config.annotation.RequiresPermissions;
import cn.qg.holmes.entity.auto.Interface; import cn.qg.holmes.entity.auto.Interface;
import cn.qg.holmes.entity.auto.Testcase; import cn.qg.holmes.entity.auto.Testcase;
import cn.qg.holmes.service.auto.InterfaceService; import cn.qg.holmes.service.auto.InterfaceService;
...@@ -71,6 +72,7 @@ public class TestcaseController { ...@@ -71,6 +72,7 @@ public class TestcaseController {
* @param testcase * @param testcase
* @return * @return
*/ */
@RequiresPermissions("auto:testcase:add")
@PostMapping("/add") @PostMapping("/add")
public JsonResult addTestcase(@RequestBody Testcase testcase) { public JsonResult addTestcase(@RequestBody Testcase testcase) {
if (testcase.getName() == null || testcase.getInterfaceId() == null || testcase.getModuleId() == null) { if (testcase.getName() == null || testcase.getInterfaceId() == null || testcase.getModuleId() == null) {
...@@ -85,6 +87,7 @@ public class TestcaseController { ...@@ -85,6 +87,7 @@ public class TestcaseController {
* @param testcase 单接口用例实体 * @param testcase 单接口用例实体
* @return * @return
*/ */
@RequiresPermissions("auto:testcase:edit")
@PostMapping("/modify") @PostMapping("/modify")
public JsonResult modifyTestcase(@RequestBody Testcase testcase) { public JsonResult modifyTestcase(@RequestBody Testcase testcase) {
if (testcase.getId() == null) { if (testcase.getId() == null) {
...@@ -99,6 +102,7 @@ public class TestcaseController { ...@@ -99,6 +102,7 @@ public class TestcaseController {
* @param testcaseId 用例id * @param testcaseId 用例id
* @return * @return
*/ */
@RequiresPermissions("auto:testcase:del")
@GetMapping("/delete") @GetMapping("/delete")
public JsonResult deleteTestcase(@RequestParam Integer testcaseId) { public JsonResult deleteTestcase(@RequestParam Integer testcaseId) {
if (testcaseService.getById(testcaseId) == null) { if (testcaseService.getById(testcaseId) == null) {
......
package cn.qg.holmes.controller.mock; package cn.qg.holmes.controller.mock;
import cn.qg.holmes.common.JsonResult; import cn.qg.holmes.common.JsonResult;
import cn.qg.holmes.config.annotation.RequiresPermissions;
import cn.qg.holmes.entity.mock.Mock; import cn.qg.holmes.entity.mock.Mock;
import cn.qg.holmes.entity.mock.MockModule; import cn.qg.holmes.entity.mock.MockModule;
import cn.qg.holmes.service.mock.MockModuleService; import cn.qg.holmes.service.mock.MockModuleService;
...@@ -33,6 +34,7 @@ public class MockController { ...@@ -33,6 +34,7 @@ public class MockController {
@RequestMapping(value = "/mock/**", method = { RequestMethod.GET, RequestMethod.POST}) @RequestMapping(value = "/mock/**", method = { RequestMethod.GET, RequestMethod.POST})
public void mockMethod() {} public void mockMethod() {}
@RequiresPermissions("mock:add")
@PostMapping("/add/mock") @PostMapping("/add/mock")
public JsonResult addMock(@RequestBody Mock mock) { public JsonResult addMock(@RequestBody Mock mock) {
return JsonResult.buildSuccessResult(mockService.save(mock)); return JsonResult.buildSuccessResult(mockService.save(mock));
...@@ -55,6 +57,7 @@ public class MockController { ...@@ -55,6 +57,7 @@ public class MockController {
return JsonResult.buildSuccessResult(map); return JsonResult.buildSuccessResult(map);
} }
@RequiresPermissions("mock:edit")
@PostMapping("/edit/mock") @PostMapping("/edit/mock")
public JsonResult modifyMock(@RequestBody Mock mock) { public JsonResult modifyMock(@RequestBody Mock mock) {
return JsonResult.buildSuccessResult(mockService.saveOrUpdate(mock)); return JsonResult.buildSuccessResult(mockService.saveOrUpdate(mock));
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment