1 files changed, 120 insertions, 0 deletions
diff --git a/src/main/java/org/onap/pnfsimulator/simulator/scheduler/EventScheduler.java b/src/main/java/org/onap/pnfsimulator/simulator/scheduler/EventScheduler.java
new file mode 100644
index 0000000..5584cb1
--- /dev/null
+++ b/src/main/java/org/onap/pnfsimulator/simulator/scheduler/EventScheduler.java
@@ -0,0 +1,120 @@
+/*
+ * ============LICENSE_START=======================================================
+ * PNF-REGISTRATION-HANDLER
+ * ================================================================================
+ * Copyright (C) 2018 Nokia. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.pnfsimulator.simulator.scheduler;
+
+
+import com.google.gson.JsonObject;
+import org.onap.pnfsimulator.simulator.KeywordsHandler;
+import org.onap.pnfsimulator.simulator.client.HttpClientAdapterImpl;
+import org.onap.pnfsimulator.simulator.client.utils.ssl.SslAuthenticationHelper;
+import org.quartz.JobBuilder;
+import org.quartz.JobDataMap;
+import org.quartz.JobDetail;
+import org.quartz.JobExecutionContext;
+import org.quartz.JobKey;
+import org.quartz.Scheduler;
+import org.quartz.SchedulerException;
+import org.quartz.SimpleTrigger;
+import org.quartz.TriggerBuilder;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.io.IOException;
+import java.security.GeneralSecurityException;
+import java.util.List;
+import java.util.Optional;
+import java.util.stream.Collectors;
+
+import static org.onap.pnfsimulator.simulator.scheduler.EventJob.BODY;
+import static org.onap.pnfsimulator.simulator.scheduler.EventJob.CLIENT_ADAPTER;
+import static org.onap.pnfsimulator.simulator.scheduler.EventJob.EVENT_ID;
+import static org.onap.pnfsimulator.simulator.scheduler.EventJob.KEYWORDS_HANDLER;
+import static org.onap.pnfsimulator.simulator.scheduler.EventJob.TEMPLATE_NAME;
+import static org.onap.pnfsimulator.simulator.scheduler.EventJob.VES_URL;
+import static org.quartz.SimpleScheduleBuilder.simpleSchedule;
+
+@Component
+public class EventScheduler {
+
+
+    private final Scheduler scheduler;
+    private final KeywordsHandler keywordsHandler;
+    private final SslAuthenticationHelper sslAuthenticationHelper;
+
+    @Autowired
+    public EventScheduler(Scheduler scheduler, KeywordsHandler keywordsHandler, SslAuthenticationHelper sslAuthenticationHelper) {
+        this.scheduler = scheduler;
+        this.keywordsHandler = keywordsHandler;
+        this.sslAuthenticationHelper = sslAuthenticationHelper;
+    }
+
+    public String scheduleEvent(String vesUrl, Integer repeatInterval, Integer repeatCount,
+                                String templateName, String eventId, JsonObject body)
+            throws SchedulerException, IOException, GeneralSecurityException {
+
+        JobDetail jobDetail = createJobDetail(vesUrl, templateName, eventId, body);
+        SimpleTrigger trigger = createTrigger(repeatInterval, repeatCount);
+
+        scheduler.scheduleJob(jobDetail, trigger);
+        return jobDetail.getKey().getName();
+    }
+
+    public boolean cancelAllEvents() throws SchedulerException {
+        List<JobKey> jobKeys = getActiveJobsKeys();
+        return scheduler.deleteJobs(jobKeys);
+    }
+
+    public boolean cancelEvent(String jobName) throws SchedulerException {
+        Optional<JobKey> activeJobKey = getActiveJobsKeys().stream().filter(e -> e.getName().equals(jobName)).findFirst();
+        return activeJobKey.isPresent() && scheduler.deleteJob(activeJobKey.get());
+    }
+
+    private SimpleTrigger createTrigger(int interval, int repeatCount) {
+        return TriggerBuilder.newTrigger()
+                .withSchedule(simpleSchedule()
+                        .withIntervalInSeconds(interval)
+                        .withRepeatCount(repeatCount - 1))
+                .build();
+    }
+
+    private JobDetail createJobDetail(String vesUrl, String templateName, String eventId, JsonObject body) throws IOException, GeneralSecurityException {
+        JobDataMap jobDataMap = new JobDataMap();
+        jobDataMap.put(TEMPLATE_NAME, templateName);
+        jobDataMap.put(VES_URL, vesUrl);
+        jobDataMap.put(EVENT_ID, eventId);
+        jobDataMap.put(KEYWORDS_HANDLER, keywordsHandler);
+        jobDataMap.put(BODY, body);
+        jobDataMap.put(CLIENT_ADAPTER, new HttpClientAdapterImpl(vesUrl, sslAuthenticationHelper));
+
+        return JobBuilder
+                .newJob(EventJob.class)
+                .withDescription(templateName)
+                .usingJobData(jobDataMap)
+                .build();
+    }
+
+    private List<JobKey> getActiveJobsKeys() throws SchedulerException {
+        return scheduler.getCurrentlyExecutingJobs()
+                .stream()
+                .map(JobExecutionContext::getJobDetail)
+                .map(JobDetail::getKey)
+                .collect(Collectors.toList());
+    }
+}