wvp-pro.git - Gitblit

			@@ -2,6 +2,8 @@

			import java.util.HashMap;
			import java.util.Map;
			import java.util.Set;
			import java.util.concurrent.ConcurrentHashMap;

			import org.springframework.http.HttpStatus;
			import org.springframework.http.ResponseEntity;
			@@ -9,34 +11,116 @@
			import org.springframework.web.context.request.async.DeferredResult;

			/**
			* @Description:TODO(这里用一句话描述这个类的作用)
			* @description: 异步请求处理
			* @author: swwheihei
			* @date: 2020年5月8日下午7:59:05
			*/
			@SuppressWarnings(value = {"rawtypes", "unchecked"})
			@Component
			public class DeferredResultHolder {

			public static final String CALLBACK_CMD_DEVICESTATUS = "CALLBACK_DEVICESTATUS";

			public static final String CALLBACK_CMD_DEVICEINFO = "CALLBACK_DEVICEINFO";

			public static final String CALLBACK_CMD_DEVICECONTROL = "CALLBACK_DEVICECONTROL";

			public static final String CALLBACK_CMD_DEVICECONFIG = "CALLBACK_DEVICECONFIG";

			public static final String CALLBACK_CMD_CONFIGDOWNLOAD = "CALLBACK_CONFIGDOWNLOAD";

			public static final String CALLBACK_CMD_CATALOG = "CALLBACK_CATALOG";

			public static final String CALLBACK_CMD_RECORDINFO = "CALLBACK_RECORDINFO";

			private Map<String, DeferredResult> map = new HashMap<String, DeferredResult>();

			public void put(String key, DeferredResult result) {
			map.put(key, result);
			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_STOP = "CALLBACK_STOP";

			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_PRESETQUERY = "CALLBACK_PRESETQUERY";

			public static final String CALLBACK_CMD_ALARM = "CALLBACK_ALARM";

			public static final String CALLBACK_CMD_BROADCAST = "CALLBACK_BROADCAST";

			private Map<String, Map<String, DeferredResult>> map = new ConcurrentHashMap<>();


			public void put(String key, String id, DeferredResult result) {
			Map<String, DeferredResult> deferredResultMap = map.get(key);
			if (deferredResultMap == null) {
			deferredResultMap = new ConcurrentHashMap<>();
			map.put(key, deferredResultMap);
			}
			deferredResultMap.put(id, result);
			}

			public DeferredResult get(String key) {
			return map.get(key);
			public DeferredResult get(String key, String id) {
			Map<String, DeferredResult> deferredResultMap = map.get(key);
			if (deferredResultMap == null) {
			return null;
			}
			return deferredResultMap.get(id);
			}


			public boolean exist(String key, String id){
			if (key == null) {
			return false;
			}
			Map<String, DeferredResult> deferredResultMap = map.get(key);
			if (id == null) {
			return deferredResultMap != null;
			}else {
			return deferredResultMap != null && deferredResultMap.get(id) != null;
			}
			}

			/**
			* 释放单个请求
			* @param msg
			*/
			public void invokeResult(RequestMessage msg) {
			DeferredResult result = map.get(msg.getId());
			Map<String, DeferredResult> deferredResultMap = map.get(msg.getKey());
			if (deferredResultMap == null) {
			return;
			}
			DeferredResult result = deferredResultMap.get(msg.getId());
			if (result == null) {
			return;
			}
			result.setResult(new ResponseEntity<>(msg.getData(),HttpStatus.OK));
			deferredResultMap.remove(msg.getId());
			if (deferredResultMap.size() == 0) {
			map.remove(msg.getKey());
			}
			}

			/**
			* 释放所有的请求
			* @param msg
			*/
			public void invokeAllResult(RequestMessage msg) {
			Map<String, DeferredResult> 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) {
			return;
			}
			result.setResult(ResponseEntity.ok().body(msg.getData()));
			}
			map.remove(msg.getKey());

			}
			}