From fcea425dbf82f34ae5e306922e02b2ea9bd20a81 Mon Sep 17 00:00:00 2001
From: ViacheslavKlimov <viacheslavklimov11@gmail.com>
Date: Mon, 18 Dec 2023 12:51:05 +0200
Subject: [PATCH] Shutdown notification scheduler in PreDestroy

---
 .../DefaultNotificationSchedulerService.java             | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/application/src/main/java/org/thingsboard/server/service/notification/DefaultNotificationSchedulerService.java b/application/src/main/java/org/thingsboard/server/service/notification/DefaultNotificationSchedulerService.java
index 171da618f4..eeeb1f0a7a 100644
--- a/application/src/main/java/org/thingsboard/server/service/notification/DefaultNotificationSchedulerService.java
+++ b/application/src/main/java/org/thingsboard/server/service/notification/DefaultNotificationSchedulerService.java
@@ -41,6 +41,7 @@ import org.thingsboard.server.service.executors.NotificationExecutorService;
 import org.thingsboard.server.service.partition.AbstractPartitionBasedService;
 
 import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
 import java.util.Collections;
 import java.util.HashSet;
 import java.util.List;
@@ -66,6 +67,7 @@ public class DefaultNotificationSchedulerService extends AbstractPartitionBasedS
 
     private final Map<NotificationRequestId, ScheduledRequestMetadata> scheduledNotificationRequests = new ConcurrentHashMap<>();
 
+    @Override
     @PostConstruct
     public void init() {
         super.init();
@@ -167,6 +169,13 @@ public class DefaultNotificationSchedulerService extends AbstractPartitionBasedS
         return "notifications-scheduler";
     }
 
+    @Override
+    @PreDestroy
+    public void stop() {
+        super.stop();
+        scheduler.shutdownNow();
+    }
+
     @Data
     private static class ScheduledRequestMetadata {
         private final TenantId tenantId;