From c12783eff23920bb7f4c39dc51f87a039736523b Mon Sep 17 00:00:00 2001 From: Fi0x Date: Wed, 7 Dec 2022 12:32:30 +0100 Subject: [PATCH] Fixed DB error --- .../com/fi0x/edct/logic/database/DBHandler.java | 6 +++--- .../edct/logic/filesystem/BlacklistHandler.java | 15 +++++++++++++-- 2 files changed, 16 insertions(+), 5 deletions(-) diff --git a/src/main/java/com/fi0x/edct/logic/database/DBHandler.java b/src/main/java/com/fi0x/edct/logic/database/DBHandler.java index f04511e..4213ef6 100644 --- a/src/main/java/com/fi0x/edct/logic/database/DBHandler.java +++ b/src/main/java/com/fi0x/edct/logic/database/DBHandler.java @@ -154,10 +154,10 @@ public static ArrayList getCommodityIDs(boolean onlyMissing, long minAv { while(results != null && results.next()) { - String name = results.getString("CommodityName"); - if(BlacklistHandler.getBlacklistedCommodities().contains(name)) + int id = results.getInt("InaraID"); + if(BlacklistHandler.isIDBlacklisted(id)) continue; - ids.add(results.getInt("InaraID")); + ids.add(id); } } catch(Exception e) { diff --git a/src/main/java/com/fi0x/edct/logic/filesystem/BlacklistHandler.java b/src/main/java/com/fi0x/edct/logic/filesystem/BlacklistHandler.java index d0628a3..41742e0 100644 --- a/src/main/java/com/fi0x/edct/logic/filesystem/BlacklistHandler.java +++ b/src/main/java/com/fi0x/edct/logic/filesystem/BlacklistHandler.java @@ -2,6 +2,7 @@ import com.fi0x.edct.Main; import com.fi0x.edct.logging.LogName; +import com.fi0x.edct.logic.database.DBHandler; import io.fi0x.javalogger.logging.Logger; import java.io.BufferedReader; @@ -15,6 +16,7 @@ public class BlacklistHandler { + private static ArrayList BLACKLISTED_IDS = null; public static ArrayList getBlacklistSystems() { ArrayList blacklistedStations = new ArrayList<>(); @@ -38,7 +40,14 @@ public static ArrayList getBlacklistSystems() return blacklistedStations; } - public static ArrayList getBlacklistedCommodities() + public static boolean isIDBlacklisted(int id) + { + if(BLACKLISTED_IDS == null) + generateBlacklistedCommodityIDs(); + return BLACKLISTED_IDS.contains(id); + } + + private static void generateBlacklistedCommodityIDs() { ArrayList blacklistedCommodities = new ArrayList<>(); try @@ -59,7 +68,9 @@ public static ArrayList getBlacklistedCommodities() Logger.log("Could not read the blacklist", LogName.WARNING, e, 992); } - return blacklistedCommodities; + BLACKLISTED_IDS = new ArrayList<>(); + for(String commodity : blacklistedCommodities) + BLACKLISTED_IDS.add(DBHandler.getCommodityIDByName(commodity)); } public static void addSystemToBlacklist(String systemName)