优化：流媒体同步任务

liusuyi

2024-07-10 016aa105789fca71e86cdbb0d26a181812f200f9

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

@@ -1,145 +1,211 @@
package com.ruoyi.media.service.impl;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.media.service.IMediaService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import com.ruoyi.media.mapper.VtduMapper;
import com.ruoyi.media.domain.Vtdu;
import com.ruoyi.media.service.IVtduService;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * 流媒体管理Service业务层处理
 *
 * @author ard
 * @date 2023-08-29
 */
@Service
@Slf4j(topic = "vtdu")
public class VtduServiceImpl implements IVtduService {
    @Resource
    private VtduMapper vtduMapper;
    @Resource
    private IMediaService mediaService;

    /**
     * 查询流媒体管理
     *
     * @param name 流媒体管理主键
     * @return 流媒体管理
     */
    @Override
    public Vtdu selectVtduByName(String name) {

        return vtduMapper.selectVtduByName(name);
    }

    /**
     * 查询流媒体管理列表
     *
     * @param vtdu 流媒体管理
     * @return 流媒体管理
     */
    @Override
    public List<Vtdu> selectVtduList(Vtdu vtdu) {
        return vtduMapper.selectVtduList(vtdu);
    }

    /**
     * 新增流媒体管理
     *
     * @param vtdu 流媒体管理
     * @return 结果
     */
    @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);
    }

    /**
     * 修改流媒体管理
     *
     * @param vtdu 流媒体管理
     * @return 结果
     */
    @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());
        vtdu.setIsCode(vtdu.getIsCode());
        vtdu.setMode(vtdu.getMode());
        vtdu.setRtspUrl(map.get("rtspUrl"));
        vtdu.setRtmpUrl(map.get("rtmpUrl"));
        vtdu.setWebrtcUrl(map.get("webrtcUrl"));
        vtdu.setUpdateTime(DateUtils.getNowDate());
        return vtduMapper.updateVtdu(vtdu);
    }

    /**
     * 批量删除流媒体管理
     *
     * @param names 需要删除的数据主键集合
     * @return 结果
     */
    @Override
    public int deleteVtduByNames(String[] names) {
        for (String name : names) {
            log.debug("流媒体【" + name + "】通道删除");
        }
        mediaService.removePath(names);
        return vtduMapper.deleteVtduByNames(names);
    }

    /**
     * 删除流媒体管理
     *
     * @param name 流媒体通道名称
     * @return 结果
     */
    @Override
    public int deleteVtduByName(String name) {
        mediaService.removePath(name);
        log.debug("流媒体【" + name + "】通道删除");
        return vtduMapper.deleteVtduByName(name);
    }

    /**
     * 删除流媒体管理信息
     *
     * @param cameraId 相机ID
     * @return 结果
     */
    @Override
    public int deleteVtduByCameraId(String cameraId) {
        Vtdu vtdu = new Vtdu();
        vtdu.setCameraId(cameraId);
        List<Vtdu> vtdus = vtduMapper.selectVtduList(vtdu);
        if (vtdus.size() > 0) {
            for (Vtdu v : vtdus) {
                mediaService.removePath(v.getName());
            }
        }
        return vtduMapper.deleteVtduByCameraId(cameraId);
    }
}
package com.ruoyi.media.service.impl;



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.channel.domain.ArdChannel;

import com.ruoyi.media.service.IMediaService;

import lombok.extern.slf4j.Slf4j;

import org.springframework.stereotype.Service;

import com.ruoyi.media.mapper.VtduMapper;

import com.ruoyi.media.domain.Vtdu;

import com.ruoyi.media.service.IVtduService;



import javax.annotation.PostConstruct;

import javax.annotation.Resource;



/**

 * 流媒体管理Service业务层处理

 *

 * @author ard

 * @date 2023-08-29

 */

@Service

@Slf4j(topic = "vtdu")

public class VtduServiceImpl implements IVtduService {

    @Resource

    private VtduMapper vtduMapper;

    @Resource

    private IMediaService mediaService;



    /**

     * 查询流媒体管理

     *

     * @param name 流媒体管理主键

     * @return 流媒体管理

     */

    @Override

    public Vtdu selectVtduByName(String name) {



        return vtduMapper.selectVtduByName(name);

    }



    /**

     * 查询流媒体管理列表

     *

     * @param vtdu 流媒体管理

     * @return 流媒体管理

     */

    @Override

    public List<Vtdu> selectVtduList(Vtdu vtdu) {

        return vtduMapper.selectVtduList(vtdu);

    }



    /**

     * 新增流媒体管理

     *

     * @param vtdu 流媒体管理

     * @return 结果

     */

    @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);

    }



    /**

     * 修改流媒体管理

     *

     * @param vtdu 流媒体管理

     * @return 结果

     */

    @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());

        vtdu.setIsCode(vtdu.getIsCode());

        vtdu.setMode(vtdu.getMode());

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

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

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

        vtdu.setUpdateTime(DateUtils.getNowDate());

        return vtduMapper.updateVtdu(vtdu);

    }



    /**

     * 批量删除流媒体管理

     *

     * @param names 需要删除的数据主键集合

     * @return 结果

     */

    @Override

    public int deleteVtduByNames(String[] names) {

        for (String name : names) {

            log.debug("流媒体【" + name + "】通道删除");

        }

        mediaService.removePath(names);

        return vtduMapper.deleteVtduByNames(names);

    }



    /**

     * 删除流媒体管理

     *

     * @param name 流媒体通道名称

     * @return 结果

     */

    @Override

    public int deleteVtduByName(String name) {

        mediaService.removePath(name);

        log.debug("流媒体【" + name + "】通道删除");

        return vtduMapper.deleteVtduByName(name);

    }



    /**

     * 删除流媒体管理信息

     *

     * @param cameraId 相机ID

     * @return 结果

     */

    @Override

    public int deleteVtduByCameraId(String cameraId) {

        Vtdu vtdu = new Vtdu();

        vtdu.setCameraId(cameraId);

        List<Vtdu> vtdus = vtduMapper.selectVtduList(vtdu);

        if (vtdus.size() > 0) {

            for (Vtdu v : vtdus) {

                mediaService.removePath(v.getName());

            }

        }

        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());

    }



    /**

     * @Author 刘苏义

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

     * @Date   2024/7/10 15:26

     * @Param  vtdus vtdu库的集合

     * @Param  names 流媒体的name集合

     * @return

     */

    @Override

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

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

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

            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());

        });

    }

}