From 21a96ad20fd75e55d03c00af8df8adb039f0c77a Mon Sep 17 00:00:00 2001 From: 648540858 <648540858@qq.com> Date: 星期二, 20 六月 2023 12:51:06 +0800 Subject: [PATCH] 修复通道刷新 --- src/main/java/com/genersoft/iot/vmp/gb28181/transmit/callback/DeferredResultHolder.java | 96 ++++++++++++++++++++++++++++++++++-------------- 1 files changed, 68 insertions(+), 28 deletions(-) diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/callback/DeferredResultHolder.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/callback/DeferredResultHolder.java index ecfa666..2175a45 100644 --- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/callback/DeferredResultHolder.java +++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/callback/DeferredResultHolder.java @@ -1,17 +1,17 @@ package com.genersoft.iot.vmp.gb28181.transmit.callback; -import java.util.HashMap; +import com.genersoft.iot.vmp.vmanager.bean.DeferredResultEx; +import org.springframework.stereotype.Component; +import org.springframework.util.ObjectUtils; +import org.springframework.web.context.request.async.DeferredResult; + +import java.util.Collection; import java.util.Map; import java.util.Set; import java.util.concurrent.ConcurrentHashMap; -import org.springframework.http.HttpStatus; -import org.springframework.http.ResponseEntity; -import org.springframework.stereotype.Component; -import org.springframework.web.context.request.async.DeferredResult; - /** - * @Description: 寮傛璇锋眰澶勭悊 + * @description: 寮傛璇锋眰澶勭悊 * @author: swwheihei * @date: 2020骞�5鏈�8鏃� 涓嬪崍7:59:05 */ @@ -35,11 +35,17 @@ public static final String CALLBACK_CMD_PLAY = "CALLBACK_PLAY"; + public static final String CALLBACK_CMD_PLAYBACK = "CALLBACK_PLAYBACK"; + + public static final String CALLBACK_CMD_DOWNLOAD = "CALLBACK_DOWNLOAD"; + + public static final String CALLBACK_CMD_PROXY = "CALLBACK_PROXY"; + public static final String CALLBACK_CMD_STOP = "CALLBACK_STOP"; - public static final String CALLBACK_ONVIF = "CALLBACK_ONVIF"; + public static final String UPLOAD_FILE_CHANNEL = "UPLOAD_FILE_CHANNEL"; - public static final String CALLBACK_CMD_MOBILEPOSITION = "CALLBACK_MOBILEPOSITION"; + public static final String CALLBACK_CMD_MOBILE_POSITION = "CALLBACK_CMD_MOBILE_POSITION"; public static final String CALLBACK_CMD_PRESETQUERY = "CALLBACK_PRESETQUERY"; @@ -47,27 +53,50 @@ public static final String CALLBACK_CMD_BROADCAST = "CALLBACK_BROADCAST"; - private Map<String, Map<String, DeferredResult>> map = new ConcurrentHashMap<>(); + public static final String CALLBACK_CMD_SNAP= "CALLBACK_SNAP"; + + private Map<String, Map<String, DeferredResultEx>> map = new ConcurrentHashMap<>(); - public void put(String key, String id, DeferredResult result) { - Map<String, DeferredResult> deferredResultMap = map.get(key); + public void put(String key, String id, DeferredResultEx result) { + Map<String, DeferredResultEx> deferredResultMap = map.get(key); if (deferredResultMap == null) { deferredResultMap = new ConcurrentHashMap<>(); map.put(key, deferredResultMap); } deferredResultMap.put(id, result); } + + public void put(String key, String id, DeferredResult result) { + Map<String, DeferredResultEx> deferredResultMap = map.get(key); + if (deferredResultMap == null) { + deferredResultMap = new ConcurrentHashMap<>(); + map.put(key, deferredResultMap); + } + deferredResultMap.put(id, new DeferredResultEx(result)); + } - public DeferredResult get(String key, String id) { - Map<String, DeferredResult> deferredResultMap = map.get(key); - if (deferredResultMap == null) return null; + public DeferredResultEx get(String key, String id) { + Map<String, DeferredResultEx> deferredResultMap = map.get(key); + if (deferredResultMap == null || ObjectUtils.isEmpty(id)) { + return null; + } return deferredResultMap.get(id); } + public Collection<DeferredResultEx> getAllByKey(String key) { + Map<String, DeferredResultEx> deferredResultMap = map.get(key); + if (deferredResultMap == null) { + return null; + } + return deferredResultMap.values(); + } + public boolean exist(String key, String id){ - if (key == null) return false; - Map<String, DeferredResult> deferredResultMap = map.get(key); + if (key == null) { + return false; + } + Map<String, DeferredResultEx> deferredResultMap = map.get(key); if (id == null) { return deferredResultMap != null; }else { @@ -80,15 +109,15 @@ * @param msg */ public void invokeResult(RequestMessage msg) { - Map<String, DeferredResult> deferredResultMap = map.get(msg.getKey()); + Map<String, DeferredResultEx> deferredResultMap = map.get(msg.getKey()); if (deferredResultMap == null) { return; } - DeferredResult result = deferredResultMap.get(msg.getId()); + DeferredResultEx result = deferredResultMap.get(msg.getId()); if (result == null) { return; } - result.setResult(new ResponseEntity<>(msg.getData(),HttpStatus.OK)); + result.getDeferredResult().setResult(msg.getData()); deferredResultMap.remove(msg.getId()); if (deferredResultMap.size() == 0) { map.remove(msg.getKey()); @@ -100,19 +129,30 @@ * @param msg */ public void invokeAllResult(RequestMessage msg) { - Map<String, DeferredResult> deferredResultMap = map.get(msg.getKey()); + Map<String, DeferredResultEx> deferredResultMap = map.get(msg.getKey()); if (deferredResultMap == null) { return; } - Set<String> ids = deferredResultMap.keySet(); - for (String id : ids) { - DeferredResult result = deferredResultMap.get(id); - if (result == null) { + synchronized (this) { + deferredResultMap = map.get(msg.getKey()); + if (deferredResultMap == null) { return; } - result.setResult(new ResponseEntity<>(msg.getData(),HttpStatus.OK)); - } - map.remove(msg.getKey()); + Set<String> ids = deferredResultMap.keySet(); + for (String id : ids) { + DeferredResultEx result = deferredResultMap.get(id); + if (result == null) { + return; + } + if (result.getFilter() != null) { + Object handler = result.getFilter().handler(msg.getData()); + result.getDeferredResult().setResult(handler); + }else { + result.getDeferredResult().setResult(msg.getData()); + } + } + map.remove(msg.getKey()); + } } } -- Gitblit v1.8.0