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