Improve initialization logic

bukkit/src/main/java/net/william278/husksync/BukkitHuskSync.java

@@ -41,6 +41,7 @@ import java.io.File;
 import java.io.IOException;
 import java.util.*;
 import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.logging.Level;
 import java.util.stream.Collectors;
 
@@ -75,104 +76,75 @@ public class BukkitHuskSync extends JavaPlugin implements HuskSync {
 
     @Override
     public void onEnable() {
-        // Process initialization stages
+        // Initialize HuskSync
-        CompletableFuture.supplyAsync(() -> {
+        final AtomicBoolean initialized = new AtomicBoolean(true);
-
+        try {
             // Set the logging adapter and resource reader
             this.logger = new BukkitLogger(this.getLogger());
             this.resourceReader = new BukkitResourceReader(this);
 
             // Load settings and locales
             getLoggingAdapter().log(Level.INFO, "Loading plugin configuration settings & locales...");
-            return reload().thenApply(loadedSettings -> {
+            initialized.set(reload().join());
-                if (loadedSettings) {
+            if (initialized.get()) {
                 logger.showDebugLogs(settings.getBooleanValue(Settings.ConfigOption.DEBUG_LOGGING));
                 getLoggingAdapter().log(Level.INFO, "Successfully loaded plugin configuration settings & locales");
             } else {
-                    getLoggingAdapter().log(Level.SEVERE, "Failed to load plugin configuration settings and/or locales");
+                throw new HuskSyncInitializationException("Failed to load plugin configuration settings and/or locales");
             }
-                return loadedSettings;
+
-            }).join();
-        }).thenApply(succeeded -> {
             // Prepare data adapter
-            if (succeeded) {
             if (settings.getBooleanValue(Settings.ConfigOption.SYNCHRONIZATION_COMPRESS_DATA)) {
                 dataAdapter = new CompressedDataAdapter();
             } else {
                 dataAdapter = new JsonDataAdapter();
             }
-            }
+
-            return succeeded;
-        }).thenApply(succeeded -> {
             // Prepare event cannon
-            if (succeeded) {
             eventCannon = new BukkitEventCannon();
-            }
+
-            return succeeded;
-        }).thenApply(succeeded -> {
             // Prepare data editor
-            if (succeeded) {
             dataEditor = new DataEditor(locales);
-            }
+
-            return succeeded;
-        }).thenApply(succeeded -> {
             // Prepare migrators
-            if (succeeded) {
             availableMigrators = new ArrayList<>();
             availableMigrators.add(new LegacyMigrator(this));
             final Plugin mySqlPlayerDataBridge = Bukkit.getPluginManager().getPlugin("MySqlPlayerDataBridge");
             if (mySqlPlayerDataBridge != null) {
                 availableMigrators.add(new MpdbMigrator(this, mySqlPlayerDataBridge));
             }
-            }
+
-            return succeeded;
+            // Prepare database connection
-        }).thenApply(succeeded -> {
-            // Establish connection to the database
-            if (succeeded) {
             this.database = new MySqlDatabase(settings, resourceReader, logger, dataAdapter, eventCannon);
             getLoggingAdapter().log(Level.INFO, "Attempting to establish connection to the database...");
-                final CompletableFuture<Boolean> databaseConnectFuture = new CompletableFuture<>();
+            initialized.set(this.database.initialize());
-                Bukkit.getScheduler().runTask(this, () -> {
+            if (initialized.get()) {
-                    final boolean initialized = this.database.initialize();
-                    if (!initialized) {
-                        getLoggingAdapter().log(Level.SEVERE, "Failed to establish a connection to the database. " + "Please check the supplied database credentials in the config file");
-                        databaseConnectFuture.completeAsync(() -> false);
-                        return;
-                    }
                 getLoggingAdapter().log(Level.INFO, "Successfully established a connection to the database");
-                    databaseConnectFuture.completeAsync(() -> true);
+            } else {
-                });
+                throw new HuskSyncInitializationException("Failed to establish a connection to the database. " +
-                return databaseConnectFuture.join();
+                                                          "Please check the supplied database credentials in the config file");
             }
-            return false;
+
-        }).thenApply(succeeded -> {
+            // Prepare redis connection
-            // Establish connection to the Redis server
-            if (succeeded) {
             this.redisManager = new RedisManager(this);
             getLoggingAdapter().log(Level.INFO, "Attempting to establish connection to the Redis server...");
-                return this.redisManager.initialize().thenApply(initialized -> {
+            initialized.set(this.redisManager.initialize().join());
-                    if (!initialized) {
+            if (initialized.get()) {
-                        getLoggingAdapter().log(Level.SEVERE, "Failed to establish a connection to the Redis server. " + "Please check the supplied Redis credentials in the config file");
-                        return false;
-                    }
                 getLoggingAdapter().log(Level.INFO, "Successfully established a connection to the Redis server");
-                    return true;
+            } else {
-                }).join();
+                throw new HuskSyncInitializationException("Failed to establish a connection to the Redis server. " +
+                                                          "Please check the supplied Redis credentials in the config file");
             }
-            return false;
+
-        }).thenApply(succeeded -> {
             // Register events
-            if (succeeded) {
             getLoggingAdapter().log(Level.INFO, "Registering events...");
             this.eventListener = new BukkitEventListener(this);
             getLoggingAdapter().log(Level.INFO, "Successfully registered events listener");
-            }
+
-            return succeeded;
-        }).thenApply(succeeded -> {
             // Register permissions
-            if (succeeded) {
             getLoggingAdapter().log(Level.INFO, "Registering permissions & commands...");
-                Arrays.stream(Permission.values()).forEach(permission -> getServer().getPluginManager().addPermission(new org.bukkit.permissions.Permission(permission.node, switch (permission.defaultAccess) {
+            Arrays.stream(Permission.values()).forEach(permission -> getServer().getPluginManager()
+                    .addPermission(new org.bukkit.permissions.Permission(permission.node, switch (permission.defaultAccess) {
                 case EVERYONE -> PermissionDefault.TRUE;
                 case NOBODY -> PermissionDefault.FALSE;
                 case OPERATORS -> PermissionDefault.OP;
@@ -186,36 +158,34 @@ public class BukkitHuskSync extends JavaPlugin implements HuskSync {
                 }
             }
             getLoggingAdapter().log(Level.INFO, "Successfully registered permissions & commands");
-            }
+
-            return succeeded;
+            // Hook into plan
-        }).thenApply(succeeded -> {
+            if (Bukkit.getPluginManager().getPlugin("Plan") != null) {
-            if (succeeded && Bukkit.getPluginManager().getPlugin("Plan") != null) {
                 getLoggingAdapter().log(Level.INFO, "Enabling Plan integration...");
                 new PlanHook(database, logger).hookIntoPlan();
                 getLoggingAdapter().log(Level.INFO, "Plan integration enabled!");
             }
-            return succeeded;
+
-        }).thenApply(succeeded -> {
             // Check for updates
-            if (succeeded && settings.getBooleanValue(Settings.ConfigOption.CHECK_FOR_UPDATES)) {
+            if (settings.getBooleanValue(Settings.ConfigOption.CHECK_FOR_UPDATES)) {
                 getLoggingAdapter().log(Level.INFO, "Checking for updates...");
-                new UpdateChecker(getPluginVersion(), getLoggingAdapter()).logToConsole();
+                CompletableFuture.runAsync(() -> new UpdateChecker(getPluginVersion(), getLoggingAdapter()).logToConsole());
-            }
+            }
-            return succeeded;
+        } catch (HuskSyncInitializationException exception) {
-        }).thenAccept(succeeded -> {
+            getLoggingAdapter().log(Level.SEVERE, exception.getMessage());
-            // Handle failed initialization
+            initialized.set(false);
-            if (!succeeded) {
+        } catch (Exception exception) {
+            getLoggingAdapter().log(Level.SEVERE, "An unhandled exception occurred initializing HuskSync!", exception);
+            initialized.set(false);
+        } finally {
+            // Validate initialization
+            if (initialized.get()) {
+                getLoggingAdapter().log(Level.INFO, "Successfully enabled HuskSync v" + getPluginVersion());
+            } else {
                 getLoggingAdapter().log(Level.SEVERE, "Failed to initialize HuskSync. The plugin will now be disabled");
                 getServer().getPluginManager().disablePlugin(this);
-            } else {
-                getLoggingAdapter().log(Level.INFO, "Successfully enabled HuskSync v" + getPluginVersion());
             }
-        }).exceptionally(exception -> {
+        }
-            getLoggingAdapter().log(Level.SEVERE, "An exception occurred initializing HuskSync. (" + exception.getMessage() + ") The plugin will now be disabled.");
-            exception.printStackTrace();
-            getServer().getPluginManager().disablePlugin(this);
-            return null;
-        });
     }
 
     @Override

common/src/main/java/net/william278/husksync/HuskSyncInitializationException.java

@@ -0,0 +1,12 @@
+package net.william278.husksync;
+
+import org.jetbrains.annotations.NotNull;
+
+/**
+ * Indicates an exception occurred while initialising the HuskSync plugin
+ */
+public class HuskSyncInitializationException extends RuntimeException {
+    public HuskSyncInitializationException(@NotNull String message) {
+        super(message);
+    }
+}

common/src/main/java/net/william278/husksync/database/MySqlDatabase.java

@@ -114,7 +114,7 @@ public class MySqlDatabase extends Database {
                 getLogger().log(Level.SEVERE, "Failed to perform database setup: " + e.getMessage());
             }
         } catch (Exception e) {
-            e.printStackTrace();
+            getLogger().log(Level.SEVERE, "An unhandled exception occurred during database setup!", e);
         }
         return false;
     }