修复请求报错404等情况下导致ResponseBody未关闭的问题

dsk

2023-06-25 cec266d3cd27e79e61b7e614d0c85e0b43492115

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

@@ -1,51 +1,139 @@
package com.genersoft.iot.vmp.gb28181.session;



import java.util.concurrent.ConcurrentHashMap;



import javax.sip.ClientTransaction;



import com.genersoft.iot.vmp.common.InviteSessionType;

import com.genersoft.iot.vmp.common.VideoManagerConstants;

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.JsonUtil;

import com.genersoft.iot.vmp.utils.redis.RedisUtil;

import gov.nist.javax.sip.message.SIPResponse;

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

import org.springframework.data.redis.core.RedisTemplate;

import org.springframework.stereotype.Component;

import org.springframework.util.ObjectUtils;



/**    
 * @Description:视频流session管理器，管理视频预览、预览回放的通信句柄 
 * @author: swwheihei

 * @date:   2020年5月13日 下午4:03:02     
import java.util.ArrayList;

import java.util.List;



/**

 * 视频流session管理器，管理视频预览、预览回放的通信句柄

 */

@Component

public class VideoStreamSessionManager {



   private ConcurrentHashMap<String, ClientTransaction> sessionMap = new ConcurrentHashMap<>();

   private ConcurrentHashMap<String, String> ssrcMap = new ConcurrentHashMap<>();

   private ConcurrentHashMap<String, String> streamIdMap = new ConcurrentHashMap<>();

   @Autowired

   private UserSetting userSetting;



   public String createPlaySsrc(){

      return SsrcUtil.getPlaySsrc();

   }

	
   public String createPlayBackSsrc(){

      return SsrcUtil.getPlayBackSsrc();

   }

	
   public void put(String deviceId, String channelId ,String ssrc, String streamId, ClientTransaction transaction){

      sessionMap.put(deviceId + "_" + channelId, transaction);

      ssrcMap.put(deviceId + "_" + channelId, ssrc);

      streamIdMap.put(deviceId + "_" + channelId, streamId);

   }

	
   public ClientTransaction getTransaction(String deviceId, String channelId){

      return sessionMap.get(deviceId + "_" + channelId);

   @Autowired

   private RedisTemplate<Object, Object> redisTemplate;



   /**

    * 添加一个点播/回放的事务信息

    * 后续可以通过流Id/callID

    * @param deviceId 设备ID

    * @param channelId 通道ID

    * @param callId 一次请求的CallID

    * @param stream 流名称

    * @param mediaServerId 所使用的流媒体ID

    * @param response 回复

    */

   public void put(String deviceId, String channelId, String callId, String stream, String ssrc, String mediaServerId, SIPResponse response, InviteSessionType type){

      SsrcTransaction ssrcTransaction = new SsrcTransaction();

      ssrcTransaction.setDeviceId(deviceId);

      ssrcTransaction.setChannelId(channelId);

      ssrcTransaction.setStream(stream);

      ssrcTransaction.setSipTransactionInfo(new SipTransactionInfo(response));

      ssrcTransaction.setCallId(callId);

      ssrcTransaction.setSsrc(ssrc);

      ssrcTransaction.setMediaServerId(mediaServerId);

      ssrcTransaction.setType(type);



      redisTemplate.opsForValue().set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId()

            + "_" +  deviceId + "_" + channelId + "_" + callId + "_" + stream, ssrcTransaction);

   }



   public String getStreamId(String deviceId, String channelId){

      return streamIdMap.get(deviceId + "_" + channelId);

   }

	
   public void remove(String deviceId, String channelId) {

      sessionMap.remove(deviceId + "_" + channelId);

      if (ssrcMap.get(deviceId + "_" + channelId) != null) {

         SsrcUtil.releaseSsrc(ssrcMap.get(deviceId + "_" + channelId));

   public SsrcTransaction getSsrcTransaction(String deviceId, String channelId, String callId, String stream){



      if (ObjectUtils.isEmpty(deviceId)) {

         deviceId ="*";

      }

      ssrcMap.remove(deviceId + "_" + channelId);

      streamIdMap.remove(deviceId + "_" + channelId);

      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(redisTemplate, key);

      if (scanResult.size() == 0) {

         return null;

      }

      return (SsrcTransaction)redisTemplate.opsForValue().get(scanResult.get(0));

   }



   public List<SsrcTransaction> getSsrcTransactionForAll(String deviceId, String channelId, String callId, String stream){

      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(redisTemplate, key);

      if (scanResult.size() == 0) {

         return null;

      }

      List<SsrcTransaction> result = new ArrayList<>();

      for (Object keyObj : scanResult) {

         result.add((SsrcTransaction)redisTemplate.opsForValue().get(keyObj));

      }

      return result;

   }



   public String getMediaServerId(String deviceId, String channelId, String stream){

      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream);

      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;

      }

      return ssrcTransaction.getSsrc();

   }

	
   public void remove(String deviceId, String channelId, String stream) {

      SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream);

      if (ssrcTransaction == null) {

         return;

      }

      redisTemplate.delete(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_"

            +  deviceId + "_" + channelId + "_" + ssrcTransaction.getCallId() + "_" + ssrcTransaction.getStream());

   }





   public List<SsrcTransaction> getAllSsrc() {

      List<Object> ssrcTransactionKeys = RedisUtil.scan(redisTemplate, String.format("%s_*_*_*_*", VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX+ userSetting.getServerId()));

      List<SsrcTransaction> result= new ArrayList<>();

      for (Object ssrcTransactionKey : ssrcTransactionKeys) {

         String key = (String) ssrcTransactionKey;

         SsrcTransaction ssrcTransaction = JsonUtil.redisJsonToObject(redisTemplate, key, SsrcTransaction.class);

         result.add(ssrcTransaction);

      }

      return result;

   }

}