gb28181版本升级-补充

zhongrj

2023-06-27 2e98b20bea4463e4465e3c19059d0744a09aec06

 src/main/java/com/genersoft/iot/vmp/gb28181/session/VideoStreamSessionManager.java

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

import javax.sip.ClientTransaction;
import javax.sip.Dialog;

import com.genersoft.iot.vmp.common.VideoManagerConstants;
import com.genersoft.iot.vmp.conf.UserSetup;
import com.genersoft.iot.vmp.conf.UserSetting;
import com.genersoft.iot.vmp.gb28181.bean.SipTransactionInfo;
import com.genersoft.iot.vmp.gb28181.bean.SsrcTransaction;
import com.genersoft.iot.vmp.utils.SerializeUtils;
import com.genersoft.iot.vmp.utils.redis.RedisUtil;
import gov.nist.javax.sip.stack.SIPDialog;
import gov.nist.javax.sip.message.SIPResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.util.ObjectUtils;

/**    
 * @description:视频流session管理器，管理视频预览、预览回放的通信句柄 
@@ -25,10 +22,13 @@
public class VideoStreamSessionManager {

   @Autowired
   private RedisUtil redisUtil;
   private UserSetting userSetting;

   @Autowired
   private UserSetup userSetup;
   public enum SessionType {
      play,
      playback,
      download
   }

   /**
    * 添加一个点播/回放的事务信息
@@ -38,113 +38,104 @@
    * @param callId 一次请求的CallID
    * @param stream 流名称
    * @param mediaServerId 所使用的流媒体ID
    * @param transaction 事务
    * @param response 回复
    */
   public void put(String deviceId, String channelId, String callId, String stream, String ssrc, String mediaServerId, ClientTransaction transaction){
   public void put(String deviceId, String channelId, String callId, String stream, String ssrc, String mediaServerId, SIPResponse response, SessionType type){
      SsrcTransaction ssrcTransaction = new SsrcTransaction();
      ssrcTransaction.setDeviceId(deviceId);
      ssrcTransaction.setChannelId(channelId);
      ssrcTransaction.setStream(stream);
      byte[] transactionByteArray = SerializeUtils.serialize(transaction);
      ssrcTransaction.setTransaction(transactionByteArray);
      ssrcTransaction.setSipTransactionInfo(new SipTransactionInfo(response));
      ssrcTransaction.setCallId(callId);
      ssrcTransaction.setSsrc(ssrc);
      ssrcTransaction.setMediaServerId(mediaServerId);
      ssrcTransaction.setType(type);

      redisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId()
      RedisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId()
            + "_" +  deviceId + "_" + channelId + "_" + callId + "_" + stream, ssrcTransaction);
      redisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId()
      RedisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId()
            + "_" +  deviceId + "_" + channelId + "_" + callId + "_" + stream, ssrcTransaction);
   }

   public void put(String deviceId, String channelId, String callId, Dialog dialog){
      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, callId, null);
      if (ssrcTransaction != null) {
         byte[] dialogByteArray = SerializeUtils.serialize(dialog);
         ssrcTransaction.setDialog(dialogByteArray);
      }
      redisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId()
            + "_" +  deviceId + "_" + channelId + "_" + ssrcTransaction.getCallId() + "_"
            + ssrcTransaction.getStream(), ssrcTransaction);
   }

	
   public ClientTransaction getTransactionByStream(String deviceId, String channelId, String stream){
      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream);
      if (ssrcTransaction == null) return null;
      byte[] transactionByteArray = ssrcTransaction.getTransaction();
      ClientTransaction clientTransaction = (ClientTransaction)SerializeUtils.deSerialize(transactionByteArray);
      return clientTransaction;
   }

   public SIPDialog getDialogByStream(String deviceId, String channelId, String stream){
      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream);
      if (ssrcTransaction == null) return null;
      byte[] dialogByteArray = ssrcTransaction.getDialog();
      if (dialogByteArray == null) return null;
      SIPDialog dialog = (SIPDialog)SerializeUtils.deSerialize(dialogByteArray);
      return dialog;
   }

   public SIPDialog getDialogByCallId(String deviceId, String channelId, String callID){
      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, callID, null);
      if (ssrcTransaction == null) return null;
      byte[] dialogByteArray = ssrcTransaction.getDialog();
      if (dialogByteArray == null) return null;
      SIPDialog dialog = (SIPDialog)SerializeUtils.deSerialize(dialogByteArray);
      return dialog;
   }

   public SsrcTransaction getSsrcTransaction(String deviceId, String channelId, String callId, String stream){
      if (StringUtils.isEmpty(callId)) callId ="*";
      if (StringUtils.isEmpty(stream)) stream ="*";
      String key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId() + "_" + deviceId + "_" + channelId + "_" + callId+ "_" + stream;
      List<Object> scanResult = redisUtil.scan(key);
      if (scanResult.size() == 0) return null;
      return (SsrcTransaction)redisUtil.get((String) scanResult.get(0));

      if (ObjectUtils.isEmpty(deviceId)) {
         deviceId ="*";
      }
      if (ObjectUtils.isEmpty(channelId)) {
         channelId ="*";
      }
      if (ObjectUtils.isEmpty(callId)) {
         callId ="*";
      }
      if (ObjectUtils.isEmpty(stream)) {
         stream ="*";
      }
      String key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_" + deviceId + "_" + channelId + "_" + callId+ "_" + stream;
      List<Object> scanResult = RedisUtil.scan(key);
      if (scanResult.size() == 0) {
         return null;
      }
      return (SsrcTransaction)RedisUtil.get((String) scanResult.get(0));
   }

   public List<SsrcTransaction> getSsrcTransactionForAll(String deviceId, String channelId, String callId, String stream){
      if (StringUtils.isEmpty(deviceId)) deviceId ="*";
      if (StringUtils.isEmpty(channelId)) channelId ="*";
      if (StringUtils.isEmpty(callId)) callId ="*";
      if (StringUtils.isEmpty(stream)) stream ="*";
      String key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId() + "_" + deviceId + "_" + channelId + "_" + callId+ "_" + stream;
      List<Object> scanResult = redisUtil.scan(key);
      if (scanResult.size() == 0) return null;
      if (ObjectUtils.isEmpty(deviceId)) {
         deviceId ="*";
      }
      if (ObjectUtils.isEmpty(channelId)) {
         channelId ="*";
      }
      if (ObjectUtils.isEmpty(callId)) {
         callId ="*";
      }
      if (ObjectUtils.isEmpty(stream)) {
         stream ="*";
      }
      String key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_" + deviceId + "_" + channelId + "_" + callId+ "_" + stream;
      List<Object> scanResult = RedisUtil.scan(key);
      if (scanResult.size() == 0) {
         return null;
      }
      List<SsrcTransaction> result = new ArrayList<>();
      for (Object keyObj : scanResult) {
         result.add((SsrcTransaction)redisUtil.get((String) keyObj));
         result.add((SsrcTransaction)RedisUtil.get((String) keyObj));
      }
      return result;
   }

   public String getMediaServerId(String deviceId, String channelId, String stream){
      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream);
      if (ssrcTransaction == null) return null;
      if (ssrcTransaction == null) {
         return null;
      }
      return ssrcTransaction.getMediaServerId();
   }

   public String getSSRC(String deviceId, String channelId, String stream){
      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream);
      if (ssrcTransaction == null) return null;
      if (ssrcTransaction == null) {
         return null;
      }
      return ssrcTransaction.getSsrc();
   }
   
   public void remove(String deviceId, String channelId, String stream) {
      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream);
      if (ssrcTransaction == null) return;
      redisUtil.del(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId() + "_"
      if (ssrcTransaction == null) {
         return;
      }
      RedisUtil.del(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_"
            +  deviceId + "_" + channelId + "_" + ssrcTransaction.getCallId() + "_" + ssrcTransaction.getStream());
   }


   public List<SsrcTransaction> getAllSsrc() {
      List<Object> ssrcTransactionKeys = redisUtil.scan(String.format("%s_*_*_*_*", VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX+ userSetup.getServerId() + "_" ));
      List<Object> ssrcTransactionKeys = RedisUtil.scan(String.format("%s_*_*_*_*", VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX+ userSetting.getServerId()));
      List<SsrcTransaction> result= new ArrayList<>();
      for (int i = 0; i < ssrcTransactionKeys.size(); i++) {
         String key = (String)ssrcTransactionKeys.get(i);
         SsrcTransaction ssrcTransaction = (SsrcTransaction)redisUtil.get(key);
         SsrcTransaction ssrcTransaction = (SsrcTransaction)RedisUtil.get(key);
         result.add(ssrcTransaction);
      }
      return result;