From 1fd15505dc8bc4335f6cb9c1a0fc3e4b687b1ac0 Mon Sep 17 00:00:00 2001
From: zhaowenyu <35589726@qq.com>
Date: Sat, 01 Nov 2025 12:40:45 +0800
Subject: [PATCH] 2025-11-1 zwy 整理代码增加注释，做权限验证的设备列表为mapper中单独写的一个自定义方法
---
ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java | 109 +++++++++++++++++++++++++++++++++++++++++++++---------
1 files changed, 91 insertions(+), 18 deletions(-)
diff --git a/ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java b/ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java
index 7466075..02ea292 100644
--- a/ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java
+++ b/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);
+// }
}
\ No newline at end of file
--
Gitblit v1.9.3