优化：流媒体

liusuyi

2024-07-11 b1084891961232e3c697ea9fc52f127cdccffb6b

 ard-work/src/main/java/com/ruoyi/media/service/impl/VtduServiceImpl.java

@@ -3,10 +3,18 @@
import java.util.ArrayList;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;



import com.ruoyi.common.utils.DateUtils;

import com.ruoyi.device.camera.domain.ArdCameras;

import com.ruoyi.device.camera.domain.CameraCmd;

import com.ruoyi.device.camera.factory.CameraSDK;

import com.ruoyi.device.camera.factory.CameraSDKFactory;

import com.ruoyi.device.channel.domain.ArdChannel;

import com.ruoyi.media.domain.StreamInfo;

import com.ruoyi.media.service.IMediaService;

import lombok.extern.slf4j.Slf4j;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.stereotype.Service;

import com.ruoyi.media.mapper.VtduMapper;

import com.ruoyi.media.domain.Vtdu;

@@ -28,7 +36,10 @@
    private VtduMapper vtduMapper;

    @Resource

    private IMediaService mediaService;



    @Resource

    private CameraSDKFactory cameraSDKFactory;

    @Value("${mediamtx.host}")

    String mediamtxHost;

    /**

     * 查询流媒体管理

     *

@@ -37,7 +48,6 @@
     */

    @Override

    public Vtdu selectVtduByName(String name) {



        return vtduMapper.selectVtduByName(name);

    }



@@ -60,17 +70,6 @@
     */

    @Override

    public int insertVtdu(Vtdu vtdu) {

        log.debug("流媒体【" + vtdu.getName() + "】通道添加");

        Map<String, String> map = mediaService.addPath(vtdu.getName(), vtdu.getRtspSource(), vtdu.getMode(), vtdu.getIsCode());

        vtdu.setRtspUrl(map.get("rtspUrl"));

        vtdu.setRtmpUrl(map.get("rtmpUrl"));

        vtdu.setWebrtcUrl(map.get("webrtcUrl"));



        vtdu.setName(vtdu.getName());

        vtdu.setRtspSource(vtdu.getRtspSource());

        vtdu.setIsCode(vtdu.getIsCode());

        vtdu.setMode(vtdu.getMode());

        vtdu.setCreateTime(DateUtils.getNowDate());

        return vtduMapper.insertVtdu(vtdu);

    }



