Added createTasks

Auxilor · May 16, 2023 · 7edc00d · 7edc00d
1 parent a51bad0
commit 7edc00d
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/eco-api/src/main/java/com/willfp/eco/core/EcoPlugin.java b/eco-api/src/main/java/com/willfp/eco/core/EcoPlugin.java
@@ -164,6 +164,11 @@ public abstract class EcoPlugin extends JavaPlugin implements PluginLike, Regist
      */
     private final ListMap<LifecyclePosition, Runnable> afterLoad = new ListMap<>();
 
+    /**
+     * The tasks to run on task creation.
+     */
+    private final ListMap<LifecyclePosition, Runnable> createTasks = new ListMap<>();
+
     /**
      * Create a new plugin.
      * <p>
@@ -632,6 +637,10 @@ public final void reload(final boolean cancelTasks) {
 
         this.handleLifecycle(this.onReload, this::handleReload);
 
+        if (cancelTasks) {
+            this.handleLifecycle(this.createTasks, this::createTasks);
+        }
+
         for (Extension extension : this.extensionLoader.getLoadedExtensions()) {
             extension.handleReload();
         }
@@ -745,6 +754,15 @@ protected void handleReload() {
 
     }
 
+    /**
+     * The plugin-specific code to create tasks.
+     * <p>
+     * Override when needed.
+     */
+    protected void createTasks() {
+
+    }
+
     /**
      * The plugin-specific code to be executed after the server is up.
      * <p>