From 237fb0b83944fc2fd4960bdc8a8b518a8ca9aaaf Mon Sep 17 00:00:00 2001 From: shedaniel Date: Sun, 8 Dec 2019 12:38:50 +0800 Subject: Warn on no container info --- .../java/me/shedaniel/rei/server/ContainerInfoHandler.java | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'src/main/java/me/shedaniel/rei/server/ContainerInfoHandler.java') diff --git a/src/main/java/me/shedaniel/rei/server/ContainerInfoHandler.java b/src/main/java/me/shedaniel/rei/server/ContainerInfoHandler.java index 51f98cb2e..3ea2dc4f4 100644 --- a/src/main/java/me/shedaniel/rei/server/ContainerInfoHandler.java +++ b/src/main/java/me/shedaniel/rei/server/ContainerInfoHandler.java @@ -12,19 +12,19 @@ import net.minecraft.util.Identifier; import java.util.Map; public class ContainerInfoHandler { - private static final Map, ContainerInfo>> containerInfoMap = Maps.newHashMap(); + private static final Map, ContainerInfo>> containerInfoMap = Maps.newHashMap(); public static void registerContainerInfo(Identifier category, ContainerInfo containerInfo) { if (!containerInfoMap.containsKey(category)) - containerInfoMap.put(category, Maps.newHashMap()); - containerInfoMap.get(category).put(containerInfo.getContainerClass(), containerInfo); + containerInfoMap.put(category.toString(), Maps.newHashMap()); + containerInfoMap.get(category.toString()).put(containerInfo.getContainerClass(), containerInfo); } public static boolean isCategoryHandled(Identifier category) { - return containerInfoMap.containsKey(category) && !containerInfoMap.get(category).isEmpty(); + return containerInfoMap.containsKey(category.toString()) && !containerInfoMap.get(category.toString()).isEmpty(); } public static ContainerInfo getContainerInfo(Identifier category, Class containerClass) { - return containerInfoMap.get(category).get(containerClass); + return isCategoryHandled(category) ? containerInfoMap.get(category.toString()).get(containerClass) : null; } } -- cgit