From 1dbb559b388831b5f15e690d0b33d24b1faaa983 Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期二, 25 一月 2022 15:43:52 +0800
Subject: [PATCH] 优化文件导入判断
---
src/main/java/com/genersoft/iot/vmp/gb28181/session/VideoStreamSessionManager.java | 113 +++++++++++++++++++++++++++++++++++++++++++++-----------
1 files changed, 91 insertions(+), 22 deletions(-)
diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/session/VideoStreamSessionManager.java b/src/main/java/com/genersoft/iot/vmp/gb28181/session/VideoStreamSessionManager.java
index f8b9350..d41b04d 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/session/VideoStreamSessionManager.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/session/VideoStreamSessionManager.java
@@ -1,41 +1,110 @@
package com.genersoft.iot.vmp.gb28181.session;
-import java.util.concurrent.ConcurrentHashMap;
+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.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;
/**
- * @Description:瑙嗛娴乻ession绠＄悊鍣紝绠＄悊瑙嗛棰勮銆侀瑙堝洖鏀剧殑閫氫俊鍙ユ焺
- * @author: songww
+ * @description:瑙嗛娴乻ession绠＄悊鍣紝绠＄悊瑙嗛棰勮銆侀瑙堝洖鏀剧殑閫氫俊鍙ユ焺
+ * @author: swwheihei
* @date: 2020骞�5鏈�13鏃� 涓嬪崍4:03:02
*/
@Component
public class VideoStreamSessionManager {
- private ConcurrentHashMap<String, ClientTransaction> sessionMap = new ConcurrentHashMap<>();
+ @Autowired
+ private RedisUtil redisUtil;
- public String createPlaySsrc(){
- String ssrc = SsrcUtil.getPlaySsrc();
- return ssrc;
+ @Autowired
+ private UserSetup userSetup;
+
+ public void put(String deviceId, String channelId ,String ssrc, String streamId, String mediaServerId, ClientTransaction transaction){
+ SsrcTransaction ssrcTransaction = new SsrcTransaction();
+ ssrcTransaction.setDeviceId(deviceId);
+ ssrcTransaction.setChannelId(channelId);
+ ssrcTransaction.setStreamId(streamId);
+ byte[] transactionByteArray = SerializeUtils.serialize(transaction);
+ ssrcTransaction.setTransaction(transactionByteArray);
+ ssrcTransaction.setSsrc(ssrc);
+ ssrcTransaction.setMediaServerId(mediaServerId);
+
+ redisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId() + "_" + deviceId + "_" + channelId, ssrcTransaction);
+ }
+
+ public void put(String deviceId, String channelId , Dialog dialog){
+ SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId);
+ if (ssrcTransaction != null) {
+ byte[] dialogByteArray = SerializeUtils.serialize(dialog);
+ ssrcTransaction.setDialog(dialogByteArray);
+ }
+ redisUtil.set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId() + "_" + deviceId + "_" + channelId, ssrcTransaction);
+ }
+
+
+ public ClientTransaction getTransaction(String deviceId, String channelId){
+ SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId);
+ if (ssrcTransaction == null) return null;
+ byte[] transactionByteArray = ssrcTransaction.getTransaction();
+ ClientTransaction clientTransaction = (ClientTransaction)SerializeUtils.deSerialize(transactionByteArray);
+ return clientTransaction;
+ }
+
+ public SIPDialog getDialog(String deviceId, String channelId){
+ SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId);
+ 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){
+ SsrcTransaction ssrcTransaction = (SsrcTransaction)redisUtil.get(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId() + "_" + deviceId + "_" + channelId);
+ return ssrcTransaction;
+ }
+
+ public String getStreamId(String deviceId, String channelId){
+ SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId);
+ if (ssrcTransaction == null) return null;
+ return ssrcTransaction.getStreamId();
+ }
+ public String getMediaServerId(String deviceId, String channelId){
+ SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId);
+ if (ssrcTransaction == null) return null;
+ return ssrcTransaction.getMediaServerId();
+ }
+
+ public String getSSRC(String deviceId, String channelId){
+ SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId);
+ if (ssrcTransaction == null) return null;
+ return ssrcTransaction.getSsrc();
}
- public String createPlayBackSsrc(){
- String ssrc = SsrcUtil.getPlayBackSsrc();
- return ssrc;
+ public void remove(String deviceId, String channelId) {
+ SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId);
+ if (ssrcTransaction == null) return;
+ redisUtil.del(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetup.getServerId() + "_" + deviceId + "_" + channelId);
}
-
- public void put(String ssrc,ClientTransaction transaction){
- sessionMap.put(ssrc, transaction);
- }
-
- public ClientTransaction get(String ssrc){
- return sessionMap.get(ssrc);
- }
-
- public void remove(String ssrc) {
- sessionMap.remove(ssrc);
- SsrcUtil.releaseSsrc(ssrc);
+
+ public List<SsrcTransaction> getAllSsrc() {
+ List<Object> ssrcTransactionKeys = redisUtil.scan(String.format("%s_*_*", VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX+ userSetup.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;
}
}
--
Gitblit v1.8.0