forked from uav/ard-uav-server
ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java
@@ -1,8 +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; @@ -264,7 +268,7 @@ /** * 发布状态回复消息 * * @param sn 序列号 * @param sn 序列号 * @param response 通用主题响应对象 */ @Override @@ -312,7 +316,7 @@ /** * 设备列表 * */ */ @Override public List<DeviceDTO> getDevicesTopoForWeb() { List<DeviceDTO> devicesList = this.getDevicesByParams( @@ -346,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) { @@ -542,7 +604,7 @@ return Optional.of(deviceEntity); } // 2025-10-20 zwy 添加时增加授权验证及数量限制，限制为3台机场设备，超出则抛出异常 try{ try { // 查询数据库中机器设备数量 // 根据参数获取设备列表 List<DeviceDTO> devicesList = this.getDevicesByParams( @@ -552,8 +614,7 @@ .build()); // 验证授权数量 validateAuthorization(devicesList.size(), 1); } catch (Exception e) { } catch (Exception e) { return Optional.empty(); } // 调用验证方法 @@ -863,9 +924,9 @@ /** * 根据域名分页获取已绑定的设备信息 * * @param page 当前页码 * @param pageSize 每页显示数量 * @param domain 域名 * @param page 当前页码 * @param pageSize 每页显示数量 * @param domain 域名 * @return PaginationData<DeviceDTO> 包含分页信息的设备DTO列表 */ @Override @@ -873,11 +934,17 @@ Long pageSize, Integer domain, Long deptId) { Page<DeviceEntity> pagination = mapper.selectPage(new Page<>(page, pageSize), new LambdaQueryWrapper<DeviceEntity>() .eq(DeviceEntity::getDomain, domain) .eq(deptId != null, DeviceEntity::getDeptId, deptId) .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())); @@ -1186,7 +1253,7 @@ * 处理获取机场组织信息的请求 * * @param receiver 接收到的消息 * @param headers 消息头 * @param headers 消息头 */ @ServiceActivator(inputChannel = ChannelName.INBOUND_REQUESTS_AIRPORT_ORGANIZATION_GET, outputChannel = ChannelName.OUTBOUND) @@ -1195,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; @@ -1219,4 +1286,10 @@ messageSender.publish(topic, builder.build()); } // @DataScopePlus(deptAlias="d") // @Override // public List<DeviceEntity> selectList(Wrapper<DeviceEntity> queryWrapper) { // return mapper.selectList(queryWrapper); // } }
