From 73bc82f63cecf23b74cbf59ebb8bdfb40fe6e5b0 Mon Sep 17 00:00:00 2001
From: zhaowenyu <35589726@qq.com>
Date: Mon, 01 Dec 2025 11:25:28 +0800
Subject: [PATCH] 2025-12-01 zwy 多路直播,将外接相机直播地址插入设备列表中。数据库增加字段“cameraWebrtcUrl”,实体增加cameraWebrtcUrl字段。

---
 ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java |  284 ++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 253 insertions(+), 31 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 955316b..d0f48f7 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,13 @@
 package com.ard.sample.manage.service.impl;
 
+import com.alibaba.fastjson2.JSONObject;
+import com.ard.device.camera.domain.ArdCamera;
+import com.ard.device.camera.service.IArdCameraService;
+import com.ard.device.channel.domain.ArdChannel;
+import com.ard.sample.component.websocket.handler.LiveStreamHandler;
+import com.ard.system.authorize.service.AuthorizeService;
 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;
@@ -64,6 +71,12 @@
 public class DeviceServiceImpl implements IDeviceService {
 
     @Autowired
+    private LiveStreamHandler _liveStreamHandler;
+
+    @Autowired
+    private IArdCameraService _cameraService;
+
+    @Autowired
     private IMessageSenderService messageSender;
 
     @Resource
@@ -89,6 +102,9 @@
 
     @Autowired
     private IDeviceFirmwareService deviceFirmwareService;
+
+    @Resource
+    private AuthorizeService authorizeService;
 
     @Autowired
     @Qualifier("gatewayOSDServiceImpl")
@@ -259,7 +275,7 @@
     /**
      * 发布状态回复消息
      *
-     * @param sn 序列号
+     * @param sn       序列号
      * @param response 通用主题响应对象
      */
     @Override
@@ -305,6 +321,9 @@
                 .collect(Collectors.toList());
     }
 
+    /**
+     * 设备列表
+     */
     @Override
     public List<DeviceDTO> getDevicesTopoForWeb() {
         List<DeviceDTO> devicesList = this.getDevicesByParams(
@@ -338,6 +357,80 @@
         // 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());
+        // 在獲取设备列表时,顺便获取直播地址
+        for(DeviceDTO devices : devicesList){
+            _liveStreamHandler.StartLiveStreaming(devices.getDeviceSn());
+            _liveStreamHandler.StartLiveStreaming(devices.getChildDeviceSn());
+            ArdCamera ardCamera = new ArdCamera();
+            ardCamera.setDockId(devices.getId());
+            List<ArdCamera> ardCameraList = _cameraService.selectArdCameraList(ardCamera);
+            for(ArdCamera ardCamera1 : ardCameraList){
+                for(ArdChannel ardChannel : ardCamera1.getChannelList()) {
+                    if(ardChannel.getChanNo() == 1) {
+                        // 修改列表中外接相机地址
+                        updateCameraWebrtcUrl(devices.getDeviceSn(), ardChannel.getArdVtdu().getWebrtcUrl());
+                    }
+                }
+            }
+        }
+
+        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) {
@@ -432,10 +525,10 @@
     @Override
     @ServiceActivator(inputChannel = ChannelName.INBOUND_OSD)
     public void handleOSD(Message<?> message) {
-        String topic = message.getHeaders().get(MqttHeaders.RECEIVED_TOPIC).toString();
-        byte[] payload = (byte[]) message.getPayload();
-        CommonTopicReceiver receiver;
         try {
+            String topic = message.getHeaders().get(MqttHeaders.RECEIVED_TOPIC).toString();
+            byte[] payload = (byte[]) message.getPayload();
+            CommonTopicReceiver receiver;
             String from = topic.substring((THING_MODEL_PRE + PRODUCT).length(),
                     topic.indexOf(OSD_SUF));
 
@@ -473,7 +566,12 @@
 
             tsaService.handleOSD(receiver, device, webSessions, wsMessage);
 
-        } catch (IOException e) {
+        } catch (NullPointerException e) {
+            log.error("处理OSD消息NPE,headers={}, payload={}",
+                    message.getHeaders(), message.getPayload(), e); // 打印完整栈
+        }
+        catch (IOException e) {
+            log.error("处理OSD消息的IO异常:{}",e.getMessage(), e);
             e.printStackTrace();
         }
     }
@@ -533,7 +631,51 @@
             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) {
+            log.error("保存设备时验证授权异常,请联系管理员!错误信息:" + e.getMessage());
+            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);
+        }
     }
 
     /**
@@ -637,6 +779,8 @@
                 .firmwareVersion(entity.getFirmwareVersion())
                 .deptId(entity.getDeptId())
                 .workType(entity.getWorkType())
+                .webrtcUrl(entity.getWebrtcUrl())
+                .cameraWebrtcUrl(entity.getCameraWebrtcUrl())
                 .workspaceName(entity.getDeptId() != null
                         ? Optional.ofNullable(sysDeptService.selectDeptById(entity.getDeptId()))
                         .map(SysDept::getDeptName)
@@ -687,6 +831,25 @@
         int update = mapper.update(null, lambdaUpdateWrapper);
         return update > 0;
     }
+
+    @Override
+    public Boolean updateWebrtcUrl(String sn, String webrtcUrl){
+        LambdaUpdateWrapper<DeviceEntity> lambdaUpdateWrapper = new LambdaUpdateWrapper<>();
+        lambdaUpdateWrapper.eq(DeviceEntity::getDeviceSn, sn)
+                .set(DeviceEntity::getWebrtcUrl, webrtcUrl);
+        int update = mapper.update(null, lambdaUpdateWrapper);
+        return update > 0;
+    }
+
+    public Boolean updateCameraWebrtcUrl(String sn, String CameraWebrtcUrl){
+        LambdaUpdateWrapper<DeviceEntity> lambdaUpdateWrapper = new LambdaUpdateWrapper<>();
+        lambdaUpdateWrapper.eq(DeviceEntity::getDeviceSn, sn)
+                .set(DeviceEntity::getCameraWebrtcUrl, CameraWebrtcUrl);
+        int update = mapper.update(null, lambdaUpdateWrapper);
+        return update > 0;
+    }
+
+
 
     @Override
     public Boolean bindDevice(DeviceDTO device) {
@@ -763,6 +926,9 @@
         BindDeviceReceiver dock = null;
         BindDeviceReceiver drone = null;
         for (BindDeviceReceiver device : devices) {
+            if(device.getDeviceModelKey().isEmpty()){
+                continue;
+            }
             int val = Integer.parseInt(device.getDeviceModelKey().split("-")[0]);
             if (val == DeviceDomainEnum.DOCK.getVal()) {
                 dock = device;
@@ -811,20 +977,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()));
@@ -952,6 +1125,16 @@
             this.deviceOnePropertySet(topic, propertyEnum, Map.entry(propertyEnum.getProperty(), param));
             return;
         }
+
+        if(param.size()==1){
+            this.deviceOnePropertySetSilent(topic,propertyEnum,param);
+            return;
+        }
+//        if(Objects.equals(param.fieldNames().next(), "silent_mode"))
+//        {
+//            this.deviceOnePropertySet(topic, propertyEnum, Map.entry(propertyEnum.getProperty(), param.get("silent_mode")));
+//            return;
+//        }
         // If there are multiple parameters, set them separately.
         for (Iterator<Map.Entry<String, JsonNode>> filed = param.fields(); filed.hasNext(); ) {
             Map.Entry<String, JsonNode> node = filed.next();
@@ -959,12 +1142,34 @@
             if (!isPublish) {
                 continue;
             }
+
             this.deviceOnePropertySet(topic, propertyEnum, Map.entry(propertyEnum.getProperty(), node));
         }
 
     }
 
+    /**
+     * 为静音模式单独列出来一个方法,只有一个值的时候都可以调用,不限于静音模式
+     * */
     @Override
