From 44d216100b45c3337c593ee82ee68e7e0f35d24b Mon Sep 17 00:00:00 2001
From: Lawrence <1934378145@qq.com>
Date: 星期三, 16 十二月 2020 20:29:19 +0800
Subject: [PATCH] 与master分支同步

---
 src/main/java/com/genersoft/iot/vmp/storager/redis/VideoManagerRedisStoragerImpl.java |  250 ++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 211 insertions(+), 39 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/storager/redis/VideoManagerRedisStoragerImpl.java b/src/main/java/com/genersoft/iot/vmp/storager/redis/VideoManagerRedisStoragerImpl.java
index a036fd0..99c7f06 100644
--- a/src/main/java/com/genersoft/iot/vmp/storager/redis/VideoManagerRedisStoragerImpl.java
+++ b/src/main/java/com/genersoft/iot/vmp/storager/redis/VideoManagerRedisStoragerImpl.java
@@ -8,6 +8,7 @@
 import com.genersoft.iot.vmp.common.StreamInfo;
 import com.genersoft.iot.vmp.conf.MediaServerConfig;
 import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
+import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
@@ -66,7 +67,6 @@
 		if (deviceMap.get(device.getDeviceId()) == null) {
 			deviceMap.put(device.getDeviceId(), new HashMap<String, HashSet<String>>());
 		}
-//		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + device.getDeviceId() + "_" + "*");
 		// 鏇存柊device涓殑閫氶亾鏁伴噺
 		device.setChannelCount(deviceMap.get(device.getDeviceId()).size());
 		// 瀛樺偍device
@@ -80,21 +80,19 @@
 		String channelId = channel.getChannelId();
 		HashMap<String, HashSet<String>> channelMap = deviceMap.get(deviceId);
 		if (channelMap == null) return;
-
 		// 浣滀负鐖惰澶�, 纭畾鑷繁鐨勫瓙鑺傜偣鏁�
 		if (channelMap.get(channelId) == null) {
 			channelMap.put(channelId, new HashSet<String>());
-		}else if (channelMap.get(channelId).size()> 0) {
+		}else if (channelMap.get(channelId).size() > 0) {
 			channel.setSubCount(channelMap.get(channelId).size());
 		}
 
 		// 瀛樺偍閫氶亾
 		redis.set(VideoManagerConstants.CACHEKEY_PREFIX + deviceId +
 						"_" + channel.getChannelId() +
-						":" + channel.getName() +
 						"_" + (channel.getStatus() == 1 ? "on":"off") +
 						"_" + (channelMap.get(channelId).size() > 0)+
-						"_" + channel.getParentId(),
+						"_" + (StringUtils.isEmpty(channel.getParentId())?null:channel.getParentId()),
 				channel);
 		// 鏇存柊device涓殑閫氶亾鏁伴噺
 		Device device = (Device)redis.get(VideoManagerConstants.DEVICE_PREFIX+deviceId);
@@ -104,17 +102,21 @@
 
 		// 濡傛灉鏈夌埗璁惧,鏇存柊鐖惰澶囧唴瀛愯妭鐐规暟
 		String parentId = channel.getParentId();
