changes

liuhaonan

2022-10-25 d495f9b8cdc83663e4189bc3cc72ac9543ff5555

 ximon-admin/src/main/java/com/sandu/ximon/admin/service/IpVolumeMissionService.java

@@ -1,14 +1,15 @@
package com.sandu.ximon.admin.service;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.pagehelper.PageHelper;
import com.sandu.common.execption.BusinessException;
import com.sandu.common.object.BaseConditionVO;
import com.sandu.common.service.impl.BaseServiceImpl;
import com.sandu.ximon.admin.config.MinIoConstant;
import com.sandu.ximon.admin.param.IpVolumeMissionParam;
import com.sandu.ximon.admin.security.SecurityUtils;
import com.sandu.ximon.admin.utils.StoreOperationRecordsUtils;
import com.sandu.ximon.admin.utils.TaskAPIUtils;
import com.sandu.ximon.admin.vo.TaskMediaFileVO;
import com.sandu.ximon.admin.vo.TaskTerminalVO;
@@ -18,6 +19,7 @@
import com.sandu.ximon.dao.domain.BroadcastV2TaskTerminal;
import com.sandu.ximon.dao.domain.IpVolumeFile;
import com.sandu.ximon.dao.domain.IpVolumeMission;
import com.sandu.ximon.dao.enums.OrderByEnums;
import com.sandu.ximon.dao.mapper.IpVolumeMissionMapper;
import lombok.AllArgsConstructor;
import org.springframework.beans.BeanUtils;
@@ -25,7 +27,6 @@

import java.math.BigInteger;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;

@@ -39,15 +40,25 @@
public class IpVolumeMissionService extends BaseServiceImpl<IpVolumeMissionMapper, IpVolumeMission> {
    private final IpVolumeMissionMapper ipVolumeMissionMapper;
    private final ClientService clientService;
    private final IpVolumeService ipVolumeService;
    private final IpVolumeFileService ipVolumeFileService;
    private final BroadcastV2TaskFileService broadcastV2TaskFileService;
    private final BroadcastV2TaskTerminalService broadcastV2TaskTerminalService;


