content: package com.genersoft.iot.vmp.service.impl;
  
  import com.alibaba.fastjson2.JSONArray;
  import com.alibaba.fastjson2.JSONObject;
  import com.genersoft.iot.vmp.common.StreamInfo;
  import com.genersoft.iot.vmp.conf.UserSetting;
  import com.genersoft.iot.vmp.conf.exception.ControllerException;
  import com.genersoft.iot.vmp.media.bean.MediaServer;
  import com.genersoft.iot.vmp.media.bean.RecordInfo;
  import com.genersoft.iot.vmp.media.event.media.MediaRecordMp4Event;
  import com.genersoft.iot.vmp.media.service.IMediaServerService;
  import com.genersoft.iot.vmp.media.zlm.AssistRESTfulUtils;
  import com.genersoft.iot.vmp.media.zlm.dto.StreamAuthorityInfo;
  import com.genersoft.iot.vmp.service.ICloudRecordService;
  import com.genersoft.iot.vmp.service.bean.CloudRecordItem;
  import com.genersoft.iot.vmp.service.bean.DownloadFileInfo;
  import com.genersoft.iot.vmp.service.bean.ErrorCallback;
  import com.genersoft.iot.vmp.service.redisMsg.IRedisRpcPlayService;
  import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
  import com.genersoft.iot.vmp.storager.dao.CloudRecordServiceMapper;
  import com.genersoft.iot.vmp.utils.DateUtil;
  import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
  import com.github.pagehelper.PageHelper;
  import com.github.pagehelper.PageInfo;
  import lombok.extern.slf4j.Slf4j;
  import org.apache.commons.lang3.ObjectUtils;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.context.event.EventListener;
  import org.springframework.scheduling.annotation.Async;
  import org.springframework.stereotype.Service;
  import org.springframework.util.Assert;
  
  import java.io.File;
  import java.time.LocalDate;
  import java.time.ZoneOffset;
  import java.util.ArrayList;
  import java.util.HashSet;
  import java.util.List;
  import java.util.Set;
  
  @Slf4j
  @Service
  public class CloudRecordServiceImpl implements ICloudRecordService {
  
      @Autowired
      private CloudRecordServiceMapper cloudRecordServiceMapper;
  
      @Autowired
      private IMediaServerService mediaServerService;
  
      @Autowired
      private IRedisCatchStorage redisCatchStorage;
  
      @Autowired
      private AssistRESTfulUtils assistRESTfulUtils;
  
      @Autowired
      private UserSetting userSetting;
  
      @Autowired
      private IRedisRpcPlayService redisRpcPlayService;
  
      @Override
      public PageInfo<CloudRecordItem> getList(int page, int count, String query, String app, String stream, String startTime,
                                               String endTime, List<MediaServer> mediaServerItems, String callId, Boolean ascOrder) {
          // 开始时间和结束时间在数据库中都是以秒为单位的
          Long startTimeStamp = null;
          Long endTimeStamp = null;
          if (startTime != null ) {
              if (!DateUtil.verification(startTime, DateUtil.formatter)) {
                  throw new ControllerException(ErrorCode.ERROR100.getCode(), "开始时间格式错误，正确格式为： " + DateUtil.formatter);
              }
              startTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestampMs(startTime);
  
          }
          if (endTime != null ) {
              if (!DateUtil.verification(endTime, DateUtil.formatter)) {
                  throw new ControllerException(ErrorCode.ERROR100.getCode(), "结束时间格式错误，正确格式为： " + DateUtil.formatter);
              }
              endTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestampMs(endTime);
  
          }
          PageHelper.startPage(page, count);
          if (query != null) {
              query = query.replaceAll("/", "//")
                      .replaceAll("%", "/%")
                      .replaceAll("_", "/_");
          }
          List<CloudRecordItem> all = cloudRecordServiceMapper.getList(query, app, stream, startTimeStamp, endTimeStamp,
                  callId, mediaServerItems, null, ascOrder);
          return new PageInfo<>(all);
      }
  
      @Override
      public List<String> getDateList(String app, String stream, int year, int month, List<MediaServer> mediaServerItems) {
          LocalDate startDate = LocalDate.of(year, month, 1);
          LocalDate endDate;
          if (month == 12) {
              endDate = LocalDate.of(year + 1, 1, 1);
          }else {
              endDate = LocalDate.of(year, month + 1, 1);
          }
          long startTimeStamp = startDate.atStartOfDay().toInstant(ZoneOffset.ofHours(8)).toEpochMilli();
          long endTimeStamp = endDate.atStartOfDay().toInstant(ZoneOffset.ofHours(8)).toEpochMilli();
          List<CloudRecordItem> cloudRecordItemList = cloudRecordServiceMapper.getList(null, app, stream, startTimeStamp,
                  endTimeStamp, null, mediaServerItems, null, null);
          if (cloudRecordItemList.isEmpty()) {
              return new ArrayList<>();
          }
          Set<String> resultSet = new HashSet<>();
          cloudRecordItemList.stream().forEach(cloudRecordItem -> {
              String date = DateUtil.timestampTo_yyyy_MM_dd(cloudRecordItem.getStartTime());
              resultSet.add(date);
          });
          return new ArrayList<>(resultSet);
      }
  
      @Async("taskExecutor")
      @EventListener
      public void onApplicationEvent(MediaRecordMp4Event event) {
          CloudRecordItem cloudRecordItem = CloudRecordItem.getInstance(event);
          cloudRecordItem.setServerId(userSetting.getServerId());
          if (ObjectUtils.isEmpty(cloudRecordItem.getCallId())) {
              StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(event.getApp(), event.getStream());
              if (streamAuthorityInfo != null) {
                  cloudRecordItem.setCallId(streamAuthorityInfo.getCallId());
              }
          }
          log.info("[添加录像记录] {}/{}, callId: {}, 内容：{}", event.getApp(), event.getStream(), cloudRecordItem.getCallId(), event.getRecordInfo());
          cloudRecordServiceMapper.add(cloudRecordItem);
      }
  
      @Override
      public String addTask(String app, String stream, MediaServer mediaServerItem, String startTime, String endTime,
                            String callId, String remoteHost, boolean filterMediaServer) {
          // 参数校验
          Assert.notNull(app,"应用名为NULL");
          Assert.notNull(stream,"流ID为NULL");
          if (mediaServerItem.getRecordAssistPort() == 0) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "为配置Assist服务");
          }
          Long startTimeStamp = null;
          Long endTimeStamp = null;
          if (startTime != null) {
              startTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(startTime);
          }
          if (endTime != null) {
              endTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(endTime);
          }
  
          List<MediaServer> mediaServers = new ArrayList<>();
          mediaServers.add(mediaServerItem);
          // 检索相关的录像文件
          List<String> filePathList = cloudRecordServiceMapper.queryRecordFilePathList(app, stream, startTimeStamp,
                  endTimeStamp, callId, filterMediaServer ? mediaServers : null);
          if (filePathList == null || filePathList.isEmpty()) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "未检索到视频文件");
          }
          JSONObject result =  assistRESTfulUtils.addTask(mediaServerItem, app, stream, startTime, endTime, callId, filePathList, remoteHost);
          if (result.getInteger("code") != 0) {
              throw new ControllerException(result.getInteger("code"), result.getString("msg"));
          }
          return result.getString("data");
      }
  
      @Override
      public JSONArray queryTask(String app, String stream, String callId, String taskId, String mediaServerId,
                                 Boolean isEnd, String scheme) {
          MediaServer mediaServerItem = null;
          if (mediaServerId == null) {
              mediaServerItem = mediaServerService.getDefaultMediaServer();
          }else {
              mediaServerItem = mediaServerService.getOne(mediaServerId);
          }
          if (mediaServerItem == null) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "未找到可用的流媒体");
          }
  
          JSONObject result =  assistRESTfulUtils.queryTaskList(mediaServerItem, app, stream, callId, taskId, isEnd, scheme);
          if (result == null || result.getInteger("code") != 0) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), result == null ? "查询任务列表失败" : result.getString("msg"));
          }
          return result.getJSONArray("data");
      }
  
      @Override
      public int changeCollect(boolean result, String app, String stream, String mediaServerId, String startTime, String endTime, String callId) {
          // 开始时间和结束时间在数据库中都是以秒为单位的
          Long startTimeStamp = null;
          Long endTimeStamp = null;
          if (startTime != null ) {
              if (!DateUtil.verification(startTime, DateUtil.formatter)) {
                  throw new ControllerException(ErrorCode.ERROR100.getCode(), "开始时间格式错误，正确格式为： " + DateUtil.formatter);
              }
              startTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(startTime);
  
          }
          if (endTime != null ) {
              if (!DateUtil.verification(endTime, DateUtil.formatter)) {
                  throw new ControllerException(ErrorCode.ERROR100.getCode(), "结束时间格式错误，正确格式为： " + DateUtil.formatter);
              }
              endTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(endTime);
  
          }
  
          List<MediaServer> mediaServerItems;
          if (!ObjectUtils.isEmpty(mediaServerId)) {
              mediaServerItems = new ArrayList<>();
              MediaServer mediaServerItem = mediaServerService.getOne(mediaServerId);
              if (mediaServerItem == null) {
                  throw new ControllerException(ErrorCode.ERROR100.getCode(), "未找到流媒体: " + mediaServerId);
              }
              mediaServerItems.add(mediaServerItem);
          } else {
              mediaServerItems = null;
          }
  
          List<CloudRecordItem> all = cloudRecordServiceMapper.getList(null, app, stream, startTimeStamp, endTimeStamp,
                  callId, mediaServerItems, null, null);
          if (all.isEmpty()) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "未找到待收藏的视频");
          }
          int limitCount = 50;
          int resultCount = 0;
          if (all.size() > limitCount) {
              for (int i = 0; i < all.size(); i += limitCount) {
                  int toIndex = i + limitCount;
                  if (i + limitCount > all.size()) {
                      toIndex = all.size();
                  }
                  resultCount += cloudRecordServiceMapper.updateCollectList(result, all.subList(i, toIndex));
  
              }
          }else {
              resultCount = cloudRecordServiceMapper.updateCollectList(result, all);
          }
          return resultCount;
      }
  
      @Override
      public int changeCollectById(Integer recordId, boolean result) {
         return cloudRecordServiceMapper.changeCollectById(result, recordId);
      }
  
      @Override
      public DownloadFileInfo getPlayUrlPath(Integer recordId) {
          CloudRecordItem recordItem = cloudRecordServiceMapper.queryOne(recordId);
          if (recordItem == null) {
              throw new ControllerException(ErrorCode.ERROR400.getCode(), "资源不存在");
          }
          if (!userSetting.getServerId().equals(recordItem.getServerId())) {
              return redisRpcPlayService.getRecordPlayUrl(recordItem.getServerId(), recordId);
          }
  
          MediaServer mediaServer = mediaServerService.getOne(recordItem.getMediaServerId());
  
          return mediaServerService.getDownloadFilePath(mediaServer, RecordInfo.getInstance(recordItem));
      }
  
      @Override
      public List<CloudRecordItem> getAllList(String query, String app, String stream, String startTime, String endTime, List<MediaServer> mediaServerItems, String callId, List<Integer> ids) {
          // 开始时间和结束时间在数据库中都是以秒为单位的
          Long startTimeStamp = null;
          Long endTimeStamp = null;
          if (startTime != null ) {
              if (!DateUtil.verification(startTime, DateUtil.formatter)) {
                  throw new ControllerException(ErrorCode.ERROR100.getCode(), "开始时间格式错误，正确格式为： " + DateUtil.formatter);
              }
              startTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestampMs(startTime);
  
          }
          if (endTime != null ) {
              if (!DateUtil.verification(endTime, DateUtil.formatter)) {
                  throw new ControllerException(ErrorCode.ERROR100.getCode(), "结束时间格式错误，正确格式为： " + DateUtil.formatter);
              }
              endTimeStamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestampMs(endTime);
  
          }
          return cloudRecordServiceMapper.getList(query, app, stream, startTimeStamp, endTimeStamp,
                  callId, mediaServerItems, ids, null);
      }
  
      @Override
      public void loadMP4File(String app, String stream, int cloudRecordId, ErrorCallback<StreamInfo> callback) {
  
          CloudRecordItem recordItem = cloudRecordServiceMapper.queryOne(cloudRecordId);
          if (recordItem == null) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "无录像");
          }
          String mediaServerId = recordItem.getMediaServerId();
          MediaServer mediaServer = mediaServerService.getOne(mediaServerId);
          if (mediaServer == null) {
              log.warn("[云端录像] 播放 未找到录制的流媒体，将自动选择低负载流媒体使用");
              mediaServer = mediaServerService.getMediaServerForMinimumLoad(null);
          }
          if (mediaServer == null) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "无可用流媒体");
          }
          String fileName = recordItem.getFileName().substring(0 , recordItem.getFileName().indexOf("."));
          String filePath = recordItem.getFilePath();
  //        if (filePath != null) {
  //            fileName = filePath.substring(0, filePath.lastIndexOf("/"));
  //        }
          mediaServerService.loadMP4File(mediaServer, app, stream, filePath, fileName, ((code, msg, streamInfo) -> {
              if (code == ErrorCode.SUCCESS.getCode()) {
                  streamInfo.setDuration(recordItem.getTimeLen());
              }
              callback.run(code, msg, streamInfo);
          }));
      }
  
      @Override
      public void loadMP4FileForDate(String app, String stream, String date, ErrorCallback<StreamInfo> callback) {
          long startTimestamp = DateUtil.yyyy_MM_dd_HH_mm_ssToTimestampMs(date + " 00:00:00");
          long endTimestamp = startTimestamp + 24 * 60 * 60 * 1000;
  
          List<CloudRecordItem> recordItemList = cloudRecordServiceMapper.getList(null, app, stream, startTimestamp, endTimestamp, null, null, null, false);
          if (recordItemList.isEmpty()) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "此时间无录像");
          }
          String mediaServerId = recordItemList.get(0).getMediaServerId();
          MediaServer mediaServer = mediaServerService.getOne(mediaServerId);
          if (mediaServer == null) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "媒体节点不存在： " + mediaServerId);
          }
          String dateDir = null;
          String filePath = recordItemList.get(0).getFilePath();
          if (filePath != null) {
              dateDir = filePath.substring(0, filePath.lastIndexOf("/"));
          }
          mediaServerService.loadMP4FileForDate(mediaServer, app, stream, date, dateDir, callback);
  
      }
  
      @Override
      public void seekRecord(String mediaServerId,String app, String stream, Double seek, String schema) {
          MediaServer mediaServer = mediaServerService.getOne(mediaServerId);
          if (mediaServer == null) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "媒体节点不存在： " + mediaServerId);
          }
          mediaServerService.seekRecordStamp(mediaServer, app, stream, seek, schema);
      }
  
      @Override
      public void setRecordSpeed(String mediaServerId, String app, String stream, Integer speed, String schema) {
          MediaServer mediaServer = mediaServerService.getOne(mediaServerId);
          if (mediaServer == null) {
              throw new ControllerException(ErrorCode.ERROR100.getCode(), "媒体节点不存在： " + mediaServerId);
          }
          mediaServerService.setRecordSpeed(mediaServer, app, stream, speed, schema);
      }
  
      @Override
      public void deleteFileByIds(Set<Integer> ids) {
          log.info("[删除录像文件] ids: {}", ids.toArray());
          List<CloudRecordItem> cloudRecordItemList = cloudRecordServiceMapper.queryRecordByIds(ids);
          if (cloudRecordItemList.isEmpty()) {
              return;
          }
          List<CloudRecordItem> cloudRecordItemIdListForDelete = new ArrayList<>();
          StringBuilder stringBuilder = new StringBuilder();
          for (CloudRecordItem cloudRecordItem : cloudRecordItemList) {
              String date = new File(cloudRecordItem.getFilePath()).getParentFile().getName();
              MediaServer mediaServer = mediaServerService.getOne(cloudRecordItem.getMediaServerId());
              try {
                  boolean deleteResult = mediaServerService.deleteRecordDirectory(mediaServer, cloudRecordItem.getApp(),
                          cloudRecordItem.getStream(), date, cloudRecordItem.getFileName());
                  if (deleteResult) {
                      log.warn("[录像文件] 删除磁盘文件成功： {}", cloudRecordItem.getFilePath());
                      cloudRecordItemIdListForDelete.add(cloudRecordItem);
                  }
              }catch (ControllerException e) {
                  if (stringBuilder.length() > 0) {
                      stringBuilder.append(", ");
                  }
                  stringBuilder.append(cloudRecordItem.getFileName());
              }
  
          }
          if (!cloudRecordItemIdListForDelete.isEmpty()) {
              cloudRecordServiceMapper.deleteList(cloudRecordItemIdListForDelete);
          }
          if (stringBuilder.length() > 0) {
              stringBuilder.append(" 删除失败");
              throw new ControllerException(ErrorCode.ERROR100.getCode(), stringBuilder.toString());
          }
      }
  }
  