优化sql

648540858

2023-10-25 cccfdace5b0af0c38627dbdce84b9c650b6c2082

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

old mode 100644
new mode 100755
@@ -1,159 +1,166 @@
package com.genersoft.iot.vmp.gb28181.session;



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

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 org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import org.springframework.util.StringUtils;



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

 * @date:   2020年5月13日 下午4:03:02     
 */

@Component

public class VideoStreamSessionManager {



   @Autowired

   private RedisUtil redisUtil;



   @Autowired

   private UserSetting userSetting;



   public enum SessionType {

      play,

      playback,

      download

   }



   /**

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

    * 后续可以通过流Id/callID

    * @param deviceId 设备ID

    * @param channelId 通道ID

    * @param callId 一次请求的CallID

    * @param stream 流名称

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

    * @param transaction 事务

    */

   public void put(String deviceId, String channelId, String callId, String stream, String ssrc, String mediaServerId, ClientTransaction transaction, SessionType type){

      SsrcTransaction ssrcTransaction = new SsrcTransaction();

      ssrcTransaction.setDeviceId(deviceId);

      ssrcTransaction.setChannelId(channelId);

      ssrcTransaction.setStream(stream);

      byte[] transactionByteArray = SerializeUtils.serialize(transaction);

      ssrcTransaction.setTransaction(transactionByteArray);

      ssrcTransaction.setCallId(callId);

      ssrcTransaction.setSsrc(ssrc);

      ssrcTransaction.setMediaServerId(mediaServerId);

      ssrcTransaction.setType(type);



      redisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId()

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

      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 + userSetting.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 + 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 + 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));

      }

      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;

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

         result.add(ssrcTransaction);

      }

      return result;

   }

}

package com.genersoft.iot.vmp.gb28181.session;

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;

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

/**
 * 视频流session管理器，管理视频预览、预览回放的通信句柄
 */
@Component
public class VideoStreamSessionManager {

   @Autowired
   private UserSetting userSetting;

   @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 SsrcTransaction getSsrcTransaction(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;
      }
      return (SsrcTransaction)redisTemplate.opsForValue().get(scanResult.get(0));
   }

   public SsrcTransaction getSsrcTransactionByCallId(String callId){

      if (ObjectUtils.isEmpty(callId)) {
         return null;
      }
      String key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_*_*_" + callId+ "_*";
      List<Object> scanResult = RedisUtil.scan(redisTemplate, key);
      if (!scanResult.isEmpty()) {
         return (SsrcTransaction)redisTemplate.opsForValue().get(scanResult.get(0));
      }else {
         key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_*_*_play_*";
         scanResult = RedisUtil.scan(redisTemplate, key);
         if (scanResult.isEmpty()) {
            return null;
         }
         for (Object keyObj : scanResult) {
            SsrcTransaction ssrcTransaction = (SsrcTransaction)redisTemplate.opsForValue().get(keyObj);
            if (ssrcTransaction.getSipTransactionInfo() != null &&
                  ssrcTransaction.getSipTransactionInfo().getCallId().equals(callId)) {
               return ssrcTransaction;
            }
         }
         return null;
      }

   }

   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;
   }
}