|
@@ -11,8 +11,10 @@ import com.huiju.estateagents.center.taUser.mapper.TaUserMapper;
|
11
|
11
|
import com.huiju.estateagents.center.taUser.service.ITaUserAuthButtonService;
|
12
|
12
|
import com.huiju.estateagents.entity.SysButtonInMenu;
|
13
|
13
|
import com.huiju.estateagents.entity.SysMenu;
|
|
14
|
+import com.huiju.estateagents.entity.TaRole;
|
14
|
15
|
import com.huiju.estateagents.entity.TaRoleButton;
|
15
|
16
|
import com.huiju.estateagents.mapper.TaRoleButtonMapper;
|
|
17
|
+import com.huiju.estateagents.mapper.TaRoleMapper;
|
16
|
18
|
import org.springframework.beans.factory.annotation.Autowired;
|
17
|
19
|
import org.springframework.stereotype.Service;
|
18
|
20
|
|
|
@@ -34,6 +36,8 @@ public class TaUserAuthButtonServiceImpl extends ServiceImpl<TaUserAuthButtonMap
|
34
|
36
|
TaRoleButtonMapper taRoleButtonMapper;
|
35
|
37
|
@Autowired
|
36
|
38
|
private TaUserMapper taUserMapper;
|
|
39
|
+ @Autowired
|
|
40
|
+ private TaRoleMapper taRoleMapper;
|
37
|
41
|
|
38
|
42
|
@Override
|
39
|
43
|
public ResponseBean getButtonByUserId(Integer pageNum, Integer pageSize, String userId) {
|
|
@@ -44,19 +48,10 @@ public class TaUserAuthButtonServiceImpl extends ServiceImpl<TaUserAuthButtonMap
|
44
|
48
|
@Override
|
45
|
49
|
public ResponseBean getAllButton(Integer pageNum, Integer pageSize) {
|
46
|
50
|
List<SysButtonInMenu> records = taUserAuthButtonMapper.sysButtonListByRoleId();
|
47
|
|
-
|
48
|
|
- QueryWrapper<TaRoleButton> queryWrapper = new QueryWrapper();
|
49
|
|
- List<TaRoleButton> taRoleButtons = taRoleButtonMapper.selectList(queryWrapper);
|
50
|
|
-
|
51
|
|
- for (SysButtonInMenu sysButtonInMenu : records){
|
52
|
|
- ArrayList<Integer> roles = new ArrayList<>();
|
53
|
|
- for (TaRoleButton taRoleButton : taRoleButtons){
|
54
|
|
- if (sysButtonInMenu.getBtnId().equals(taRoleButton.getBtnId()) && !roles.contains(taRoleButton.getBtnId())){
|
55
|
|
- roles.add(taRoleButton.getBtnId());
|
56
|
|
- }
|
57
|
|
- }
|
58
|
|
- sysButtonInMenu.setRoles(roles);
|
59
|
|
- }
|
|
51
|
+ records.forEach(e -> {
|
|
52
|
+ List<TaRole> taRoles = taRoleMapper.getRolesByBtnId(e.getBtnId());
|
|
53
|
+ e.setRolesList(taRoles);
|
|
54
|
+ });
|
60
|
55
|
|
61
|
56
|
return ResponseBean.success(records);
|
62
|
57
|
}
|