From e48fa711a3664bece9b3e58840a75fe7c05bc47c Mon Sep 17 00:00:00 2001
From: panlinlin <648540858@qq.com>
Date: 星期六, 08 五月 2021 17:14:05 +0800
Subject: [PATCH] 添加截图(快照)功能

---
 src/main/java/com/genersoft/iot/vmp/storager/impl/VideoManagerStoragerImpl.java |  107 ++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 97 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/storager/impl/VideoManagerStoragerImpl.java b/src/main/java/com/genersoft/iot/vmp/storager/impl/VideoManagerStoragerImpl.java
index b2756e2..cbaa404 100644
--- a/src/main/java/com/genersoft/iot/vmp/storager/impl/VideoManagerStoragerImpl.java
+++ b/src/main/java/com/genersoft/iot/vmp/storager/impl/VideoManagerStoragerImpl.java
@@ -2,13 +2,18 @@
 
 import java.util.*;
 
+import com.genersoft.iot.vmp.common.StreamInfo;
 import com.genersoft.iot.vmp.gb28181.bean.*;
+import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
 import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem;
+import com.genersoft.iot.vmp.media.zlm.dto.StreamPushItem;
 import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
 import com.genersoft.iot.vmp.storager.dao.*;
-import com.genersoft.iot.vmp.vmanager.platform.bean.ChannelReduce;
+import com.genersoft.iot.vmp.vmanager.gb28181.platform.bean.ChannelReduce;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.jdbc.datasource.DataSourceTransactionManager;
 import org.springframework.stereotype.Component;
