From 30ae9e929fad80f624ab632c53081db3d2dc9aec Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期四, 25 五月 2023 17:28:57 +0800
Subject: [PATCH] 合并主线
---
src/main/java/com/genersoft/iot/vmp/service/impl/InviteStreamServiceImpl.java | 182 +++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 182 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/genersoft/iot/vmp/service/impl/InviteStreamServiceImpl.java b/src/main/java/com/genersoft/iot/vmp/service/impl/InviteStreamServiceImpl.java
new file mode 100644
index 0000000..f067cf0
--- /dev/null
+++ b/src/main/java/com/genersoft/iot/vmp/service/impl/InviteStreamServiceImpl.java
@@ -0,0 +1,182 @@
+package com.genersoft.iot.vmp.service.impl;
+
+import com.alibaba.fastjson2.JSON;
+import com.genersoft.iot.vmp.common.InviteInfo;
+import com.genersoft.iot.vmp.common.InviteSessionStatus;
+import com.genersoft.iot.vmp.common.InviteSessionType;
+import com.genersoft.iot.vmp.common.VideoManagerConstants;
+import com.genersoft.iot.vmp.service.IInviteStreamService;
+import com.genersoft.iot.vmp.service.bean.InviteErrorCallback;
+import com.genersoft.iot.vmp.utils.redis.RedisUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.CopyOnWriteArrayList;
+
+@Service
+public class InviteStreamServiceImpl implements IInviteStreamService {
+
+ private final Logger logger = LoggerFactory.getLogger(InviteStreamServiceImpl.class);
+
+ private final Map<String, List<InviteErrorCallback<Object>>> inviteErrorCallbackMap = new ConcurrentHashMap<>();
+
+ @Autowired
+ private RedisTemplate<Object, Object> redisTemplate;
+
+ @Override
+ public void updateInviteInfo(InviteInfo inviteInfo) {
+ if (inviteInfo == null || (inviteInfo.getDeviceId() == null || inviteInfo.getChannelId() == null)) {
+ logger.warn("[鏇存柊Invite淇℃伅]锛屽弬鏁颁笉鍏細 {}", JSON.toJSON(inviteInfo));
+ return;
+ }
+ InviteInfo inviteInfoForUpdate = null;
+
+ if (InviteSessionStatus.ready == inviteInfo.getStatus()) {
+ if (inviteInfo.getDeviceId() == null
+ || inviteInfo.getChannelId() == null
+ || inviteInfo.getType() == null
+ || inviteInfo.getStream() == null
+ ) {
+ return;
+ }
+ inviteInfoForUpdate = inviteInfo;
+ } else {
+ InviteInfo inviteInfoInRedis = getInviteInfo(inviteInfo.getType(), inviteInfo.getDeviceId(),
+ inviteInfo.getChannelId(), inviteInfo.getStream());
+ if (inviteInfoInRedis == null) {
+ logger.warn("[鏇存柊Invite淇℃伅]锛屾湭浠庣紦瀛樹腑璇诲彇鍒癐nvite淇℃伅锛� deviceId: {}, channel: {}, stream: {}",
+ inviteInfo.getDeviceId(), inviteInfo.getChannelId(), inviteInfo.getStream());
+ return;
+ }
+ if (inviteInfo.getStreamInfo() != null) {
+ inviteInfoInRedis.setStreamInfo(inviteInfo.getStreamInfo());
+ }
+ if (inviteInfo.getSsrcInfo() != null) {
+ inviteInfoInRedis.setSsrcInfo(inviteInfo.getSsrcInfo());
+ }
+ if (inviteInfo.getStreamMode() != null) {
+ inviteInfoInRedis.setStreamMode(inviteInfo.getStreamMode());
+ }
+ if (inviteInfo.getReceiveIp() != null) {
+ inviteInfoInRedis.setReceiveIp(inviteInfo.getReceiveIp());
+ }
+ if (inviteInfo.getReceivePort() != null) {
+ inviteInfoInRedis.setReceivePort(inviteInfo.getReceivePort());
+ }
+ if (inviteInfo.getStatus() != null) {
+ inviteInfoInRedis.setStatus(inviteInfo.getStatus());
+ }
+
+ inviteInfoForUpdate = inviteInfoInRedis;
+
+ }
+ String key = VideoManagerConstants.INVITE_PREFIX +
+ "_" + inviteInfoForUpdate.getType() +
+ "_" + inviteInfoForUpdate.getDeviceId() +
+ "_" + inviteInfoForUpdate.getChannelId() +
+ "_" + inviteInfoForUpdate.getStream();
+ redisTemplate.opsForValue().set(key, inviteInfoForUpdate);
+ }
+
+ @Override
+ public InviteInfo getInviteInfo(InviteSessionType type, String deviceId, String channelId, String stream) {
+ String key = VideoManagerConstants.INVITE_PREFIX +
+ "_" + (type != null ? type : "*") +
+ "_" + (deviceId != null ? deviceId : "*") +
+ "_" + (channelId != null ? channelId : "*") +
+ "_" + (stream != null ? stream : "*");
+ List<Object> scanResult = RedisUtil.scan(redisTemplate, key);
+ if (scanResult.size() != 1) {
+ return null;
+ }
+
+ return (InviteInfo) redisTemplate.opsForValue().get(scanResult.get(0));
+ }
+
+ @Override
+ public InviteInfo getInviteInfoByDeviceAndChannel(InviteSessionType type, String deviceId, String channelId) {
+ return getInviteInfo(type, deviceId, channelId, null);
+ }
+
+ @Override
+ public InviteInfo getInviteInfoByStream(InviteSessionType type, String stream) {
+ return getInviteInfo(type, null, null, stream);
+ }
+
+ @Override
+ public void removeInviteInfo(InviteSessionType type, String deviceId, String channelId, String stream) {
+ String scanKey = VideoManagerConstants.INVITE_PREFIX +
+ "_" + (type != null ? type : "*") +
+ "_" + (deviceId != null ? deviceId : "*") +
+ "_" + (channelId != null ? channelId : "*") +
+ "_" + (stream != null ? stream : "*");
+ List<Object> scanResult = RedisUtil.scan(redisTemplate, scanKey);
+ if (scanResult.size() > 0) {
+ for (Object keyObj : scanResult) {
+ String key = (String) keyObj;
+ InviteInfo inviteInfo = (InviteInfo) redisTemplate.opsForValue().get(key);
+ if (inviteInfo == null) {
+ continue;
+ }
+ redisTemplate.delete(key);
+ inviteErrorCallbackMap.remove(buildKey(type, deviceId, channelId, inviteInfo.getStream()));
+ }
+ }
+ }
+
+ @Override
+ public void removeInviteInfoByDeviceAndChannel(InviteSessionType inviteSessionType, String deviceId, String channelId) {
+ removeInviteInfo(inviteSessionType, deviceId, channelId, null);
+ }
+
+ @Override
+ public void removeInviteInfo(InviteInfo inviteInfo) {
+ removeInviteInfo(inviteInfo.getType(), inviteInfo.getDeviceId(), inviteInfo.getChannelId(), inviteInfo.getStream());
+ }
+
+ @Override
+ public void once(InviteSessionType type, String deviceId, String channelId, String stream, InviteErrorCallback<Object> callback) {
+ String key = buildKey(type, deviceId, channelId, stream);
+ List<InviteErrorCallback<Object>> callbacks = inviteErrorCallbackMap.get(key);
+ if (callbacks == null) {
+ callbacks = new CopyOnWriteArrayList<>();
+ inviteErrorCallbackMap.put(key, callbacks);
+ }
+ callbacks.add(callback);
+
+ }
+
+ @Override
+ public void call(InviteSessionType type, String deviceId, String channelId, String stream, int code, String msg, Object data) {
+ String key = buildKey(type, deviceId, channelId, stream);
+ List<InviteErrorCallback<Object>> callbacks = inviteErrorCallbackMap.get(key);
+ if (callbacks == null) {
+ return;
+ }
+ for (InviteErrorCallback<Object> callback : callbacks) {
+ callback.run(code, msg, data);
+ }
+ inviteErrorCallbackMap.remove(key);
+ }
+
+ private String buildKey(InviteSessionType type, String deviceId, String channelId, String stream) {
+ String key = type + "_" + deviceId + "_" + channelId;
+ // 濡傛灉ssrc鏈猲ull閭ｄ箞鍙互瀹炵幇涓�涓�氶亾鍙兘涓�娆℃搷浣滐紝ssrc涓嶄负null鍒欏彲浠ユ敮鎸佷竴涓�氶亾澶氭invite
+ if (stream != null) {
+ key += ("_" + stream);
+ }
+ return key;
+ }
+
+
+ @Override
+ public void clearInviteInfo(String deviceId) {
+ removeInviteInfo(null, deviceId, null, null);
+ }
+}
--
Gitblit v1.8.0