DisplayElement now cleans itself up properly

ArcticRaven · ArcticRaven · commit 19645d4cecf4 · 2024-08-27T17:27:57.000-04:00
diff --git a/src/main/java/dev/arctic/interactivemenuapi/objects/elements/DisplayElement.java b/src/main/java/dev/arctic/interactivemenuapi/objects/elements/DisplayElement.java
@@ -63,6 +63,12 @@ public void setDisplayItem(ItemStack displayItem) {
         displayEntity.setItemStack(displayItem);
     }
 
+    @Override
+    public void cleanup() {
+        super.cleanup();
+        displayEntity.remove();
+    }
+
     /**
      * Gets the display entity associated with this element.
      *

Original file line number	Diff line number	Diff line change
`@@ -63,6 +63,12 @@ public void setDisplayItem(ItemStack displayItem) {`
`63`	`63`	`displayEntity.setItemStack(displayItem);`
`64`	`64`	`}`
`65`	`65`
	`66`	`+ @Override`
	`67`	`+ public void cleanup() {`
	`68`	`+ super.cleanup();`
	`69`	`+ displayEntity.remove();`
	`70`	`+ }`
	`71`	`+`
`66`	`72`	`/**`
`67`	`73`	`* Gets the display entity associated with this element.`
`68`	`74`	`*`