notify cluster formation changed nug fixes

components/heartbeat-management/io.entgra.device.mgt.core.server.bootup.heartbeat.beacon/src/main/java/io/entgra/device/mgt/core/server/bootup/heartbeat/beacon/internal/HeartBeatBeaconComponent.java

@@ -61,10 +61,6 @@ public class HeartBeatBeaconComponent {
                 DataSourceConfig dsConfig = HeartBeatBeaconConfig.getInstance().getDataSourceConfig();
                 HeartBeatBeaconDAOFactory.init(dsConfig);
 
-                //Setting up executors to notify heart beat status */
-                HeartBeatExecutor.setUpNotifiers(HeartBeatBeaconUtils.getServerDetails());
-            }
-
                 ClusterFormationChangedNotifierRepository clusterFormationChangedNotifierRepository
                         = new ClusterFormationChangedNotifierRepository();
                 List<String> notifiers = HeartBeatBeaconConfig.getInstance().getNotifiers();
@@ -76,6 +72,12 @@ public class HeartBeatBeaconComponent {
                 HeartBeatBeaconDataHolder.getInstance().setClusterFormationChangedNotifierRepository(
                         clusterFormationChangedNotifierRepository);
 
+                //Setting up executors to notify heart beat status */
+                HeartBeatExecutor.setUpNotifiers(HeartBeatBeaconUtils.getServerDetails());
+            }
+
+
+
             if (log.isDebugEnabled()) {
                 log.debug("Heart Beat Notifier bundle has been successfully initialized");
             }

components/heartbeat-management/io.entgra.device.mgt.core.server.bootup.heartbeat.beacon/src/main/java/io/entgra/device/mgt/core/server/bootup/heartbeat/beacon/service/HeartBeatManagementServiceImpl.java

@@ -261,7 +261,9 @@ public class HeartBeatManagementServiceImpl implements HeartBeatManagementServic
     public void notifyClusterFormationChanged(int elapsedTimeInSeconds) throws HeartBeatManagementException {
         if (HeartBeatBeaconConfig.getInstance().isEnabled()) {
             try {
+                HeartBeatBeaconDAOFactory.beginTransaction();
                 Map<String, ServerContext> servers = heartBeatDAO.getActiveServerDetails(elapsedTimeInSeconds);
+                HeartBeatBeaconDAOFactory.commitTransaction();
                 if (servers != null && !servers.isEmpty()) {
                     String serverUUID = HeartBeatBeaconDataHolder.getInstance().getLocalServerUUID();
                     ServerContext serverContext = servers.get(serverUUID);
@@ -289,6 +291,13 @@ public class HeartBeatManagementServiceImpl implements HeartBeatManagementServic
                 String msg = "Error occurred while notifyClusterFormationChanged.";
                 log.error(msg, e);
                 throw new HeartBeatManagementException(msg, e);
+            } catch (TransactionManagementException e) {
+                HeartBeatBeaconDAOFactory.rollbackTransaction();
+                String msg = "Error occurred while electing candidate for dynamic task execution. Issue in opening a connection to the underlying data source";
+                log.error(msg, e);
+                throw new HeartBeatManagementException(msg, e);
+            } finally {
+                HeartBeatBeaconDAOFactory.closeConnection();
             }
         } else {
             String msg = "Heart Beat Configuration Disabled. Error while notifyClusterFormationChanged.";