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 |  316 ++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 289 insertions(+), 27 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 4547d76..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,22 +2,25 @@
 
 import java.util.*;
 
-import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
-import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
-import com.genersoft.iot.vmp.gb28181.bean.ParentPlatformCatch;
+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.DeviceChannelMapper;
-import com.genersoft.iot.vmp.storager.dao.DeviceMapper;
-import com.genersoft.iot.vmp.storager.dao.ParentPlatformMapper;
-import com.genersoft.iot.vmp.storager.dao.PatformChannelMapper;
-import com.genersoft.iot.vmp.vmanager.platform.bean.ChannelReduce;
+import com.genersoft.iot.vmp.storager.dao.*;
+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;
 
-import com.genersoft.iot.vmp.gb28181.bean.Device;
 import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
+import org.springframework.transaction.TransactionDefinition;
+import org.springframework.transaction.TransactionStatus;
 import org.springframework.transaction.annotation.Transactional;
 
 /**    
@@ -25,14 +28,26 @@
  * @author: swwheihei
  * @date:   2020骞�5鏈�6鏃� 涓嬪崍2:31:42
  */
+@SuppressWarnings("rawtypes")
 @Component
 public class VideoManagerStoragerImpl implements IVideoManagerStorager {
+
+	private Logger logger = LoggerFactory.getLogger(VideoManagerStoragerImpl.class);
+
+	@Autowired
+	DataSourceTransactionManager dataSourceTransactionManager;
+
+	@Autowired
+	TransactionDefinition transactionDefinition;
 
 	@Autowired
     private DeviceMapper deviceMapper;
 
 	@Autowired
-    private DeviceChannelMapper deviceChannelMapper;
+	private DeviceChannelMapper deviceChannelMapper;
+
+	@Autowired
+	private DeviceMobilePositionMapper deviceMobilePositionMapper;
 
 	@Autowired
     private ParentPlatformMapper platformMapper;
@@ -41,9 +56,19 @@
     private IRedisCatchStorage redisCatchStorage;
 
 	@Autowired
-    private PatformChannelMapper patformChannelMapper;
+    private PlatformChannelMapper platformChannelMapper;
 
+	@Autowired
+    private StreamProxyMapper streamProxyMapper;
 
+	@Autowired
+    private StreamPushMapper streamPushMapper;
+
+	@Autowired
+    private GbStreamMapper gbStreamMapper;
+
+	@Autowired
+    private VideoStreamSessionManager streamSession;
 
 
 	/**
@@ -91,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);
@@ -180,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;
 	}
 
 	/**
@@ -194,11 +233,11 @@
 	@Override
 	public synchronized boolean online(String deviceId) {
 		Device device = deviceMapper.getDeviceByDeviceId(deviceId);
-		device.setOnline(1);
-		System.out.println("鏇存柊璁惧鍦ㄧ嚎");
 		if (device == null) {
 			return false;
 		}
+		device.setOnline(1);
+		logger.info("鏇存柊璁惧鍦ㄧ嚎: " + deviceId);
 		return deviceMapper.update(device) > 0;
 	}
 
@@ -210,16 +249,52 @@
 	 */
 	@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;
+	}
 
+	/**
+	 * 娓呯┖閫氶亾
+	 * @param deviceId
+	 */
 	@Override
 	public void cleanChannelsForDevice(String deviceId) {
-		int result = deviceChannelMapper.cleanChannelsByDeviceId(deviceId);
+		deviceChannelMapper.cleanChannelsByDeviceId(deviceId);
+	}
+
+	/**
+	 * 娣诲姞Mobile Position璁惧绉诲姩浣嶇疆
+	 * @param mobilePosition
+	 */
+	@Override
+	public synchronized boolean insertMobilePosition(MobilePosition mobilePosition) {
+		return deviceMobilePositionMapper.insertNewPosition(mobilePosition) > 0;
+	}
+
+	/**
+	 * 鏌ヨ绉诲姩浣嶇疆杞ㄨ抗
+	 * @param deviceId
+	 * @param startTime
+	 * @param endTime
+	 */
+	@Override
+	public synchronized List<MobilePosition> queryMobilePositions(String deviceId, String startTime, String endTime) {
+		return deviceMobilePositionMapper.queryPositionByDeviceIdAndTime(deviceId, startTime, endTime);
 	}
 
 	@Override
