ves-agent/src/main/java/org/commscope/tr069adapter/vesagent/timer/ScheduleTaskService.java

   1 /*
   2  * ============LICENSE_START========================================================================
   3  * ONAP : tr-069-adapter
   4  * =================================================================================================
   5  * Copyright (C) 2020 CommScope Inc Intellectual Property.
   6  * =================================================================================================
   7  * This tr-069-adapter software file is distributed by CommScope Inc under the Apache License,
   8  * Version 2.0 (the "License"); you may not use this file except in compliance with the License. You
   9  * may obtain a copy of the License at
  10  *
  11  * http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
  14  * either express or implied. See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  * ===============LICENSE_END=======================================================================
  17  */
  18
  19 package org.commscope.tr069adapter.vesagent.timer;\r
  20 \r
  21 import java.util.HashMap;\r
  22 import java.util.Map;\r
  23 import java.util.concurrent.ScheduledFuture;\r
  24 import java.util.concurrent.TimeUnit;\r
  25 \r
  26 import org.springframework.beans.factory.annotation.Autowired;\r
  27 import org.springframework.scheduling.TaskScheduler;\r
  28 import org.springframework.scheduling.annotation.EnableScheduling;\r
  29 import org.springframework.scheduling.support.PeriodicTrigger;\r
  30 import org.springframework.stereotype.Service;\r
  31 \r
  32 /**\r
  33  * \r
  34  * @version 1.0\r
  35  * @since June 5, 2020\r
  36  * @author Prashant Kumar\r
  37  */\r
  38 \r
  39 @Service\r
  40 @EnableScheduling\r
  41 public class ScheduleTaskService {\r
  42 \r
  43   @Autowired\r
  44   TaskScheduler scheduler;\r
  45 \r
  46   Map<String, ScheduledFuture<?>> jobsMap = new HashMap<>();\r
  47 \r
  48   public ScheduleTaskService(TaskScheduler scheduler) {\r
  49     this.scheduler = scheduler;\r
  50   }\r
  51 \r
  52   public void schedule(String jobId, ScheduleInfo scheduleInfo, Runnable callBackTask) {\r
  53     cancelSchedule(jobId);\r
  54 \r
  55     PeriodicTrigger trigger =\r
  56         new PeriodicTrigger(scheduleInfo.getInterval(), scheduleInfo.getTimeUnit());\r
  57     trigger.setInitialDelay(scheduleInfo.getInterval());\r
  58 \r
  59     ScheduledFuture<?> scheduledTask = scheduler.schedule(callBackTask, trigger);\r
  60     jobsMap.put(jobId, scheduledTask);\r
  61   }\r
  62 \r
  63   public void cancelSchedule(String id) {\r
  64     ScheduledFuture<?> scheduledTask = jobsMap.get(id);\r
  65     if (scheduledTask != null) {\r
  66       scheduledTask.cancel(true);\r
  67       jobsMap.remove(id);\r
  68     }\r
  69   }\r
  70 \r
  71   public ScheduledFuture<?> getSchedule(String id) {\r
  72     return jobsMap.get(id);\r
  73   }\r
  74 \r
  75   public Long getTimeRemainingTillNextExecution(String id, TimeUnit timeUnit) {\r
  76     ScheduledFuture<?> scheduledTask = jobsMap.get(id);\r
  77 \r
  78     if (scheduledTask != null) {\r
  79       return scheduledTask.getDelay(timeUnit);\r
  80     } else {\r
  81       return null;\r
  82     }\r
  83   }\r
  84 }\r