forked from uav/ard-uav-server
ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java
@@ -1,6 +1,12 @@ package com.ard.sample.manage.service.impl; import com.alibaba.fastjson2.JSONObject; import com.ard.common.annotation.DataScope; import com.ard.common.annotation.DataScopePlus; import com.ard.system.authorize.service.AuthorizeService; import com.baomidou.mybatisplus.core.conditions.Wrapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.ard.common.core.domain.entity.SysDept; @@ -26,7 +32,7 @@ import com.ard.sample.manage.model.param.DeviceQueryParam; import com.ard.sample.manage.model.receiver.*; import com.ard.sample.manage.service.*; import com.ard.system.service.system.ISysDeptService; import com.ard.system.service.ISysDeptService; import com.fasterxml.jackson.core.type.TypeReference; import com.fasterxml.jackson.databind.JsonNode; import com.fasterxml.jackson.databind.ObjectMapper; @@ -89,6 +95,9 @@ @Autowired private IDeviceFirmwareService deviceFirmwareService; @Resource private AuthorizeService authorizeService; @Autowired @Qualifier("gatewayOSDServiceImpl") @@ -259,7 +268,7 @@ /** * 发布状态回复消息 * * @param sn 序列号 * @param sn 序列号 * @param response 通用主题响应对象 */ @Override @@ -305,6 +314,9 @@ .collect(Collectors.toList()); } /** * 设备列表 */ @Override public List<DeviceDTO> getDevicesTopoForWeb() { List<DeviceDTO> devicesList = this.getDevicesByParams( @@ -338,6 +350,64 @@ // payloads // subDevice.setPayloadsList(payloadService.getDevicePayloadEntitiesByDeviceSn(gateway.getChildDeviceSn())); } //region 单独写一个方法做权限验证，由于有好多定时任务也在实时获取设备列表，原方法不适用再添加权限注解 @Override public List<DeviceDTO> getDockList() { List<DeviceDTO> devicesList = this.getDockListByParams( DeviceQueryParam.builder() .domains(List.of(DeviceDomainEnum.GATEWAY.getVal(), DeviceDomainEnum.DOCK.getVal())) .build()); devicesList.stream() .filter(gateway -> DeviceDomainEnum.DOCK.getVal() == gateway.getDomain() || RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + gateway.getDeviceSn())) .forEach(this::spliceDeviceTopoPer); return devicesList; } @Override public void spliceDeviceTopoPer(DeviceDTO gateway) { gateway.setStatus(RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + gateway.getDeviceSn())); // sub device if (!StringUtils.hasText(gateway.getChildDeviceSn())) { return; } DeviceDTO subDevice = getDevicesByParams(DeviceQueryParam.builder().deviceSn(gateway.getChildDeviceSn()).build()).get(0); subDevice.setStatus(RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + subDevice.getDeviceSn())); gateway.setChildren(subDevice); } @Override public List<DeviceDTO> getDockListByParams(DeviceQueryParam param){ // 在mapper中添加查询方法做权限验证 return mapper.selectListPer( new LambdaQueryWrapper<DeviceEntity>() .eq(StringUtils.hasText(param.getDeviceSn()), DeviceEntity::getDeviceSn, param.getDeviceSn()) .eq(param.getDeviceType() != null, DeviceEntity::getDeviceType, param.getDeviceType()) .eq(param.getSubType() != null, DeviceEntity::getSubType, param.getSubType()) .eq(StringUtils.hasText(param.getChildSn()), DeviceEntity::getChildSn, param.getChildSn()) .and(!CollectionUtils.isEmpty(param.getDomains()), wrapper -> { for (Integer domain : param.getDomains()) { wrapper.eq(DeviceEntity::getDomain, domain).or(); } }) .eq(param.getBoundStatus() != null, DeviceEntity::getBoundStatus, param.getBoundStatus()) .orderBy(param.isOrderBy(), param.isAsc(), DeviceEntity::getId)) .stream() .map(this::deviceEntityConvertToDTO) .collect(Collectors.toList()); } //endregion @Override public Optional<TopologyDeviceDTO> getDeviceTopoForPilot(String sn) { @@ -533,7 +603,50 @@ mapper.updateById(entity); return Optional.of(deviceEntity); } // 2025-10-20 zwy 添加时增加授权验证及数量限制，限制为3台机场设备，超出则抛出异常 try { // 查询数据库中机器设备数量 // 根据参数获取设备列表 List<DeviceDTO> devicesList = this.getDevicesByParams( DeviceQueryParam.builder() // 过滤设备域为GATEWAY和DOCK的设备 .domains(List.of(DeviceDomainEnum.GATEWAY.getVal(), DeviceDomainEnum.DOCK.getVal())) .build()); // 验证授权数量 validateAuthorization(devicesList.size(), 1); } catch (Exception e) { return Optional.empty(); } // 调用验证方法 return mapper.insert(entity) > 0 ? Optional.of(entity) : Optional.empty(); } /** * 验证授权数量 */ private void validateAuthorization(int existingCount, int newCount) { int totalCount = existingCount + newCount; JSONObject authInfo = authorizeService.getAuthorizationInfo(); String authType = authInfo.getString("Type"); switch (authType) { case "normal": int allowedDockNum = authInfo.getIntValue("dockNum"); if (totalCount > allowedDockNum) { throw new RuntimeException(String.format( "授权数量%d，当前操作将导致总数达到%d，超过授权限制", allowedDockNum, totalCount)); } break; case "expired": throw new RuntimeException("授权文件已过期"); case "mismatch": throw new RuntimeException("授权文件不匹配"); default: throw new RuntimeException("未知的授权类型: " + authType); } } /** @@ -636,6 +749,7 @@ .nickname(entity.getNickname()) .firmwareVersion(entity.getFirmwareVersion()) .deptId(entity.getDeptId()) .workType(entity.getWorkType()) .workspaceName(entity.getDeptId() != null ? Optional.ofNullable(sysDeptService.selectDeptById(entity.getDeptId())) .map(SysDept::getDeptName) @@ -671,6 +785,19 @@ public Boolean updateDevice(DeviceDTO deviceDTO) { int update = mapper.update(this.deviceDTO2Entity(deviceDTO), new LambdaUpdateWrapper<DeviceEntity>().eq(DeviceEntity::getDeviceSn, deviceDTO.getDeviceSn())); return update > 0; } @Override public Boolean updateWorkType(String sn, Long workType) { // 方式1：使用LambdaUpdateWrapper（推荐，类型安全，避免字段名硬编码） LambdaUpdateWrapper<DeviceEntity> lambdaUpdateWrapper = new LambdaUpdateWrapper<>(); // 设置更新条件：deviceSn = sn lambdaUpdateWrapper.eq(DeviceEntity::getDeviceSn, sn) // 只更新workType字段 .set(DeviceEntity::getWorkType, workType); // 执行更新（第一个参数传null，因为更新字段已通过wrapper的set指定） int update = mapper.update(null, lambdaUpdateWrapper); return update > 0; } @@ -797,19 +924,27 @@ /** * 根据域名分页获取已绑定的设备信息 * * @param page 当前页码 * @param pageSize 每页显示数量 * @param domain 域名 * @param page 当前页码 * @param pageSize 每页显示数量 * @param domain 域名 * @return PaginationData<DeviceDTO> 包含分页信息的设备DTO列表 */ @Override public PaginationData<DeviceDTO> getBoundDevicesWithDomain(Long page, Long pageSize, Integer domain) { Long pageSize, Integer domain, Long deptId) { Page<DeviceEntity> pagination = mapper.selectPage(new Page<>(page, pageSize), new LambdaQueryWrapper<DeviceEntity>() .eq(DeviceEntity::getDomain, domain) .eq(DeviceEntity::getBoundStatus, true)); new QueryWrapper<DeviceEntity>() .eq("m.domain", domain) .eq(deptId != null, "m.dept_id", deptId) .isNotNull("m.dept_id")); //以前的写法 // new LambdaQueryWrapper<DeviceEntity>() // .eq(DeviceEntity::getDomain, domain) // .eq(deptId != null, DeviceEntity::getDeptId, deptId) // .isNotNull(DeviceEntity::getDeptId); List<DeviceDTO> devicesList = pagination.getRecords().stream().map(this::deviceEntityConvertToDTO) .peek(device -> { device.setStatus(RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + device.getDeviceSn())); @@ -835,6 +970,7 @@ DeviceDTO device = DeviceDTO.builder() .deviceSn(deviceSn) .boundStatus(false) .deptId(null) .build(); this.updateDevice(device); } @@ -1009,6 +1145,7 @@ .childSn(dto.getChildDeviceSn()) .domain(dto.getDomain()) .firmwareVersion(dto.getFirmwareVersion()) .workType(dto.getWorkType()) .compatibleStatus(dto.getFirmwareStatus() == null ? null : DeviceFirmwareStatusEnum.CONSISTENT_UPGRADE != DeviceFirmwareStatusEnum.find(dto.getFirmwareStatus())) .build(); @@ -1116,7 +1253,7 @@ * 处理获取机场组织信息的请求 * * @param receiver 接收到的消息 * @param headers 消息头 * @param headers 消息头 */ @ServiceActivator(inputChannel = ChannelName.INBOUND_REQUESTS_AIRPORT_ORGANIZATION_GET, outputChannel = ChannelName.OUTBOUND) @@ -1125,10 +1262,10 @@ OrganizationGetReceiver.class); CommonTopicResponse.CommonTopicResponseBuilder<RequestsReply> builder = CommonTopicResponse.<RequestsReply>builder() .tid(receiver.getTid()) .bid(receiver.getBid()) .method(RequestsMethodEnum.AIRPORT_ORGANIZATION_GET.getMethod()) .timestamp(System.currentTimeMillis()); .tid(receiver.getTid()) .bid(receiver.getBid()) .method(RequestsMethodEnum.AIRPORT_ORGANIZATION_GET.getMethod()) .timestamp(System.currentTimeMillis()); String topic = headers.get(MqttHeaders.RECEIVED_TOPIC).toString() + TopicConst._REPLY_SUF; @@ -1149,4 +1286,10 @@ messageSender.publish(topic, builder.build()); } // @DataScopePlus(deptAlias="d") // @Override // public List<DeviceEntity> selectList(Wrapper<DeviceEntity> queryWrapper) { // return mapper.selectList(queryWrapper); // } }
