清洗代码

src/main/java/com/huiju/estateagents/cleancode/CleanCode.java

@@ -2,12 +2,8 @@ package com.huiju.estateagents.cleancode;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.huiju.estateagents.base.ResponseBean;
-import com.huiju.estateagents.entity.SysMenu;
-import com.huiju.estateagents.entity.TaRole;
-import com.huiju.estateagents.entity.TaRoleMenu;
-import com.huiju.estateagents.service.ISysMenuService;
-import com.huiju.estateagents.service.ITaRoleMenuService;
-import com.huiju.estateagents.service.ITaRoleService;
+import com.huiju.estateagents.entity.*;
+import com.huiju.estateagents.service.*;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
@@ -30,8 +26,14 @@ public class CleanCode {
     @Autowired
     private ITaRoleMenuService taRoleMenuService;
 
+    @Autowired
+    private ITaRoleButtonService taRoleButtonService;
+
+    @Autowired
+    private ISysButtonInMenuService sysButtonInMenuService;
+
     @RequestMapping(value = "/clean/menurole", method = RequestMethod.GET)
-    public void helpActivityList() {
+    public void cleanMenu() {
         List<SysMenu> sysMenuList = sysMenuService.list();
 
         QueryWrapper<TaRole> roleQueryWrapper = new QueryWrapper<>();
@@ -47,4 +49,22 @@ public class CleanCode {
             });
         });
     }
+
+    @RequestMapping(value = "/clean/buttonrole", method = RequestMethod.GET)
+    public void cleanButton() {
+        List<SysButtonInMenu> buttlist = sysButtonInMenuService.list();
+
+        QueryWrapper<TaRole> roleQueryWrapper = new QueryWrapper<>();
+        roleQueryWrapper.eq("is_admin",1);
+        List<TaRole> roleList = taRoleService.list(roleQueryWrapper);
+
+        roleList.forEach(e -> {
+            buttlist.forEach(s -> {
+                TaRoleButton taRoleButton = new TaRoleButton();
+                taRoleButton.setBtnId(s.getBtnId());
+                taRoleButton.setRoleId(e.getRoleId());
+                taRoleButtonService.save(taRoleButton);
+            });
+        });
+    }
 }

src/main/java/com/huiju/estateagents/interceptor/AccessInterceptor.java

@@ -65,7 +65,8 @@ public class AccessInterceptor implements HandlerInterceptor {
             "/webjars",
             "/webjars/springfox-swagger-ui",
             "/webjars/springfox-swagger-ui/css/typography.css",
-            "/clean/menurole"
+            "/clean/menurole",
+            "/clean/buttonrole"
     };
 
     /*