巡检模块

airport-admin/src/main/java/com/sundot/airport/web/controller/check/CheckRecordController.java

@@ -7,13 +7,12 @@ import javax.servlet.http.HttpServletResponse;
 
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.StrUtil;
-import com.sundot.airport.common.core.domain.entity.SysRole;
+import com.sundot.airport.common.core.domain.DataPermissionResult;
 import com.sundot.airport.common.core.domain.entity.SysUser;
 import com.sundot.airport.common.enums.CheckRecordStatusEnum;
-import com.sundot.airport.common.enums.RoleTypeEnum;
 import com.sundot.airport.common.enums.SourceTypeEnum;
-import com.sundot.airport.common.utils.SecurityUtils;
 import com.sundot.airport.system.service.ISysUserService;
+import com.sundot.airport.web.core.utils.DataPermissionUtils;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -54,40 +53,27 @@ public class CheckRecordController extends BaseController {
     @PreAuthorize("@ss.hasPermi('check:checkRecord:list')")
     @GetMapping("/list")
     public TableDataInfo list(CheckRecord checkRecord, @RequestHeader(value = "X-Request-Source", defaultValue = "mobile") String source) {
+        DataPermissionResult dataPermission = DataPermissionUtils.getDataPermission(getUserId(), getDeptId(), getLoginUser());
         List<CheckRecord> list;
-        List<SysRole> roles = SecurityUtils.getLoginUser().getUser().getRoles();
-        if (CollUtil.isEmpty(roles)) {
-            return getDataTable(Collections.emptyList());
-        }
-        List<String> roleKeyList = roles.stream().map(SysRole::getRoleKey).collect(Collectors.toList());
         if (StrUtil.equals(SourceTypeEnum.mobile.getCode(), source)) {
             startPage();
             list = checkRecordService.selectCheckRecordList(checkRecord);
         } else if (StrUtil.equals(SourceTypeEnum.web.getCode(), source)) {
-            if (roleKeyList.contains(RoleTypeEnum.admin.getCode())) {
-
-            } else if (roleKeyList.contains(RoleTypeEnum.test.getCode()) || roleKeyList.contains(RoleTypeEnum.zhijianke.getCode())) {
-
-            } else if (roleKeyList.contains(RoleTypeEnum.jingli.getCode()) || roleKeyList.contains(RoleTypeEnum.xingzheng.getCode())) {
-                Long deptId = SecurityUtils.getLoginUser().getDeptId();
-                List<SysUser> sysUserList = sysUserService.selectUserByDeptId(deptId);
-                if (CollUtil.isEmpty(sysUserList)) {
-                    return getDataTable(Collections.emptyList());
-                }
-                List<Long> userIdList = sysUserList.stream().map(SysUser::getUserId).distinct().collect(Collectors.toList());
-                checkRecord.setQueryCheckerIdList(userIdList);
-            } else if (roleKeyList.contains(RoleTypeEnum.kezhang.getCode())) {
-                Long deptId = SecurityUtils.getLoginUser().getDeptId();
-                List<SysUser> sysUserList = sysUserService.selectUserByDeptId(deptId);
-                if (CollUtil.isEmpty(sysUserList)) {
+            switch (dataPermission.getPermissionType()) {
+                case ALL:
+                case STATION:
+                    break;
+                case BRIGADE:
+                case DEPARTMENT:
+                    List<SysUser> sysUserList = sysUserService.selectUserByDeptId(dataPermission.getValue());
+                    if (CollUtil.isEmpty(sysUserList)) {
+                        return getDataTable(Collections.emptyList());
+                    }
+                    List<Long> userIdList = sysUserList.stream().map(SysUser::getUserId).distinct().collect(Collectors.toList());
+                    checkRecord.setQueryCheckerIdList(userIdList);
+                    break;
+                default:
                     return getDataTable(Collections.emptyList());
-                }
-                List<Long> userIdList = sysUserList.stream().map(SysUser::getUserId).distinct().collect(Collectors.toList());
-                checkRecord.setQueryCheckerIdList(userIdList);
-            } else if (roleKeyList.contains(RoleTypeEnum.banzuzhang.getCode())) {
-                return getDataTable(Collections.emptyList());
-            } else {
-                return getDataTable(Collections.emptyList());
             }
             startPage();
             list = checkRecordService.selectCheckRecordList(checkRecord);
@@ -104,35 +90,24 @@ public class CheckRecordController extends BaseController {
     @Log(title = "检查记录", businessType = BusinessType.EXPORT)
     @PostMapping("/export")
     public void export(HttpServletResponse response, CheckRecord checkRecord) {
+        DataPermissionResult dataPermission = DataPermissionUtils.getDataPermission(getUserId(), getDeptId(), getLoginUser());
         List<CheckRecord> list = Collections.emptyList();
-        List<SysRole> roles = SecurityUtils.getLoginUser().getUser().getRoles();
-        if (CollUtil.isNotEmpty(roles)) {
-            List<String> roleKeyList = roles.stream().map(SysRole::getRoleKey).collect(Collectors.toList());
-            if (roleKeyList.contains(RoleTypeEnum.admin.getCode())) {
-                list = checkRecordService.selectCheckRecordList(checkRecord);
-            } else if (roleKeyList.contains(RoleTypeEnum.test.getCode()) || roleKeyList.contains(RoleTypeEnum.zhijianke.getCode())) {
+        switch (dataPermission.getPermissionType()) {
+            case ALL:
+            case STATION:
                 list = checkRecordService.selectCheckRecordList(checkRecord);
-            } else if (roleKeyList.contains(RoleTypeEnum.jingli.getCode()) || roleKeyList.contains(RoleTypeEnum.xingzheng.getCode())) {
-                Long deptId = SecurityUtils.getLoginUser().getDeptId();
-                List<SysUser> sysUserList = sysUserService.selectUserByDeptId(deptId);
-                if (CollUtil.isNotEmpty(sysUserList)) {
-                    List<Long> userIdList = sysUserList.stream().map(SysUser::getUserId).distinct().collect(Collectors.toList());
-                    checkRecord.setQueryCheckerIdList(userIdList);
-                    list = checkRecordService.selectCheckRecordList(checkRecord);
-                }
-            } else if (roleKeyList.contains(RoleTypeEnum.kezhang.getCode())) {
-                Long deptId = SecurityUtils.getLoginUser().getDeptId();
-                List<SysUser> sysUserList = sysUserService.selectUserByDeptId(deptId);
+                break;
+            case BRIGADE:
+            case DEPARTMENT:
+                List<SysUser> sysUserList = sysUserService.selectUserByDeptId(dataPermission.getValue());
                 if (CollUtil.isNotEmpty(sysUserList)) {
                     List<Long> userIdList = sysUserList.stream().map(SysUser::getUserId).distinct().collect(Collectors.toList());
                     checkRecord.setQueryCheckerIdList(userIdList);
                     list = checkRecordService.selectCheckRecordList(checkRecord);
                 }
-            } else if (roleKeyList.contains(RoleTypeEnum.banzuzhang.getCode())) {
-                list = Collections.emptyList();
-            } else {
-                list = Collections.emptyList();
-            }
+                break;
+            default:
+                break;
         }
         ExcelUtil<CheckRecord> util = new ExcelUtil<CheckRecord>(CheckRecord.class);
         util.exportExcel(response, list, "检查记录数据");