@@ -26,6 +31,9 @@
 @SuppressWarnings("rawtypes")
 @Component
 public class VideoManagerStoragerImpl implements IVideoManagerStorager {
+
+	private Logger logger = LoggerFactory.getLogger(VideoManagerStoragerImpl.class);
+
 	@Autowired
 	DataSourceTransactionManager dataSourceTransactionManager;
 
@@ -54,7 +62,13 @@
     private StreamProxyMapper streamProxyMapper;
 
 	@Autowired
+    private StreamPushMapper streamPushMapper;
+
+	@Autowired
     private GbStreamMapper gbStreamMapper;
+
+	@Autowired
+    private VideoStreamSessionManager streamSession;
 
 
 	/**
@@ -102,6 +116,7 @@
 	public synchronized void updateChannel(String deviceId, DeviceChannel channel) {
 		String channelId = channel.getChannelId();
 		channel.setDeviceId(deviceId);
+		channel.setStreamId(streamSession.getStreamId(deviceId, channel.getChannelId()));
 		DeviceChannel deviceChannel = deviceChannelMapper.queryChannel(deviceId, channelId);
 		if (deviceChannel == null) {
 			deviceChannelMapper.add(channel);
@@ -191,9 +206,22 @@
 	 */
 	@Override
 	public boolean delete(String deviceId) {
-		int result = deviceMapper.del(deviceId);
-
-		return result > 0;
+		TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
+		boolean result = false;
+		try {
+			if (platformChannelMapper.delChannelForDeviceId(deviceId) <0  // 鍒犻櫎涓庡浗鏍囧钩鍙扮殑鍏宠仈
+					|| deviceChannelMapper.cleanChannelsByDeviceId(deviceId) < 0 // 鍒犻櫎浠栫殑閫氶亾
+					|| deviceMapper.del(deviceId) < 0 // 绉婚櫎璁惧淇℃伅
+			) {
+				//浜嬪姟鍥炴粴
+				dataSourceTransactionManager.rollback(transactionStatus);
+			}
+			result = true;
+			dataSourceTransactionManager.commit(transactionStatus);     //鎵嬪姩鎻愪氦
+		}catch (Exception e) {
+			dataSourceTransactionManager.rollback(transactionStatus);
+		}
+		return result;
 	}
 
 	/**
@@ -209,7 +237,7 @@
 			return false;
 		}
 		device.setOnline(1);
-		System.out.println("鏇存柊璁惧鍦ㄧ嚎");
+		logger.info("鏇存柊璁惧鍦ㄧ嚎: " + deviceId);
 		return deviceMapper.update(device) > 0;
 	}
 
@@ -221,10 +249,23 @@
 	 */
 	@Override
 	public synchronized boolean outline(String deviceId) {
+		logger.info("鏇存柊璁惧绂荤嚎: " + deviceId);
 		Device device = deviceMapper.getDeviceByDeviceId(deviceId);
+		if (device == null) return false;
 		device.setOnline(0);
-		System.out.println("鏇存柊璁惧绂荤嚎");
 		return deviceMapper.update(device) > 0;
+	}
+
+	/**
+	 * 鏇存柊鎵�鏈夎澶囩绾�
+	 *
+	 * @return true锛氭洿鏂版垚鍔�  false锛氭洿鏂板け璐�
+	 */
+	@Override
+	public synchronized boolean outlineForAll() {
+		logger.info("鏇存柊鎵�鏈夎澶囩绾�");
+		int result = deviceMapper.outlineForAll();
+		return result > 0;
 	}
 
 	/**
@@ -266,15 +307,21 @@
 	public boolean updateParentPlatform(ParentPlatform parentPlatform) {
 		int result = 0;
 		ParentPlatformCatch parentPlatformCatch = redisCatchStorage.queryPlatformCatchInfo(parentPlatform.getServerGBId()); // .getDeviceGBId());
-		if ( platformMapper.getParentPlatById(parentPlatform.getServerGBId()) == null) {
+		if (parentPlatform.getId() == null ) {
 			result = platformMapper.addParentPlatform(parentPlatform);
-
 			if (parentPlatformCatch == null) {
 				parentPlatformCatch = new ParentPlatformCatch();
 				parentPlatformCatch.setParentPlatform(parentPlatform);
 				parentPlatformCatch.setId(parentPlatform.getServerGBId());
 			}
 		}else {
+			if (parentPlatformCatch == null) { // serverGBId 宸插彉鍖�
+				ParentPlatform parentPlatById = platformMapper.getParentPlatById(parentPlatform.getId());
+				// 浣跨敤鏃х殑鏌ュ嚭缂撳瓨ID
+				parentPlatformCatch = redisCatchStorage.queryPlatformCatchInfo(parentPlatById.getServerGBId());
+				parentPlatformCatch.setId(parentPlatform.getServerGBId());
+				redisCatchStorage.delPlatformCatchInfo(parentPlatById.getServerGBId());
+			}
 			result = platformMapper.updateParentPlatform(parentPlatform);
 		}
 		// 鏇存柊缂撳瓨
@@ -300,8 +347,8 @@
 	}
 
 	@Override
-	public ParentPlatform queryParentPlatById(String platformGbId) {
-		return platformMapper.getParentPlatById(platformGbId);
+	public ParentPlatform queryParentPlatByServerGBId(String platformGbId) {
+		return platformMapper.getParentPlatByServerGBId(platformGbId);
 	}
 
 	@Override
@@ -403,6 +450,7 @@
 		TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
 		boolean result = false;
 		streamProxyItem.setStreamType("proxy");
+		streamProxyItem.setStatus(true);
 		try {
 			if (gbStreamMapper.add(streamProxyItem)<0 || streamProxyMapper.add(streamProxyItem) < 0) {
 				//浜嬪姟鍥炴粴
@@ -434,6 +482,7 @@
 			dataSourceTransactionManager.commit(transactionStatus);     //鎵嬪姩鎻愪氦
 			result = true;
 		}catch (Exception e) {
+			e.printStackTrace();
 			dataSourceTransactionManager.rollback(transactionStatus);
 		}
 		return result;
@@ -504,4 +553,42 @@
 	public StreamProxyItem queryStreamProxy(String app, String stream){
 		return streamProxyMapper.selectOne(app, stream);
 	}
+
+	@Override
+	public void updateMediaList(List<StreamPushItem> streamPushItems) {
+		if (streamPushItems == null || streamPushItems.size() == 0) return;
+		logger.info("updateMediaList:  " + streamPushItems.size());
+		streamPushMapper.addAll(streamPushItems);
+		// TODO 寰呬紭鍖�
+		for (int i = 0; i < streamPushItems.size(); i++) {
+			gbStreamMapper.setStatus(streamPushItems.get(i).getApp(), streamPushItems.get(i).getStream(), true);
+		}
+	}
+
+	@Override
+	public void updateMedia(StreamPushItem streamPushItem) {
+		streamPushMapper.del(streamPushItem.getApp(), streamPushItem.getStream());
+		streamPushMapper.add(streamPushItem);
+		gbStreamMapper.setStatus(streamPushItem.getApp(), streamPushItem.getStream(), true);
+	}
+
+	@Override
+	public void removeMedia(String app, String stream) {
+		streamPushMapper.del(app, stream);
+	}
+
+	@Override
+	public void clearMediaList() {
+		streamPushMapper.clear();
+	}
+
+	@Override
+	public void mediaOutline(String app, String streamId) {
+		gbStreamMapper.setStatus(app, streamId, false);
+	}
+
+	@Override
+	public void updateParentPlatformStatus(String platformGbID, boolean online) {
+		platformMapper.updateParentPlatformStatus(platformGbID, online);
+	}
 }

--
Gitblit v1.8.0