    /*public boolean add1() {
      return   broadcastV2TaskTerminalService.save();
    }*/
    public boolean addMission(IpVolumeMissionParam missionparam) {

        List<Integer> fileIds = missionparam.getFileIds();
        List<IpVolumeFile> ipVolumeFiles = ipVolumeFileService.listByIds(fileIds);
        ipVolumeFiles.forEach(file -> {
            if (SecurityUtils.getClientId() != null) {
                //非超管用户
                if (!SecurityUtils.getUserId().equals(file.getClientId()) && !SecurityUtils.getUserId().equals(file.getUserId())) {
                    //用户id和客户id都不匹配
                    throw new BusinessException("文件归属异常,创建任务失败");
                }
            }
        });

        String taskName = missionparam.getMissionName();

@@ -63,14 +74,14 @@

        if (missionparam.getTerminalIds().size() <= 0) {
            throw new BusinessException("播放终端不可为空");
           // return 3;
            // return 3;
        }

        //  封装播放终端
        List<TaskTerminalVO> taskTerminalVOList = new ArrayList<>();
        //  封装数据库中任务终端绑定
        List<BroadcastV2TaskTerminal> taskTerminalEntityList = new ArrayList<>();
        for(Integer id: new HashSet<>(missionparam.getTerminalIds())) {
        for (Integer id : new HashSet<>(missionparam.getTerminalIds())) {
            TaskTerminalVO taskTerminalVO = new TaskTerminalVO();
            BroadcastV2TaskTerminal taskTerminalEntity = new BroadcastV2TaskTerminal();

@@ -85,22 +96,21 @@
        List<TaskMediaFileVO> taskMediaFileVOS = new ArrayList<>();
        List<BroadcastV2TaskFile> taskFileEntityList = new ArrayList<>();
        int i = 1;
        for(Integer fileId: new HashSet<>(missionparam.getFileIds())) {
        for (Integer fileId : new HashSet<>(missionparam.getFileIds())) {
            TaskMediaFileVO taskMediaFileVO = new TaskMediaFileVO();
            BroadcastV2TaskFile taskFileEntity = new BroadcastV2TaskFile();

            taskFileEntity.setFileId(fileId);

            IpVolumeFile fileEntity = ipVolumeFileService.getById(fileId);
            String absolutePath = MinIoConstant.getAbsolutePath(
                    MinIoConstant.BROADCAST_FILE.getBucketName() + fileEntity.getUserId(),
                    fileEntity.getFileName()
            );
            if (fileEntity == null) {
                throw new BusinessException("fileId为" + fileId + "的文件不存在");
            }

            taskMediaFileVO.setI(i);
            i++;
            taskMediaFileVO.setT(0);
            taskMediaFileVO.setN(absolutePath);
            taskMediaFileVO.setN(fileEntity.getFileUrl());
            taskMediaFileVO.setS(BigInteger.valueOf(fileEntity.getOriginSize()));

            taskMediaFileVOS.add(taskMediaFileVO);
@@ -112,19 +122,19 @@
        missionparam.setTerminals(taskTerminalVOList.toArray(new TaskTerminalVO[taskTerminalVOList.size()]));
        System.out.println("---------------------------");
        //  向IP音柱服务器添加任务
        //Integer taskId = TaskAPIUtils.addTask(missionparam);
       // System.out.println("---------------------------");
        Integer taskId = 1111;
        Integer taskId = TaskAPIUtils.addTask(missionparam);
//        Integer taskId = 1111;

        boolean flag;
        if (null != taskId && taskId != 0) {
            IpVolumeMission byId = getById(taskId);
            for (BroadcastV2TaskTerminal v2TaskTerminal: taskTerminalEntityList) {
            for (BroadcastV2TaskTerminal v2TaskTerminal : taskTerminalEntityList) {
                v2TaskTerminal.setTaskId(taskId);
            }
            for (BroadcastV2TaskFile v2TaskFile :taskFileEntityList) {
            for (BroadcastV2TaskFile v2TaskFile : taskFileEntityList) {
                v2TaskFile.setTaskId(taskId);
            }
            if (byId==null) {
            if (byId == null) {
                //  向数据库添加该任务记录

                IpVolumeMission mission = new IpVolumeMission();
@@ -139,9 +149,8 @@
                mission.setOffDate(missionparam.getOffDate());
                mission.setWeek(Integer.parseInt(missionparam.getWeek(), 2));//时间由二进制字符串转换为十进制
                mission.setMissionVolume(missionparam.getMissionVolume());
                //mission.setMissionFileUrl(missionparam.getFiles().toString());
                mission.setNumberInPlay(missionparam.getTerminals().length);
                /*TaskTerminalVO[] terminals = missionparam.getTerminals();*/


                System.out.println(save(mission));
                System.out.println("----------------------------------------------------------");
@@ -158,37 +167,45 @@
                mission.setMissionVolume(missionparam.getMissionVolume());
                mission.setMissionFileUrl(missionparam.getFiles().toString());
                mission.setNumberInPlay(missionparam.getTerminals().length);
                System.out.println(updateById(mission));
                System.out.println("----------------------------------------------------------");
            }
            broadcastV2TaskFileService.deleteByTaskId(taskId);
            broadcastV2TaskTerminalService.deleteByTaskId(taskId);
//            broadcastV2TaskTerminalService.removeById(taskId);
            broadcastV2TaskTerminalService.saveTaskTerminal(taskTerminalEntityList);
            broadcastV2TaskFileService.saveBatch(taskFileEntityList);
//            broadcastV2TaskTerminalService.insertTaskTerminal(taskTerminalEntityList);
//            broadcastV2TaskTerminalService.saveBatch(taskTerminalEntityList);

            return true;
            flag = true;
        } else {
            return false;
            flag = false;
        }

        /**
         * IP音柱任务添加日志记录开始
         */
        String content = "{任务ID：" + taskId
                + "，任务名：" + taskName
                + "，设备ID集合：" + missionparam.getTerminalIds()
                + " }";
        StoreOperationRecordsUtils.storeOperationData(missionparam.getTerminalIds(), null, "IP音柱任务添加", content);
        /**
         * IP音柱任务添加日志记录结束
         */


        return flag;
    }

    public Integer updateMission(Integer taskId, IpVolumeMissionParam missionparam) {

        String taskName = missionparam.getMissionName();

        if(null == taskName || "".equals(taskName)) {
        if (null == taskName || "".equals(taskName)) {
            return 4;
        }

        if(missionparam.getFileIds().size()<=0){
        if (missionparam.getFileIds().size() <= 0) {
            return 5;
        }

        if(missionparam.getTerminalIds().size()<=0){
        if (missionparam.getTerminalIds().size() <= 0) {
            return 6;
        }

@@ -197,7 +214,7 @@
        List<TaskTerminalVO> taskTerminalVOList = new ArrayList<>();
        //  封装数据库中任务终端绑定
        List<BroadcastV2TaskTerminal> taskTerminalEntityList = new ArrayList<>();
        for(Integer id: new HashSet<>(missionparam.getTerminalIds())) {
        for (Integer id : new HashSet<>(missionparam.getTerminalIds())) {
            TaskTerminalVO taskTerminalVO = new TaskTerminalVO();
            BroadcastV2TaskTerminal taskTerminalEntity = new BroadcastV2TaskTerminal();

@@ -212,22 +229,18 @@
        List<TaskMediaFileVO> taskMediaFileVOS = new ArrayList<>();
        List<BroadcastV2TaskFile> taskFileEntityList = new ArrayList<>();
        int i = 1;
        for(Integer fileId: new HashSet<>(missionparam.getFileIds())) {
        for (Integer fileId : new HashSet<>(missionparam.getFileIds())) {
            TaskMediaFileVO taskMediaFileVO = new TaskMediaFileVO();
            BroadcastV2TaskFile taskFileEntity = new BroadcastV2TaskFile();

            taskFileEntity.setFileId(fileId);

            IpVolumeFile fileEntity = ipVolumeFileService.getById(fileId);
            String absolutePath = MinIoConstant.getAbsolutePath(
                    MinIoConstant.BROADCAST_FILE.getBucketName() + fileEntity.getUserId(),
                    fileEntity.getFileName()
            );

            taskMediaFileVO.setI(i);
            i++;
            taskMediaFileVO.setT(0);
            taskMediaFileVO.setN(absolutePath);
            taskMediaFileVO.setN(fileEntity.getFileUrl());
            taskMediaFileVO.setS(BigInteger.valueOf(fileEntity.getOriginSize()));

            taskMediaFileVOS.add(taskMediaFileVO);
@@ -237,17 +250,16 @@

        missionparam.setFiles(taskMediaFileVOS.toArray(new TaskMediaFileVO[taskMediaFileVOS.size()]));
        missionparam.setTerminals(taskTerminalVOList.toArray(new TaskTerminalVO[taskTerminalVOList.size()]));
        System.out.println("---------------------------");

        //  向IP音柱服务器更新任务
        Integer result = TaskAPIUtils.updateTask(taskId, missionparam);

        if(result==0){
        if (result == 0) {
            IpVolumeMission byId = getById(taskId);
            for (BroadcastV2TaskTerminal v2TaskTerminal: taskTerminalEntityList) {
            for (BroadcastV2TaskTerminal v2TaskTerminal : taskTerminalEntityList) {
                v2TaskTerminal.setTaskId(taskId);
            }
            for (BroadcastV2TaskFile v2TaskFile :taskFileEntityList) {
            for (BroadcastV2TaskFile v2TaskFile : taskFileEntityList) {
                v2TaskFile.setTaskId(taskId);
            }
            if (null == byId) {
@@ -265,7 +277,7 @@
                mission.setOffDate(missionparam.getOffDate());
                mission.setWeek(Integer.parseInt(missionparam.getWeek(), 2));//时间由二进制字符串转换为十进制
                mission.setMissionVolume(missionparam.getMissionVolume());
               // mission.setMissionFileUrl(missionparam.getFiles().toString());
                // mission.setMissionFileUrl(missionparam.getFiles().toString());
                mission.setNumberInPlay(missionparam.getTerminals().length);
                /*TaskTerminalVO[] terminals = missionparam.getTerminals();*/

@@ -280,7 +292,7 @@
                mission.setOffDate(missionparam.getOffDate());
                mission.setWeek(Integer.parseInt(missionparam.getWeek(), 2));//时间由二进制字符串转换为十进制
                mission.setMissionVolume(missionparam.getMissionVolume());
               // mission.setMissionFileUrl(missionparam.getFiles().toString());
                // mission.setMissionFileUrl(missionparam.getFiles().toString());
                mission.setNumberInPlay(missionparam.getTerminals().length);
                System.out.println(updateById(mission));
                System.out.println("----------------------------------------------------------");
@@ -289,107 +301,222 @@
            broadcastV2TaskTerminalService.deleteByTaskId(taskId);
            broadcastV2TaskTerminalService.saveBatch(taskTerminalEntityList);
            broadcastV2TaskFileService.saveBatch(taskFileEntityList);
            return 0;

        }else {
            return result;
        }
        /**
         * IP音柱任务修改 日志记录开始
         */
        String content = "{任务ID：" + taskId
                + "，任务名：" + taskName
                + "，设备ID集合：" + missionparam.getTerminalIds()
                + "，任务编辑结果：" + result
                + " }";
        StoreOperationRecordsUtils.storeOperationData(missionparam.getTerminalIds(), null, "IP音柱任务修改", content);
        /**
         * IP音柱任务修改 日志记录结束
         */
        return result;

    }

    /**
     * 删除任务
     *
     * @param taskId
     * @return0：成功   1：任务不处于空闲状态 2:服务器删除失败
     * @return0：成功 1：任务不处于空闲状态 2:服务器删除失败
     */
    public Integer deleteMission(Integer taskId) {
        Integer del = TaskAPIUtils.delTask(taskId);
        if(del==0){
        if (del == 0) {
            IpVolumeMission byId = getById(taskId);
            if (byId == null) {
               return -1;
                return -1;
            }
            removeById(taskId);
            broadcastV2TaskFileService.removeById(taskId);
            broadcastV2TaskTerminalService.removeById(taskId);
            return 0;
        }
        else {
            return del;
            broadcastV2TaskFileService.deleteByTaskId(taskId);
            broadcastV2TaskTerminalService.deleteByTaskId(taskId);
//            broadcastV2TaskTerminalService.removeById(taskId);
        }

        /**
         * IP音柱任务删除 日志记录开始
         */
        String content = "{删除的任务ID：" + taskId
                + "，任务删除结果：" + del
                + " }";
        StoreOperationRecordsUtils.storeOperationData(null, null, "IP音柱任务删除", content);
        /**
         * IP音柱任务删除 日志记录结束
         */


        return del;
    }

//    public List<IpVolumeMission> listMission(ReceiveParam receiveParam) {
//
//        LambdaQueryWrapper<IpVolumeMission> wrapper = Wrappers.lambdaQuery(IpVolumeMission.class).eq(IpVolumeMission::getClientId, SecurityUtils.getUserId()).like(IpVolumeMission::getMissionName, receiveParam.getKeyWord());
//
//        return list(wrapper);
//    }

    public List<IpVolumeMissionBo> missionList(BaseConditionVO baseConditionVO,String keyword) {
        PageHelper.startPage(baseConditionVO.getPageNo(),baseConditionVO.getPageSize());
        boolean r = clientService.findClientId();  //ture代表有上级客户  查询通过userID
    public List<IpVolumeMissionBo> missionList(BaseConditionVO baseConditionVO, String keyword, Integer order, Integer seq) {


        //排序字段
        String orderByResult = "task_id";
        //正序、倒叙
        String orderBySeq = "ASC";
        if (order != null) {
            switch (order) {
                case 1:
                    orderByResult = OrderByEnums.IP_VOLUME_MISSION_CREATE_TIME.getCode();
                    break;
                default:
            }
        }
        if (seq != null) {
            switch (seq) {
                case 1:
                    orderBySeq = " ASC";
                    break;
                case 2:
                    orderBySeq = " DESC";
                    break;
                default:
                    break;
            }
        }
        //排序方式
        String orderBy = orderByResult + " " + orderBySeq;
        if (baseConditionVO != null) {
            PageHelper.startPage(baseConditionVO.getPageNo(), baseConditionVO.getPageSize(), orderBy);
        }

        List<IpVolumeMissionBo> ipVolumeMissionBos = new ArrayList<>();
        LambdaQueryWrapper<IpVolumeMission> eq=new LambdaQueryWrapper<>();
        LambdaQueryWrapper<IpVolumeMission> eq = Wrappers.lambdaQuery(IpVolumeMission.class);

//        if(r.equals(-1L)){
//            ipVolumeMissionBos = ipVolumeMissionMapper.listAllByClientId(r);
//        }else {
//            ipVolumeMissionBos = ipVolumeMissionMapper.listAllByUserId(SecurityUtils.getUserId());
//        }
        if(r){
           eq = Wrappers.lambdaQuery(IpVolumeMission.class).eq(IpVolumeMission::getUserId, SecurityUtils.getUserId());
        }else {
           eq = Wrappers.lambdaQuery(IpVolumeMission.class).eq(IpVolumeMission::getClientId,SecurityUtils.getUserId());

        //不是超管
        if (SecurityUtils.getClientId() != null) {
            eq = eq.eq(IpVolumeMission::getUserId, SecurityUtils.getUserId()).or(w -> {
                w.eq(IpVolumeMission::getClientId, SecurityUtils.getUserId());
            });
        }
        if(!keyword.isEmpty()){
            eq.like(IpVolumeMission::getMissionName,keyword);
        if (keyword != null && !keyword.isEmpty()) {
            eq.like(IpVolumeMission::getMissionName, keyword);
        }
        List<IpVolumeMission> list = list(eq);


        //  ipVolumeMissionBos = ipVolumeMissionMapper.listAllByClientId();

        for (IpVolumeMission mission: list) {
        for (IpVolumeMission mission : list) {
            List<String> fileName = broadcastV2TaskFileService.findByTaskid(mission.getTaskId());
            IpVolumeMissionBo bo=new IpVolumeMissionBo();
            IpVolumeMissionBo bo = new IpVolumeMissionBo();
            BeanUtils.copyProperties(mission, bo);
            bo.setFileNames(fileName);
            ipVolumeMissionBos.add(bo);

        }

//        for (IpVolumeMissionBo missionBo: ipVolumeMissionBos) {
//            missionBo.setFileNames(broadcastV2TaskFileService.findByTaskid(missionBo.getTaskId()));
//
//        }

        return ipVolumeMissionBos;
        //return new ArrayList<>();
    }

    public IpVolumeMission getByTaskId(Integer taskId){
    public TaskVO getByTaskId(Integer taskId) {
        TaskVO task = TaskAPIUtils.getTask(taskId);
        if(task==null){
        if (task == null) {
            throw new BusinessException("服务器未找到该任务");
        }

        IpVolumeMission byId = getById(taskId);
        if(byId==null){
        if (byId == null) {
            throw new BusinessException("本地未找到该任务");
        }
        if (SecurityUtils.getClientId() != null) {
            if (clientService.findClientId()) {
                if (!byId.getUserId().equals(SecurityUtils.getUserId())) {
                    throw new BusinessException("任务归属错误");
                }

        byId.setWeeks("0"+Integer.toBinaryString(byId.getWeek()));
            } else {
                if (!byId.getClientId().equals(SecurityUtils.getUserId()) && !byId.getUserId().equals(SecurityUtils.getUserId())) {
                    throw new BusinessException("任务归属错误");
                }
            }

        return byId;

        }
        task.setWeeks("0" + Integer.toBinaryString(task.getWeek()));
//        byId.setWeeks("0" + Integer.toBinaryString(byId.getWeek()));

        List<BroadcastV2TaskFile> taskFiles = broadcastV2TaskFileService.list(Wrappers.lambdaQuery(BroadcastV2TaskFile.class)
                .eq(BroadcastV2TaskFile::getTaskId, taskId));
        List<BroadcastV2TaskTerminal> taskTerminals = broadcastV2TaskTerminalService.list(Wrappers.lambdaQuery(BroadcastV2TaskTerminal.class)
                .eq(BroadcastV2TaskTerminal::getTaskId, taskId));


        if (!taskFiles.isEmpty()) {
            taskFiles.forEach(
                    fileId -> {
//                           IpVolumeFile volumeFile = ipVolumeFileService.getById(fileId.getFileId());
//                           if(volumeFile==null){
//                               throw new BusinessException("任务文件不存在");
//                           }
                        task.getFileIds().add(fileId.getFileId());
//                           byId.getFileIds().add(fileId.getFileId());
                    }
            );

        }
        if (!taskTerminals.isEmpty()) {
            taskTerminals.forEach(
                    terminalId -> {
//                        BroadcastTerminalV2Entity v2Entity = ipVolumeService.getById(terminalId.getTerminalId());
//                        if(v2Entity==null){
//                            throw new BusinessException("该任务暂无设备使用");
//                        }
                        task.getTerminalIds().add(terminalId.getTerminalId());
//                        byId.getTerminalIds().add(terminalId.getTerminalId());
                    }
            );
        }


        System.out.println(JSON.toJSONString(task));
        System.out.println(task);
        return task;


    }

    /**
     * 推送
     *
     * @param taskId
     * @return
     */
    public boolean playTask(Integer taskId) {
        return TaskAPIUtils.playTask(taskId);
    }


    /**
     * 停止
     *
     * @param taskId
     * @return
     */
    public boolean stopTask(Integer taskId) {
        return TaskAPIUtils.stopTask(taskId);
    }

    /**
     * 暂停
     *
     * @param taskId
     * @return
     */
    public boolean pauseTask(Integer taskId) {
        return TaskAPIUtils.pauseTask(taskId);
    }

    /**
     * 继续
     *
     * @param taskId
     * @return
     */
    public boolean continueTask(Integer taskId) {
        return TaskAPIUtils.continueTask(taskId);
    }
}