From ba8633e99a27b90c55b47e8f7d6225f1f0efc6bb Mon Sep 17 00:00:00 2001 From: 648540858 <456panlinlin> Date: 星期五, 08 四月 2022 18:09:23 +0800 Subject: [PATCH] 优化地图-添加地图页面以及设备树 --- src/main/java/com/genersoft/iot/vmp/conf/DynamicTask.java | 51 +++++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 45 insertions(+), 6 deletions(-) diff --git a/src/main/java/com/genersoft/iot/vmp/conf/DynamicTask.java b/src/main/java/com/genersoft/iot/vmp/conf/DynamicTask.java index 145c29b..0c57bde 100644 --- a/src/main/java/com/genersoft/iot/vmp/conf/DynamicTask.java +++ b/src/main/java/com/genersoft/iot/vmp/conf/DynamicTask.java @@ -1,12 +1,14 @@ package com.genersoft.iot.vmp.conf; +import com.genersoft.iot.vmp.gb28181.task.ISubscribeTask; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Bean; import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler; -import org.springframework.scheduling.support.CronTrigger; import org.springframework.stereotype.Component; +import java.util.Date; import java.util.Map; +import java.util.Set; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ScheduledFuture; @@ -20,23 +22,60 @@ private ThreadPoolTaskScheduler threadPoolTaskScheduler; private Map<String, ScheduledFuture<?>> futureMap = new ConcurrentHashMap<>(); + private Map<String, Runnable> runnableMap = new ConcurrentHashMap<>(); @Bean public ThreadPoolTaskScheduler threadPoolTaskScheduler() { return new ThreadPoolTaskScheduler(); } - public String startCron(String key, Runnable task, String corn) { - stopCron(key); - ScheduledFuture future = threadPoolTaskScheduler.schedule(task, new CronTrigger(corn)); + /** + * 寰幆鎵ц鐨勪换鍔� + * @param key 浠诲姟ID + * @param task 浠诲姟 + * @param cycleForCatalog 闂撮殧 + * @return + */ + public void startCron(String key, Runnable task, int cycleForCatalog) { + stop(key); + // scheduleWithFixedDelay 蹇呴』绛夊緟涓婁竴涓换鍔$粨鏉熸墠寮�濮嬭鏃秔eriod锛� cycleForCatalog琛ㄧず鎵ц鐨勯棿闅� + ScheduledFuture future = threadPoolTaskScheduler.scheduleWithFixedDelay(task, cycleForCatalog * 1000L); futureMap.put(key, future); - return "startCron"; + runnableMap.put(key, task); } - public void stopCron(String key) { + /** + * 寤舵椂浠诲姟 + * @param key 浠诲姟ID + * @param task 浠诲姟 + * @param delay 寤舵椂 /姣 + * @return + */ + public void startDelay(String key, Runnable task, int delay) { + stop(key); + Date starTime = new Date(System.currentTimeMillis() + delay); + // scheduleWithFixedDelay 蹇呴』绛夊緟涓婁竴涓换鍔$粨鏉熸墠寮�濮嬭鏃秔eriod锛� cycleForCatalog琛ㄧず鎵ц鐨勯棿闅� + ScheduledFuture future = threadPoolTaskScheduler.schedule(task, starTime); + futureMap.put(key, future); + } + + public void stop(String key) { if (futureMap.get(key) != null && !futureMap.get(key).isCancelled()) { futureMap.get(key).cancel(true); + Runnable runnable = runnableMap.get(key); + if (runnable instanceof ISubscribeTask) { + ISubscribeTask subscribeTask = (ISubscribeTask) runnable; + subscribeTask.stop(); + } } } + public boolean contains(String key) { + return futureMap.get(key) != null; + } + + public Set<String> getAllKeys() { + return futureMap.keySet(); + } + } -- Gitblit v1.8.0