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; @@ -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); } } /** @@ -811,20 +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) .isNotNull(DeviceEntity::getDeptId)); // .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())); @@ -981,18 +1101,18 @@ .build(), 2); while (true) { reply = (Map<String, Object>) reply.get(value.getKey()); if (value.getValue() instanceof JsonNode) { break; } value = (Map.Entry) value.getValue(); } SetReply setReply = objectMapper.convertValue(reply, SetReply.class); if (SetReplyStatusResultEnum.SUCCESS.getVal() != setReply.getResult()) { throw new RuntimeException("Failed to set " + value.getKey() + "; Error Code: " + setReply.getResult()); } // 接收返回信息在serviceReply中，订阅thing/product/7CTDLCE00A1UPE/property/set_reply上报的信息，此处不处理 // while (true) { // reply = (Map<String, Object>) reply.get(value.getKey()); // if (value.getValue() instanceof JsonNode) { // break; // } // value = (Map.Entry) value.getValue(); // } // SetReply setReply = objectMapper.convertValue(reply, SetReply.class); // if (SetReplyStatusResultEnum.SUCCESS.getVal() != setReply.getResult()) { // throw new RuntimeException("Failed to set " + value.getKey() + "; Error Code: " + setReply.getResult()); // } } @@ -1133,7 +1253,7 @@ * 处理获取机场组织信息的请求 * * @param receiver 接收到的消息 * @param headers 消息头 * @param headers 消息头 */ @ServiceActivator(inputChannel = ChannelName.INBOUND_REQUESTS_AIRPORT_ORGANIZATION_GET, outputChannel = ChannelName.OUTBOUND) @@ -1142,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; @@ -1166,4 +1286,10 @@ messageSender.publish(topic, builder.build()); } // @DataScopePlus(deptAlias="d") // @Override // public List<DeviceEntity> selectList(Wrapper<DeviceEntity> queryWrapper) { // return mapper.selectList(queryWrapper); // } }
