PermissionController.java
4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package com.uccc.admin.controller;
import com.alibaba.fastjson.JSONArray;
import com.uccc.admin.domain.Permission;
import com.uccc.admin.domain.Role;
import com.uccc.admin.exception.ApiException;
import com.uccc.admin.service.PermissionService;
import com.uccc.admin.service.RoleService;
import com.uccc.pretty.common.Result;
import com.uccc.pretty.common.User;
import com.uccc.pretty.constants.ErrorCode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import java.util.List;
import static com.uccc.pretty.constants.ResultEnum.RESULT_ERR;
import static com.uccc.pretty.constants.ResultEnum.RESULT_OK;
/**
* Created by bert on 2021-09-14 11:33
*/
@RestController
@RequestMapping("/permission/")
public class PermissionController {
private Logger logger = LoggerFactory.getLogger(PermissionController.class);
@Autowired
private PermissionService permissionService;
@Autowired
private RoleService roleService;
/**
* permission tree struct
* @return
* @throws NullPointerException
*/
@RequestMapping(value = "tree",method = RequestMethod.GET)
public Result permissionTree(@RequestParam int type) throws NullPointerException{
if (!(type == 1) && !(type ==2)) {
throw new ApiException(ErrorCode.INVALID_PARAM);
}
Result result = new Result();
JSONArray jsonArray;
//only admin can watch tree struct
List<Permission> permissions = permissionService.selectAllUserPermission(type);
if (permissions.size() == 0) {
throw new ApiException(ErrorCode.USER_PERMISSION_NOT_FOUND);
}else {
jsonArray = permissionService.formatPermissionList(permissions);
}
if (jsonArray.size() > 0) {
result.setCode(RESULT_OK.getCode());
result.setMessage(RESULT_OK.getMessage());
}else {
result.setCode(RESULT_ERR.getCode());
result.setMessage(RESULT_ERR.getMessage());
}
result.setData(jsonArray);
return result;
}
/**
* create role and config role_permission
* @param role
* @return
* @throws NullPointerException
*/
@RequestMapping(value = "role",method = RequestMethod.POST)
public Result createRole(@RequestBody Role role) throws NullPointerException{
Result result = new Result();
if (role == null) throw new ApiException(ErrorCode.INVALID_PARAM);
if (role.getPermission() == null) throw new ApiException(ErrorCode.INVALID_PARAM);
if (role.getEnName() == null) throw new ApiException(ErrorCode.INVALID_PARAM);
if (!(role.getPermission().length > 0)) throw new ApiException(ErrorCode.USER_PERMISSION_NOT_FOUND);
if (role.getName() == null) throw new ApiException(ErrorCode.INVALID_PARAM);
boolean repeat = roleService.getRoleNameRepeat(role.getName());
if (repeat) throw new ApiException(ErrorCode.ROLE_EXISTED);
int createRoleCount = roleService.createRole(role);
if (createRoleCount > 0) {
Role newRole = roleService.getRoleByName(role.getName());
//create permission
boolean addRolePermissionRes = permissionService.createRolePermission(role.getPermission(), newRole.getId());
if (addRolePermissionRes) {
result.setCode(RESULT_OK.getCode());
result.setMessage(RESULT_OK.getMessage());
}else {
throw new ApiException(ErrorCode.ADD_FAILED);
}
}else {
throw new ApiException(ErrorCode.ADD_FAILED);
}
return result;
}
/**
* create role and config role_permission
* @param user
* @return
* @throws NullPointerException
*/
@RequestMapping(value = "role",method = RequestMethod.PUT)
public Result updateRole(@RequestBody User user) throws NullPointerException{
Result result = new Result();
return result;
}
}