From 3721eb9648e1a6a74a956a7688e380115007b6ba Mon Sep 17 00:00:00 2001
From: zhaowenyu <35589726@qq.com>
Date: Mon, 03 Nov 2025 11:47:52 +0800
Subject: [PATCH] 2025-11-3 zwy 机场详情页面中的机场与飞行器状态发给前端的JSON调整

---
 ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java |  153 ++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 139 insertions(+), 14 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 cd29c0f..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,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,9 +924,9 @@
     /**
      * 根据域名分页获取已绑定的设备信息
      *
-     * @param page      当前页码
-     * @param pageSize  每页显示数量
-     * @param domain    域名
+     * @param page     当前页码
+     * @param pageSize 每页显示数量
+     * @param domain   域名
      * @return PaginationData<DeviceDTO> 包含分页信息的设备DTO列表
      */
     @Override
@@ -821,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()));
@@ -1134,7 +1253,7 @@
      * 处理获取机场组织信息的请求
      *
      * @param receiver 接收到的消息
-     * @param headers 消息头
+     * @param headers  消息头
      */
     @ServiceActivator(inputChannel = ChannelName.INBOUND_REQUESTS_AIRPORT_ORGANIZATION_GET, outputChannel =
             ChannelName.OUTBOUND)
@@ -1143,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;
 
@@ -1167,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