优化按钮接口

src/main/java/com/huiju/estateagents/center/taUser/controller/TaUserController.java

             List<SysMenu> allMenus = (List<SysMenu>) iTaUserAuthMenuService.getAllMenu(null, null).getData();
             List<SysButtonInMenu> allButtons = (List<SysButtonInMenu>) iTaUserAuthButtonService.getAllButton(null, null).getData();
 
-            allButtons.forEach(e -> {
-                List<TaRole> taRoles = taRoleMapper.getRolesByBtnId(e.getBtnId());
-                e.setRolesList(taRoles);
-            });
-
             userMap.put("taUser", taUser);
             userMap.put("menuList", allMenus);
             userMap.put("buttonList", allButtons);

src/main/java/com/huiju/estateagents/center/taUser/service/impl/TaUserAuthButtonServiceImpl.java

 import com.huiju.estateagents.center.taUser.service.ITaUserAuthButtonService;
 import com.huiju.estateagents.entity.SysButtonInMenu;
 import com.huiju.estateagents.entity.SysMenu;
+import com.huiju.estateagents.entity.TaRole;
 import com.huiju.estateagents.entity.TaRoleButton;
 import com.huiju.estateagents.mapper.TaRoleButtonMapper;
+import com.huiju.estateagents.mapper.TaRoleMapper;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
     TaRoleButtonMapper taRoleButtonMapper;
     @Autowired
     private TaUserMapper taUserMapper;
+    @Autowired
+    private TaRoleMapper taRoleMapper;
 
     @Override
     public ResponseBean getButtonByUserId(Integer pageNum, Integer pageSize, String userId) {
     @Override
     public ResponseBean getAllButton(Integer pageNum, Integer pageSize) {
         List<SysButtonInMenu> records = taUserAuthButtonMapper.sysButtonListByRoleId();
-
-        QueryWrapper<TaRoleButton> queryWrapper = new QueryWrapper();
-        List<TaRoleButton> taRoleButtons = taRoleButtonMapper.selectList(queryWrapper);
-
-        for (SysButtonInMenu sysButtonInMenu : records){
-            ArrayList<Integer> roles = new ArrayList<>();
-            for (TaRoleButton taRoleButton : taRoleButtons){
-                if (sysButtonInMenu.getBtnId().equals(taRoleButton.getBtnId()) && !roles.contains(taRoleButton.getBtnId())){
-                    roles.add(taRoleButton.getBtnId());
-                }
-            }
-            sysButtonInMenu.setRoles(roles);
-        }
+        records.forEach(e -> {
+            List<TaRole> taRoles = taRoleMapper.getRolesByBtnId(e.getBtnId());
+            e.setRolesList(taRoles);
+        });
 
         return ResponseBean.success(records);
     }