DataMap: List type compatibility improvements

xDec0de · Nov 2, 2024 · 0c29fab · 0c29fab
1 parent ed447be
commit 0c29fab
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/shared/src/main/java/net/codersky/mcutils/storage/DataMap.java b/shared/src/main/java/net/codersky/mcutils/storage/DataMap.java
@@ -130,7 +130,7 @@ public Set<String> getKeys(@NotNull String parent) {
 	@SuppressWarnings("unchecked")
 	private <T> T getFromMap(@NotNull Map<?, ?> source, @NotNull String key, @NotNull Class<T> type) {
 		final Object obj = source.get(key);
-		return obj != null && obj.getClass().equals(type) ? (T) obj : null;
+		return obj != null && type.isAssignableFrom(obj.getClass()) ? (T) obj : null;
 	}
 
 	// - Objects - //
@@ -154,9 +154,12 @@ public <T> T get(@NotNull String key, T def) {
 	@Nullable
 	@SuppressWarnings("unchecked")
 	public <T> List<T> getList(@NotNull String key, @NotNull Class<T> type) {
-		final Object obj = get(key, Object.class);
-		if (obj instanceof LinkedList<?> lst)
-			return lst.getClass().getTypeParameters()[0].getClass().equals(type) ? (List<T>) lst : null;
+		final Object obj = get(key, List.class);
+		if (obj instanceof List<?> lst) {
+			if (lst.isEmpty())
+				return List.of();
+			return lst.getFirst().getClass().equals(type) ? (List<T>) lst : null;
+		}
 		return null;
 	}