写死过滤菜单权限

haoban-manage3-service/src/main/java/com/gic/haoban/manage/service/service/out/impl/role/HaobanRoleApiServiceImpl.java

@@ -27,6 +27,7 @@ import com.gic.haoban.manage.service.service.content.adaptor.EnterpriseAdaptor;
 import com.gic.haoban.manage.service.service.role.HaobanMenuService;
 import com.gic.haoban.manage.service.service.role.HaobanRoleMenuService;
 import com.gic.haoban.manage.service.service.role.HaobanRoleService;
+import com.gic.haoban.manage.service.util.ApolloUtils;
 import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.logging.log4j.LogManager;
@@ -287,10 +288,26 @@ public class HaobanRoleApiServiceImpl implements HaobanRoleApiService {
                 haobanMenuList.add(bo) ;
             }
         }
+        haobanMenuList = filterHaobanMenu(enterpriseId, haobanMenuList);
         retBO.setMenuList(EntityUtil.changeEntityListNew(HaobanMenuDTO.class, haobanMenuList));
         return ServiceResponse.success(EntityUtil.changeEntityNew(HaobanRoleDTO.class, retBO));
     }
 
+    private List<HaobanMenuBO> filterHaobanMenu(String enterpriseId,List<HaobanMenuBO> allMenuList) {
+        if (CollectionUtils.isEmpty(allMenuList)) {
+            return allMenuList;
+        }
+        boolean filterEid = ApolloUtils.filterEid(enterpriseId);
+        if (!filterEid) {
+            return allMenuList;
+        }
+        List<String> list = ApolloUtils.filterMenuCode();
+        if (CollectionUtils.isEmpty(list)) {
+            return allMenuList;
+        }
+        return allMenuList.stream().filter(menu -> !list.contains(menu.getMenuCode())).collect(Collectors.toList());
+    }
+
     private List<String> listRoleMenuCode(List<HaobanMenuBO> allMenuList , String wxEnterpriseId , Long  roleId) {
         List<String> menuCodes = haobanRoleMenuService.getByWxEnterpriseIdAndRoleId(wxEnterpriseId, roleId);
         // 添加关联的权限

haoban-manage3-service/src/main/java/com/gic/haoban/manage/service/util/ApolloUtils.java

@@ -7,7 +7,11 @@ import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 public class ApolloUtils {
 
@@ -103,4 +107,22 @@ public class ApolloUtils {
         String connectionMaxIdleMs = config.getProperty(HoloDataSource.HOLO_CONNECTIONMAXIDLEMS, "9000");
         return Integer.parseInt(connectionMaxIdleMs);
     }
+
+    public static boolean filterEid(String eid) {
+        Config config = ConfigService.getAppConfig();
+        String filterEid = config.getProperty("filterEid", "");
+        if (StringUtils.isBlank(filterEid)) {
+            return false;
+        }
+        return filterEid.contains(eid);
+    }
+
+    public static List<String> filterMenuCode() {
+        Config config = ConfigService.getAppConfig();
+        String filterMenuCode = config.getProperty("filterMenuCode", "");
+        if (StringUtils.isBlank(filterMenuCode)) {
+            return new ArrayList<>();
+        }
+        return Arrays.stream(filterMenuCode.split(",")).collect(Collectors.toList());
+    }
 }