+    public void deviceOnePropertySetSilent(String topic, DeviceSetPropertyEnum propertyEnum,
+                                     JsonNode value) {
+//        if (Objects.isNull(value) || Objects.isNull(value.getValue())) {
+//            throw new IllegalArgumentException(CommonErrorEnum.ILLEGAL_ARGUMENT.getErrorMsg());
+//        }
+
+        Map reply = messageSender.publishWithReply(
+                Map.class, topic,
+                CommonTopicResponse.builder()
+                        .bid(UUID.randomUUID().toString())
+                        .tid(UUID.randomUUID().toString())
+                        .timestamp(System.currentTimeMillis())
+                        .data(value)
+                        .build(),
+                2);
+    }
+
+        @Override
     public void deviceOnePropertySet(String topic, DeviceSetPropertyEnum propertyEnum,
                                      Map.Entry<String, Object> value) {
         if (Objects.isNull(value) || Objects.isNull(value.getValue())) {
@@ -981,20 +1186,29 @@
                         .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());
+//        }
 
     }
+
+//    public JsonNode getJsonNode(JsonNode jsonNode) {
+//        ObjectNode parentNode = objectMapper.createObjectNode();
+//        parentNode.put("bid", UUID.randomUUID().toString());
+//        parentNode.set("data", jsonNode);
+//        parentNode.put("tid", UUID.randomUUID().toString());
+//        parentNode.put("timestamp", System.currentTimeMillis());
+//        return parentNode;
+//    }
 
     @Override
     public Boolean checkDeviceOnline(String sn) {
@@ -1026,6 +1240,8 @@
                 .domain(dto.getDomain())
                 .firmwareVersion(dto.getFirmwareVersion())
                 .workType(dto.getWorkType())
+                .webrtcUrl(dto.getWebrtcUrl())
+                .cameraWebrtcUrl(dto.getCameraWebrtcUrl())
                 .compatibleStatus(dto.getFirmwareStatus() == null ? null :
                         DeviceFirmwareStatusEnum.CONSISTENT_UPGRADE != DeviceFirmwareStatusEnum.find(dto.getFirmwareStatus()))
                 .build();
@@ -1085,7 +1301,7 @@
                 .sn(device.getDeviceSn())
                 .deviceCallsign(device.getNickname())
                 .isDeviceBindOrganization(device.getBoundStatus())
-                .organizationId(device.getDeptId().toString())
+                .organizationId(String.valueOf(device.getDeptId()))
                 .organizationName(device.getWorkspaceName())
                 .build();
     }
@@ -1133,7 +1349,7 @@
      * 处理获取机场组织信息的请求
      *
      * @param receiver 接收到的消息
-     * @param headers 消息头
+     * @param headers  消息头
      */
     @ServiceActivator(inputChannel = ChannelName.INBOUND_REQUESTS_AIRPORT_ORGANIZATION_GET, outputChannel =
             ChannelName.OUTBOUND)
@@ -1142,10 +1358,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 +1382,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