Add onRemove for cleanup code

CitizensDev · Nov 23, 2022 · b28d689 · b28d689
1 parent 5408e70
commit b28d689
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/main/src/main/java/net/citizensnpcs/trait/ScoreboardTrait.java b/main/src/main/java/net/citizensnpcs/trait/ScoreboardTrait.java
@@ -96,6 +96,11 @@ public void onDespawn() {
         }
     }
 
+    @Override
+    public void onRemove() {
+        onDespawn();
+    }
+
     @Override
     public void onSpawn() {
         changed = true;

diff --git a/main/src/main/java/net/citizensnpcs/trait/SitTrait.java b/main/src/main/java/net/citizensnpcs/trait/SitTrait.java
@@ -35,6 +35,11 @@ public void onDespawn() {
         }
     }
 
+    @Override
+    public void onRemove() {
+        onDespawn();
+    }
+
     @Override
     public void run() {
         if (!npc.isSpawned() || !isSitting()) {

diff --git a/main/src/main/java/net/citizensnpcs/trait/versioned/BossBarTrait.java b/main/src/main/java/net/citizensnpcs/trait/versioned/BossBarTrait.java
@@ -105,6 +105,11 @@ public void onDespawn() {
         }
     }
 
+    @Override
+    public void onRemove() {
+        onDespawn();
+    }
+
     @Override
     public void run() {
         if (!npc.isSpawned())