StoredEntityData#remove

api/src/main/java/com/viaversion/viaversion/api/data/entity/StoredEntityData.java

@@ -51,6 +51,15 @@ public interface StoredEntityData {
      */
     @Nullable <T> T get(Class<T> objectClass);
 
+    /**
+     * Removes and returns an object from the storage if present.
+     *
+     * @param objectClass class of the object to remove
+     * @param <T>         object type
+     * @return removed object if present
+     */
+    @Nullable <T> T remove(Class<T> objectClass);
+
     /**
      * Stores an object based on its class.
      *

common/src/main/java/com/viaversion/viaversion/data/entity/StoredEntityImpl.java

@@ -42,6 +42,11 @@ public final class StoredEntityImpl implements StoredEntityData {
         return (T) storedObjects.get(objectClass);
     }
 
+    @Override
+    public <T> @Nullable T remove(Class<T> objectClass) {
+        return (T) storedObjects.remove(objectClass);
+    }
+
     @Override
     public boolean has(Class<?> objectClass) {
         return storedObjects.containsKey(objectClass);