@@ -232,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);
 		}
 		// 鏇存柊缂撳瓨
@@ -254,7 +335,7 @@
 	public boolean deleteParentPlatform(ParentPlatform parentPlatform) {
 		int result = platformMapper.delParentPlatform(parentPlatform);
 		// 鍒犻櫎鍏宠仈鐨勯�氶亾
-		patformChannelMapper.cleanChannelForGB(parentPlatform.getDeviceGBId());
+		platformChannelMapper.cleanChannelForGB(parentPlatform.getServerGBId());
 		return result > 0;
 	}
 
@@ -266,8 +347,8 @@
 	}
 
 	@Override
-	public ParentPlatform queryParentPlatById(String platformGbId) {
-		return platformMapper.getParentPlatById(platformGbId);
+	public ParentPlatform queryParentPlatByServerGBId(String platformGbId) {
+		return platformMapper.getParentPlatByServerGBId(platformGbId);
 	}
 
 	@Override
@@ -283,7 +364,7 @@
 
 	@Override
 	public PageInfo<ChannelReduce> queryAllChannelList(int page, int count, String query, Boolean online,
-														 Boolean channelType, String platformId, Boolean inPlatform) {
+													   Boolean channelType, String platformId, Boolean inPlatform) {
 		PageHelper.startPage(page, count);
 		List<ChannelReduce> all = deviceChannelMapper.queryChannelListInAll(query, online, channelType, platformId, inPlatform);
 		return new PageInfo<>(all);
@@ -304,7 +385,7 @@
 		}
 		List<String> deviceAndChannelList = new ArrayList<>(deviceAndChannels.keySet());
 		// 鏌ヨ褰撳墠宸茬粡瀛樺湪鐨�
-		List<String> relatedPlatformchannels = patformChannelMapper.findChannelRelatedPlatform(platformId, deviceAndChannelList);
+		List<String> relatedPlatformchannels = platformChannelMapper.findChannelRelatedPlatform(platformId, deviceAndChannelList);
 		if (relatedPlatformchannels != null) {
 			deviceAndChannelList.removeAll(relatedPlatformchannels);
 		}
@@ -315,7 +396,7 @@
 		// 瀵瑰墿涓嬬殑鏁版嵁杩涜瀛樺偍
 		int result = 0;
 		if (channelReducesToAdd.size() > 0) {
-			result = patformChannelMapper.addChannels(platformId, channelReducesToAdd);
+			result = platformChannelMapper.addChannels(platformId, channelReducesToAdd);
 		}
 
 		return result;
@@ -325,8 +406,189 @@
 	@Override
 	public int delChannelForGB(String platformId, List<ChannelReduce> channelReduces) {
 
-		int result = patformChannelMapper.delChannelForGB(platformId, channelReduces);
+		int result = platformChannelMapper.delChannelForGB(platformId, channelReduces);
 
 		return result;
 	}
+
+	@Override
+	public DeviceChannel queryChannelInParentPlatform(String platformId, String channelId) {
+		DeviceChannel channel = platformChannelMapper.queryChannelInParentPlatform(platformId, channelId);
+		return channel;
+	}
+
+	@Override
+	public Device queryVideoDeviceByPlatformIdAndChannelId(String platformId, String channelId) {
+		Device device = platformChannelMapper.queryVideoDeviceByPlatformIdAndChannelId(platformId, channelId);
+		return device;
+	}
+
+	/**
+	 * 鏌ヨ鏈�鏂扮Щ鍔ㄤ綅缃�
+	 * @param deviceId
+	 */
+	@Override
+	public MobilePosition queryLatestPosition(String deviceId) {
+		return deviceMobilePositionMapper.queryLatestPositionByDevice(deviceId);
+	}
+
+	/**
+	 * 鍒犻櫎鎸囧畾璁惧鐨勬墍鏈夌Щ鍔ㄤ綅缃�
+	 * @param deviceId
+	 */
+	public int clearMobilePositionsByDeviceId(String deviceId) {
+		return deviceMobilePositionMapper.clearMobilePositionsByDeviceId(deviceId);
+	}
+
+	/**
+	 * 鏂板浠g悊娴�
+	 * @param streamProxyItem
+	 * @return
+	 */
+	@Override
+	public boolean addStreamProxy(StreamProxyItem streamProxyItem) {
+		TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
+		boolean result = false;
+		streamProxyItem.setStreamType("proxy");
+		streamProxyItem.setStatus(true);
+		try {
+			if (gbStreamMapper.add(streamProxyItem)<0 || streamProxyMapper.add(streamProxyItem) < 0) {
+				//浜嬪姟鍥炴粴
+				dataSourceTransactionManager.rollback(transactionStatus);
+			}
+			result = true;
+			dataSourceTransactionManager.commit(transactionStatus);     //鎵嬪姩鎻愪氦
+		}catch (Exception e) {
+			dataSourceTransactionManager.rollback(transactionStatus);
+		}
+		return result;
+	}
+
+	/**
+	 * 鏇存柊浠g悊娴�
+	 * @param streamProxyItem
+	 * @return
+	 */
+	@Override
+	public boolean updateStreamProxy(StreamProxyItem streamProxyItem) {
+		TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
+		boolean result = false;
+		streamProxyItem.setStreamType("proxy");
+		try {
+			if (gbStreamMapper.update(streamProxyItem)<0 || streamProxyMapper.update(streamProxyItem) < 0) {
+				//浜嬪姟鍥炴粴
+				dataSourceTransactionManager.rollback(transactionStatus);
+			}
+			dataSourceTransactionManager.commit(transactionStatus);     //鎵嬪姩鎻愪氦
+			result = true;
+		}catch (Exception e) {
+			e.printStackTrace();
+			dataSourceTransactionManager.rollback(transactionStatus);
+		}
+		return result;
+	}
+
+	/**
+	 * 绉婚櫎浠g悊娴�
+	 * @param app
+	 * @param stream
+	 * @return
+	 */
+	@Override
+	public int deleteStreamProxy(String app, String stream) {
+		return streamProxyMapper.del(app, stream);
+	}
+
+	/**
+	 * 鏍规嵁鏄惁鍚敤鑾峰彇浠g悊娴佸垪琛�
+	 * @param enable
+	 * @return
+	 */
+	@Override
+	public List<StreamProxyItem> getStreamProxyListForEnable(boolean enable) {
+		return streamProxyMapper.selectForEnable(enable);
+	}
+
+	/**
+	 * 鍒嗛〉鏌ヨ浠g悊娴佸垪琛�
+	 * @param page
+	 * @param count
+	 * @return
+	 */
+	@Override
+	public PageInfo<StreamProxyItem> queryStreamProxyList(Integer page, Integer count) {
+		PageHelper.startPage(page, count);
+		List<StreamProxyItem> all = streamProxyMapper.selectAll();
+		return new PageInfo<>(all);
+	}
+
+	/**
+	 * 鏍规嵁鍥芥爣ID鑾峰彇骞冲彴鍏宠仈鐨勭洿鎾祦
+	 * @param platformId
+	 * @param gbId
+	 * @return
+	 */
+	@Override
+	public GbStream queryStreamInParentPlatform(String platformId, String gbId) {
+		return gbStreamMapper.queryStreamInPlatform(platformId, gbId);
+	}
+
+	/**
+	 * 鑾峰彇骞冲彴鍏宠仈鐨勭洿鎾祦
+	 * @param platformId
+	 * @return
+	 */
+	@Override
+	public List<GbStream> queryGbStreamListInPlatform(String platformId) {
+		return gbStreamMapper.queryGbStreamListInPlatform(platformId);
+	}
+
+	/**
+	 * 鎸夌収鏄痑pp鍜宻tream鑾峰彇浠g悊娴�
+	 * @param app
+	 * @param stream
+	 * @return
+	 */
+	@Override
+	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