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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=======================================================================
19 package org.commscope.tr069adapter.vesagent.timer;
\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
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
35 * @since June 5, 2020
\r
36 * @author Prashant Kumar
\r
41 public class ScheduleTaskService {
\r
44 TaskScheduler scheduler;
\r
46 Map<String, ScheduledFuture<?>> jobsMap = new HashMap<>();
\r
48 public ScheduleTaskService(TaskScheduler scheduler) {
\r
49 this.scheduler = scheduler;
\r
52 public void schedule(String jobId, ScheduleInfo scheduleInfo, Runnable callBackTask) {
\r
53 cancelSchedule(jobId);
\r
55 PeriodicTrigger trigger =
\r
56 new PeriodicTrigger(scheduleInfo.getInterval(), scheduleInfo.getTimeUnit());
\r
57 trigger.setInitialDelay(scheduleInfo.getInterval());
\r
59 ScheduledFuture<?> scheduledTask = scheduler.schedule(callBackTask, trigger);
\r
60 jobsMap.put(jobId, scheduledTask);
\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
71 public ScheduledFuture<?> getSchedule(String id) {
\r
72 return jobsMap.get(id);
\r
75 public Long getTimeRemainingTillNextExecution(String id, TimeUnit timeUnit) {
\r
76 ScheduledFuture<?> scheduledTask = jobsMap.get(id);
\r
78 if (scheduledTask != null) {
\r
79 return scheduledTask.getDelay(timeUnit);
\r