diff --git a/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java b/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java index dccaa495..d23572c9 100644 --- a/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java +++ b/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java @@ -98,18 +98,35 @@ public class DeviceTypeServiceImpl extends ServiceImpl queryDeviceTypes() { DeviceTypeQueryCriteria criteria = new DeviceTypeQueryCriteria(); - // 管理员 - String username = LoginHelper.getUsername(); - if (!username.equals("admin")) { - criteria.setCustomerId(LoginHelper.getUserId()); + // // 管理员 + // String username = LoginHelper.getUsername(); + // if (!username.equals("admin")) { + // criteria.setCustomerId(LoginHelper.getUserId()); + // + // Long userId = LoginHelper.getUserId(); + // criteria.setCustomerId(userId); + // } - Long userId = LoginHelper.getUserId(); - criteria.setCustomerId(userId); + // 角色管理员 + Long userId = LoginHelper.getUserId(); + List roles = roleService.selectRolesAuthByUserId(userId); + boolean isAdmin = false; + if (CollectionUtil.isNotEmpty(roles)) { + for (SysRoleVo role : roles) { + if (role.getRoleKey().contains("admin")) { + isAdmin = true; + break; + } + } + } + if (!isAdmin) { + criteria.setCustomerId(LoginHelper.getUserId()); } return deviceTypeMapper.findAll(criteria); } + /** * 根据设备类型名称查询设备类型 *