forked from uav/ard-uav-server
blame | history | patch | commit | commitdiff | ignore whitespace
zhaowenyu
2025-11-29 2e5e7a94ada4509fa286d8438b6b127aaa906570 
 ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java


@@ -1,6 +1,7 @@


package com.ard.sample.manage.service.impl;




import com.alibaba.fastjson2.JSONObject;


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;


@@ -65,6 +66,9 @@


@Slf4j


@Transactional


public class DeviceServiceImpl implements IDeviceService {




    @Autowired


    private LiveStreamHandler _liveStreamHandler;




    @Autowired


    private IMessageSenderService messageSender;


@@ -355,6 +359,11 @@


                DeviceQueryParam.builder()


                        .domains(List.of(DeviceDomainEnum.GATEWAY.getVal(), DeviceDomainEnum.DOCK.getVal()))


                        .build());


        // 在獲取设备列表时，顺便获取直播地址


        for(DeviceDTO devices : devicesList){


            _liveStreamHandler.StartLiveStreaming(devices.getDeviceSn());


            _liveStreamHandler.StartLiveStreaming(devices.getChildDeviceSn());


        }




        devicesList.stream()


                .filter(gateway -> DeviceDomainEnum.DOCK.getVal() == gateway.getDomain() ||


@@ -753,6 +762,7 @@


                .firmwareVersion(entity.getFirmwareVersion())


                .deptId(entity.getDeptId())


                .workType(entity.getWorkType())


                .webrtcUrl(entity.getWebrtcUrl())


                .workspaceName(entity.getDeptId() != null


                        ? Optional.ofNullable(sysDeptService.selectDeptById(entity.getDeptId()))


                        .map(SysDept::getDeptName)


@@ -800,6 +810,15 @@


                // 只更新workType字段


                .set(DeviceEntity::getWorkType, workType);


        // 执行更新（第一个参数传null，因为更新字段已通过wrapper的set指定）


        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;


    }


@@ -1193,6 +1212,7 @@


                .domain(dto.getDomain())


                .firmwareVersion(dto.getFirmwareVersion())


                .workType(dto.getWorkType())


                .webrtcUrl(dto.getWebrtcUrl())


                .compatibleStatus(dto.getFirmwareStatus() == null ? null :


                        DeviceFirmwareStatusEnum.CONSISTENT_UPGRADE != DeviceFirmwareStatusEnum.find(dto.getFirmwareStatus()))


                .build();