-		if (!StringUtils.isEmpty(parentId)) {
+		if (!StringUtils.isEmpty(parentId) && !parentId.equals(deviceId)) {
 
 			if (channelMap.get(parentId) == null) {
-				channelMap.put(parentId, new HashSet<>());
+				channelMap.put(parentId, new HashSet<String>());
 			}
 			channelMap.get(parentId).add(channelId);
 
 			DeviceChannel deviceChannel = queryChannel(deviceId, parentId);
 			if (deviceChannel != null) {
 				deviceChannel.setSubCount(channelMap.get(parentId).size());
-				redis.set(VideoManagerConstants.CACHEKEY_PREFIX+deviceId + "_" + deviceChannel.getChannelId(),
+				redis.set(VideoManagerConstants.CACHEKEY_PREFIX + deviceId +
+								"_" + deviceChannel.getChannelId() +
+								"_" + (deviceChannel.getStatus() == 1 ? "on":"off") +
+								"_" + (channelMap.get(deviceChannel.getChannelId()).size() > 0)+
+								"_" + (StringUtils.isEmpty(deviceChannel.getParentId())?null:deviceChannel.getParentId()),
 						deviceChannel);
 
 			}
@@ -135,6 +137,8 @@
 
 	@Override
 	public PageResult queryChannelsByDeviceId(String deviceId, String query, Boolean hasSubChannel, String online, int page, int count) {
+		// 鑾峰彇鍒版墍鏈夋鍦ㄦ挱鏀剧殑娴�
+		Map<String, StreamInfo> stringStreamInfoMap = queryPlayByDeviceId(deviceId);
 		List<DeviceChannel> result = new ArrayList<>();
 		PageResult pageResult = new PageResult<DeviceChannel>();
 		String queryContent = "*";
@@ -148,14 +152,26 @@
 				"_" + queryOnline + // 鎼滅储鏄惁鍦ㄧ嚎
 				"_" + queryHasSubChannel + // 鎼滅储鏄惁鍚湁瀛愯妭鐐�
 				"_" + "*";
-		List<Object> deviceChannelList = redis.keys(queryStr);
+//		List<Object> deviceChannelList = redis.keys(queryStr);
+		List<Object> deviceChannelList = redis.scan(queryStr);
+		//瀵规煡璇㈢粨鏋滄帓搴忥紝閬垮厤鍑虹幇閫氶亾鎺掑垪椤哄簭涔卞簭鐨勬儏鍐�
+		Collections.sort(deviceChannelList,new Comparator<Object>(){
+			@Override
+			public int compare(Object o1, Object o2) {
+				return o1.toString().compareToIgnoreCase(o2.toString());
+			}
+		});
 		pageResult.setPage(page);
 		pageResult.setCount(count);
 		pageResult.setTotal(deviceChannelList.size());
 		int maxCount = (page + 1 ) * count;
 		if (deviceChannelList != null && deviceChannelList.size() > 0 ) {
 			for (int i = page * count; i < (pageResult.getTotal() > maxCount ? maxCount : pageResult.getTotal() ); i++) {
-				result.add((DeviceChannel)redis.get((String)deviceChannelList.get(i)));
+				DeviceChannel deviceChannel = (DeviceChannel)redis.get((String)deviceChannelList.get(i));
+				StreamInfo streamInfo = stringStreamInfoMap.get(deviceId + "_" + deviceChannel.getChannelId());
+				deviceChannel.setPlay(streamInfo != null);
+				if (streamInfo != null) deviceChannel.setSsrc(streamInfo.getSsrc());
+				result.add(deviceChannel);
 			}
 			pageResult.setData(result);
 		}
@@ -163,13 +179,17 @@
 		return pageResult;
 	}
 
+
+
 	@Override
 	public List<DeviceChannel> queryChannelsByDeviceId(String deviceId) {
 		List<DeviceChannel> result = new ArrayList<>();
-		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
+//		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
+		List<Object> deviceChannelList = redis.scan(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
+
 		if (deviceChannelList != null && deviceChannelList.size() > 0 ) {
 			for (int i = 0; i < deviceChannelList.size(); i++) {
-				result.add((DeviceChannel)redis.get((String)deviceChannelList.get(i)));
+				result.add((DeviceChannel)redis.get((String) deviceChannelList.get(i)));
 			}
 		}
 		return result;
@@ -190,7 +210,8 @@
 				"_" + queryHasSubChannel + // 鎼滅储鏄惁鍚湁瀛愯妭鐐�
 				"_" + parentChannelId;
 
-		List<Object> deviceChannelList = redis.keys(queryStr);
+//		List<Object> deviceChannelList = redis.keys(queryStr);
+		List<Object> deviceChannelList = redis.scan(queryStr);
 
 		if (deviceChannelList != null && deviceChannelList.size() > 0 ) {
 			for (int i = 0; i < deviceChannelList.size(); i++) {
@@ -216,7 +237,8 @@
 
 	public List<DeviceChannel> querySubChannels(String deviceId, String parentChannelId) {
 		List<DeviceChannel> allDeviceChannels = new ArrayList<>();
-		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
+//		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
+		List<Object> deviceChannelList = redis.scan(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
 
 		if (deviceChannelList != null && deviceChannelList.size() > 0 ) {
 			for (int i = 0; i < deviceChannelList.size(); i++) {
@@ -232,7 +254,14 @@
 
 	@Override
 	public DeviceChannel queryChannel(String deviceId, String channelId) {
-		return (DeviceChannel)redis.get(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + channelId + "_");
+		DeviceChannel deviceChannel = null;
+//		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + deviceId +
+		List<Object> deviceChannelList = redis.scan(VideoManagerConstants.CACHEKEY_PREFIX + deviceId +
+				"_" + channelId  + "*");
+		if (deviceChannelList != null && deviceChannelList.size() > 0 ) {
+			deviceChannel = (DeviceChannel)redis.get((String)deviceChannelList.get(0));
+		}
+		return deviceChannel;
 	}
 
 
@@ -248,18 +277,30 @@
 		PageResult pageResult = new PageResult<Device>();
 		pageResult.setPage(page);
 		pageResult.setCount(count);
+		Device device = null;
 
 		if (deviceIds == null || deviceIds.length == 0) {
 
-			List<Object> deviceIdList = redis.keys(VideoManagerConstants.DEVICE_PREFIX+"*");
+//			List<Object> deviceIdList = redis.keys(VideoManagerConstants.DEVICE_PREFIX+"*");
+			List<Object> deviceIdList = redis.scan(VideoManagerConstants.DEVICE_PREFIX+"*");
 			pageResult.setTotal(deviceIdList.size());
 			int maxCount = (page + 1)* count;
 			for (int i = page * count; i < (pageResult.getTotal() > maxCount ? maxCount : pageResult.getTotal() ); i++) {
-				devices.add((Device)redis.get((String)deviceIdList.get(i)));
+				// devices.add((Device)redis.get((String)deviceIdList.get(i)));
+				device =(Device)redis.get((String)deviceIdList.get(i));
+				if (redis.scan(VideoManagerConstants.KEEPLIVEKEY_PREFIX+device.getDeviceId()).size() == 0){
+					// outline(device.getDeviceId());
+				}
+				devices.add(device);
 			}
 		} else {
 			for (int i = 0; i < deviceIds.length; i++) {
-				devices.add((Device)redis.get(VideoManagerConstants.DEVICE_PREFIX+deviceIds[i]));
+				// devices.add((Device)redis.get(VideoManagerConstants.DEVICE_PREFIX+deviceIds[i]));
+				device = (Device)redis.get(VideoManagerConstants.DEVICE_PREFIX+deviceIds[i]);
+				if (redis.scan(VideoManagerConstants.KEEPLIVEKEY_PREFIX+device.getDeviceId()).size() == 0){
+					// outline(device.getDeviceId());
+				}
+				devices.add(device);
 			}
 		}
 		pageResult.setData(devices);
@@ -275,15 +316,25 @@
 	@Override
 	public List<Device> queryVideoDeviceList(String[] deviceIds) {
 		List<Device> devices = new ArrayList<>();
+		Device device = null;
 
 		if (deviceIds == null || deviceIds.length == 0) {
-			List<Object> deviceIdList = redis.keys(VideoManagerConstants.DEVICE_PREFIX+"*");
+//			List<Object> deviceIdList = redis.keys(VideoManagerConstants.DEVICE_PREFIX+"*");
+			List<Object> deviceIdList = redis.scan(VideoManagerConstants.DEVICE_PREFIX+"*");
 			for (int i = 0; i < deviceIdList.size(); i++) {
-				devices.add((Device)redis.get((String)deviceIdList.get(i)));
+				device =(Device)redis.get((String)deviceIdList.get(i));
+				if (redis.scan(VideoManagerConstants.KEEPLIVEKEY_PREFIX+device.getDeviceId()).size() == 0){
+					outline(device.getDeviceId());
+				}
+				devices.add(device);
 			}
 		} else {
 			for (int i = 0; i < deviceIds.length; i++) {
-				devices.add((Device)redis.get(VideoManagerConstants.DEVICE_PREFIX+deviceIds[i]));
+				device = (Device)redis.get(VideoManagerConstants.DEVICE_PREFIX+deviceIds[i]);
+				if (redis.scan(VideoManagerConstants.KEEPLIVEKEY_PREFIX+device.getDeviceId()).size() == 0){
+					outline(device.getDeviceId());
+				}
+				devices.add(device);
 			}
 		}
 		return devices;
@@ -330,40 +381,71 @@
 	/**
 	 * 寮�濮嬫挱鏀炬椂灏嗘祦瀛樺叆redis
 	 *
-	 * @param deviceId 璁惧ID
-	 * @param channelId 閫氶亾ID
 	 * @return
 	 */
 	@Override
-	public boolean startPlay(String deviceId, String channelId, StreamInfo stream) {
-		return redis.set(String.format("%S_%s_%s", VideoManagerConstants.PLAYER_PREFIX, deviceId, channelId),
+	public boolean startPlay(StreamInfo stream) {
+		return redis.set(String.format("%S_%s_%s_%s", VideoManagerConstants.PLAYER_PREFIX, stream.getSsrc(),stream.getDeviceID(), stream.getCahnnelId()),
 				stream);
 	}
 
 	/**
 	 * 鍋滄鎾斁鏃朵粠redis鍒犻櫎
 	 *
-	 * @param deviceId 璁惧ID
-	 * @param channelId 閫氶亾ID
 	 * @return
 	 */
 	@Override
-	public boolean stopPlay(String deviceId, String channelId) {
-		return redis.del(String.format("%S_%s_%s", VideoManagerConstants.PLAYER_PREFIX, deviceId, channelId));
+	public boolean stopPlay(StreamInfo streamInfo) {
+		if (streamInfo == null) return false;
+		DeviceChannel deviceChannel = queryChannel(streamInfo.getDeviceID(), streamInfo.getCahnnelId());
+		if (deviceChannel != null) {
+			deviceChannel.setSsrc(null);
+			deviceChannel.setPlay(false);
+			updateChannel(streamInfo.getDeviceID(), deviceChannel);
+		}
+		return redis.del(String.format("%S_%s_%s_%s", VideoManagerConstants.PLAYER_PREFIX,
+				streamInfo.getSsrc(),
+				streamInfo.getDeviceID(),
+				streamInfo.getCahnnelId()));
 	}
 
 	/**
 	 * 鏌ヨ鎾斁鍒楄〃
-	 * @param deviceId 璁惧ID
-	 * @param channelId 閫氶亾ID
 	 * @return
 	 */
 	@Override
-	public StreamInfo queryPlay(String deviceId, String channelId) {
-		return (StreamInfo)redis.get(String.format("%S_%s_%s", VideoManagerConstants.PLAYER_PREFIX, deviceId, channelId));
+	public StreamInfo queryPlay(StreamInfo streamInfo) {
+		return (StreamInfo)redis.get(String.format("%S_%s_%s_%s",
+				VideoManagerConstants.PLAYER_PREFIX,
+				streamInfo.getSsrc(),
+				streamInfo.getDeviceID(),
+				streamInfo.getCahnnelId()));
+	}
+	@Override
+	public StreamInfo queryPlayBySSRC(String ssrc) {
+//		List<Object> playLeys = redis.keys(String.format("%S_%s_*", VideoManagerConstants.PLAYER_PREFIX, ssrc));
+		List<Object> playLeys = redis.scan(String.format("%S_%s_*", VideoManagerConstants.PLAYER_PREFIX, ssrc));
+		if (playLeys == null || playLeys.size() == 0) return null;
+		return (StreamInfo)redis.get(playLeys.get(0).toString());
 	}
 
+	@Override
+	public StreamInfo queryPlaybackBySSRC(String ssrc) {
+//		List<Object> playLeys = redis.keys(String.format("%S_%s_*", VideoManagerConstants.PLAYER_PREFIX, ssrc));
+		List<Object> playLeys = redis.scan(String.format("%S_%s_*", VideoManagerConstants.PLAY_BLACK_PREFIX, ssrc));
+		if (playLeys == null || playLeys.size() == 0) return null;
+		return (StreamInfo)redis.get(playLeys.get(0).toString());
+	}
 
+	@Override
+	public StreamInfo queryPlayByDevice(String deviceId, String code) {
+//		List<Object> playLeys = redis.keys(String.format("%S_*_%s_%s", VideoManagerConstants.PLAYER_PREFIX,
+		List<Object> playLeys = redis.scan(String.format("%S_*_%s_%s", VideoManagerConstants.PLAYER_PREFIX,
+				deviceId,
+				code));
+		if (playLeys == null || playLeys.size() == 0) return null;
+		return (StreamInfo)redis.get(playLeys.get(0).toString());
+	}
 
 	/**
 	 * 鏇存柊娴佸獟浣撲俊鎭�
@@ -393,36 +475,126 @@
 		for (Device device : devices) {
 			// 鏇存柊璁惧涓嬬殑閫氶亾
 			HashMap<String, HashSet<String>> channelMap = new HashMap<String, HashSet<String>>();
-			List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX +
+			List<Object> deviceChannelList = redis.scan(VideoManagerConstants.CACHEKEY_PREFIX +
 					device.getDeviceId() + "_" + "*");
 			if (deviceChannelList != null && deviceChannelList.size() > 0 ) {
 				for (int i = 0; i < deviceChannelList.size(); i++) {
 					String key = (String)deviceChannelList.get(i);
 					String[] s = key.split("_");
-					String channelId = s[3].split(":")[0];
+					String channelId = s[3];
 					HashSet<String> subChannel = channelMap.get(channelId);
 					if (subChannel == null) {
 						subChannel = new HashSet<>();
 					}
-					if ("null".equals(s[6])) {
-						subChannel.add(s[6]);
+					System.out.println(key);
+					if (s.length == 6 && !"null".equals(s[5])) {
+						subChannel.add(s[5]);
 					}
 					channelMap.put(channelId, subChannel);
-					System.out.println();
 				}
 			}
 			deviceMap.put(device.getDeviceId(),channelMap);
 		}
+		System.out.println();
 	}
 
 	@Override
 	public void cleanChannelsForDevice(String deviceId) {
 		List<DeviceChannel> result = new ArrayList<>();
-		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
+//		List<Object> deviceChannelList = redis.keys(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
+		List<Object> deviceChannelList = redis.scan(VideoManagerConstants.CACHEKEY_PREFIX + deviceId + "_" + "*");
 		if (deviceChannelList != null && deviceChannelList.size() > 0 ) {
 			for (int i = 0; i < deviceChannelList.size(); i++) {
 				redis.del((String)deviceChannelList.get(i));
 			}
 		}
 	}
+
+	@Override
+	public Map<String, StreamInfo> queryPlayByDeviceId(String deviceId) {
+		Map<String, StreamInfo> streamInfos = new HashMap<>();
+//		List<Object> playLeys = redis.keys(String.format("%S_*_%S_*", VideoManagerConstants.PLAYER_PREFIX, deviceId));
+		List<Object> playLeys = redis.scan(String.format("%S_*_%S_*", VideoManagerConstants.PLAYER_PREFIX, deviceId));
+		if (playLeys.size() == 0) return streamInfos;
+		for (int i = 0; i < playLeys.size(); i++) {
+			String key = (String) playLeys.get(i);
+			StreamInfo streamInfo = (StreamInfo)redis.get(key);
+			streamInfos.put(streamInfo.getDeviceID() + "_" + streamInfo.getCahnnelId(), streamInfo);
+		}
+		return streamInfos;
+	}
+
+
+	@Override
+	public boolean startPlayback(StreamInfo stream) {
+		return redis.set(String.format("%S_%s_%s_%s", VideoManagerConstants.PLAY_BLACK_PREFIX, stream.getSsrc(),stream.getDeviceID(), stream.getCahnnelId()),
+				stream);
+	}
+
+
+	@Override
+	public boolean stopPlayback(StreamInfo streamInfo) {
+		if (streamInfo == null) return false;
+		DeviceChannel deviceChannel = queryChannel(streamInfo.getDeviceID(), streamInfo.getCahnnelId());
+		if (deviceChannel != null) {
+			deviceChannel.setSsrc(null);
+			deviceChannel.setPlay(false);
+			updateChannel(streamInfo.getDeviceID(), deviceChannel);
+		}
+		return redis.del(String.format("%S_%s_%s_%s", VideoManagerConstants.PLAY_BLACK_PREFIX,
+				streamInfo.getSsrc(),
+				streamInfo.getDeviceID(),
+				streamInfo.getCahnnelId()));
+	}
+
+	@Override
+	public StreamInfo queryPlaybackByDevice(String deviceId, String code) {
+		String format = String.format("%S_*_%s_%s", VideoManagerConstants.PLAY_BLACK_PREFIX,
+				deviceId,
+				code);
+		List<Object> playLeys = redis.scan(String.format("%S_*_%s_%s", VideoManagerConstants.PLAY_BLACK_PREFIX,
+				deviceId,
+				code));
+		if (playLeys == null || playLeys.size() == 0) {
+			playLeys = redis.scan(String.format("%S_*_*_%s", VideoManagerConstants.PLAY_BLACK_PREFIX,
+				deviceId));
+		}
+		if (playLeys == null || playLeys.size() == 0) return null;
+		return (StreamInfo)redis.get(playLeys.get(0).toString());
+	}
+
+	@Override
+	public boolean updateParentPlatform(ParentPlatform parentPlatform) {
+
+		// 瀛樺偍device
+		return redis.set(VideoManagerConstants.PLATFORM_PREFIX + parentPlatform.getDeviceGBId(), parentPlatform);
+	}
+
+	@Override
+	public boolean deleteParentPlatform(ParentPlatform parentPlatform) {
+		return redis.del(VideoManagerConstants.PLATFORM_PREFIX + parentPlatform.getDeviceGBId());
+	}
+
+	@Override
+	public PageResult<ParentPlatform> queryParentPlatformList(int page, int count) {
+		PageResult pageResult = new PageResult<Device>();
+		pageResult.setPage(page);
+		pageResult.setCount(count);
+		List<ParentPlatform> resultData = new ArrayList<>();
+		List<Object> parentPlatformList = redis.scan(VideoManagerConstants.PLATFORM_PREFIX + "*");
+		pageResult.setTotal(parentPlatformList.size());
+		int maxCount = (page + 1)* count;
+		for (int i = page * count; i < (pageResult.getTotal() > maxCount ? maxCount : pageResult.getTotal() ); i++) {
+			ParentPlatform parentPlatform =(ParentPlatform)redis.get((String)parentPlatformList.get(i));
+			resultData.add(parentPlatform);
+
+		}
+		pageResult.setData(resultData);
+		return pageResult;
+	}
+
+	@Override
+	public ParentPlatform queryParentPlatById(String platformGbId) {
+		return (ParentPlatform)redis.get(VideoManagerConstants.PLATFORM_PREFIX + platformGbId);
+	}
 }

--
Gitblit v1.8.0