@@ -82,7 +81,6 @@
     */

    @Override

    public int updateVtdu(Vtdu vtdu) {

        log.debug("流媒体【" + vtdu.getName() + "】通道更新");

        Map<String, String> map = mediaService.editPath(vtdu.getName(), vtdu.getRtspSource(), vtdu.getMode(), vtdu.getIsCode());

        vtdu.setName(vtdu.getName());

        vtdu.setRtspSource(vtdu.getRtspSource());

@@ -141,4 +139,111 @@
        }

        return vtduMapper.deleteVtduByCameraId(cameraId);

    }



    /**

     * @return

     * @Author 刘苏义

     * @Description 清除所有流媒体数据

     * @Date 2024/7/10 13:31

     * @Param

     */

    public void clearVtdu() {

        vtduMapper.clearVtdu();

    }





    //需要更新的（流媒体和vtdu相同）

    @Override

    public List<Vtdu> sameList(List<Vtdu> vtdus, List<String> names) {

        return vtdus.stream().filter(vtdu -> names.contains(vtdu.getName())).collect(Collectors.toList());

    }



    //需要删除的（流媒体中多的）

    @Override

    public List<String> diffListToDel(List<Vtdu> vtdus, List<String> names) {

        if (vtdus.size() >= names.size()) {

            List<String> nameList = vtdus.stream().map(Vtdu::getName)

                    .filter(item -> !names.contains(item))

                    .collect(Collectors.toList());

            return nameList;

        } else {

            List<String> nameList = names.stream().filter(item -> !vtdus.stream().map(Vtdu::getName).collect(Collectors.toList()).contains(item))

                    .collect(Collectors.toList());

            return nameList;

        }

    }



    //需要新增的（流媒体中少的）

    @Override

    public List<Vtdu> diffListToAdd(List<Vtdu> vtdus, List<String> names) {

        return vtdus.stream().filter(vtdu -> !names.contains(vtdu.getName())).collect(Collectors.toList());

    }



    /**

     * @return

     * @Author 刘苏义

     * @Description 同步本地vtdu库和流媒体中的数据

     * @Date 2024/7/10 15:26

     * @Param vtdus vtdu库的集合

     * @Param names 流媒体的name集合

     */

    @Override

    public void asyncVtdu(List<Vtdu> vtdus, List<String> names) {

        //需要更新的数据,参数顺序注意

        sameList(vtdus, names).stream().forEach(vtdu -> {

            StreamInfo streamInfo = mediaService.getPathInfo(vtdu.getName());

            if (!streamInfo.getRtspSource().equals(vtdu.getRtspSource())) {

                mediaService.editPath(vtdu.getName(), vtdu.getRtspSource(), vtdu.getMode(), vtdu.getIsCode());

            }

        });

        //需要删除的数据

        diffListToDel(vtdus, names).stream().forEach(name -> {

            mediaService.removePath(name);

        });

        //需要新增的数据

        diffListToAdd(vtdus, names).stream().forEach(vtdu -> {

            mediaService.addPath(vtdu.getName(), vtdu.getRtspSource(), vtdu.getMode(), vtdu.getIsCode());

        });

    }

    @Override

    public void addChanToVtdu(ArdCameras camera, ArdChannel channel) {

        String name = camera.getId() + "_" + channel.getChanNo();

        String rtspSource="";

        switch (camera.getFactory())

        {

            case "1": rtspSource = "rtsp://" + camera.getUsername() + ":" + camera.getPassword() + "@" + camera.getIp() + ":" + camera.getRtspPort() + "/h264/ch" + channel.getChanNo() + "/main/av_stream";break;

            case "2": rtspSource = "rtsp://" + camera.getUsername() + ":" + camera.getPassword() + "@" + camera.getIp() + ":" + camera.getRtspPort() + "/cam/realmonitor?channel=" + channel.getChanNo() + "&subtype=0";break;

        }



        //删除流媒体

        if (selectVtduByName(name) != null) {

            deleteVtduByName(name);

        }

        //添加到流媒体

        CameraCmd cmd = new CameraCmd(camera.getId(), channel.getChanNo());

        CameraSDK cameraSDK = cameraSDKFactory.createCameraSDK(camera.getFactory());

        Map<String, Object> videoCompressionCfg = cameraSDK.getVideoCompressionCfg(cmd);

        Vtdu vtdu = new Vtdu();

        if (videoCompressionCfg.get("videoEncType") != null) {

            if (videoCompressionCfg.get("videoEncType").equals("标准h264")) {

                vtdu.setIsCode("0");//默认不转码

            } else {

                vtdu.setIsCode("1");//默认转码

            }

        } else {

            vtdu.setIsCode("0");//默认不转码

        }

        vtdu.setRtspSource(rtspSource);

        vtdu.setName(name);

        vtdu.setMode("1");//默认CPU软解码

        vtdu.setCameraId(camera.getId());

        String rtspUrl = "rtsp://" + mediamtxHost + ":8554/" + name;

        String rtmpUrl = "rtmp://" + mediamtxHost + ":1935/" + name;

        String webrtcUrl = "http://" + mediamtxHost + ":8889/" + name;

        vtdu.setRtmpUrl(rtmpUrl);

        vtdu.setWebrtcUrl(webrtcUrl);

        vtdu.setRtspUrl(rtspUrl);

        insertVtdu(vtdu);

    }



}