greg6775/Hadder
1
0
Fork 0
Code Issues 9 Pull requests 5 Projects Releases 54 Packages Wiki Activity Actions

Prevent error in getByID function

Browse source
This commit is contained in:
Greg 2021-03-27 12:26:52 +01:00
parent 5c5a53fdd2
commit b308a8bc25
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/main/java/one/bbn/hadder/db/Mongo.java
View file

@ -43,7 +43,11 @@ public class Mongo {
BasicDBObject whereQuery = new BasicDBObject(); BasicDBObject whereQuery = new BasicDBObject();
whereQuery.put(where, what); whereQuery.put(where, what);
FindIterable<Document> it = collection.find(whereQuery); FindIterable<Document> it = collection.find(whereQuery);
return it.cursor().next().get(column); try {
return it.cursor().next().get(column);
} catch (NoSuchElementException ignore) {
return null;
}
} }
public JSONObject getObjectByID(String collection, String id) { public JSONObject getObjectByID(String collection, String id) {
@ -133,8 +137,7 @@ public class Mongo {
} }
public boolean hasStarboardMessage(String message_id) { public boolean hasStarboardMessage(String message_id) {
this.getByID("stars", "id", message_id, "guild"); return this.getByID("stars", "id", message_id, "guild") != null;
return true;
} }
public void insertStarboardMessage(String message_id, String guild_id, String starboard_message_id) { public void insertStarboardMessage(String message_id, String guild_id, String starboard_message_id) {