diff --git a/src/main/java/net/frozenorb/apiv3/model/BannedAsn.java b/src/main/java/net/frozenorb/apiv3/model/BannedAsn.java index 1440070..2a4a4ba 100644 --- a/src/main/java/net/frozenorb/apiv3/model/BannedAsn.java +++ b/src/main/java/net/frozenorb/apiv3/model/BannedAsn.java @@ -82,6 +82,8 @@ public final class BannedAsn { } public void insert(SingleResultCallback callback) { + bannedAsnCache.add(this); + bannedAsnIdCache.put(id, this); bannedAsnsCollection.insertOne(this, callback); } @@ -90,6 +92,8 @@ public final class BannedAsn { } public void delete(SingleResultCallback callback) { + bannedAsnCache.remove(this); + bannedAsnIdCache.remove(id); bannedAsnsCollection.deleteOne(new Document("_id", id), callback); } diff --git a/src/main/java/net/frozenorb/apiv3/model/Rank.java b/src/main/java/net/frozenorb/apiv3/model/Rank.java index 8aeecb7..ec1e86b 100644 --- a/src/main/java/net/frozenorb/apiv3/model/Rank.java +++ b/src/main/java/net/frozenorb/apiv3/model/Rank.java @@ -78,10 +78,14 @@ public final class Rank { } public void insert(SingleResultCallback callback) { + rankCache.add(this); + rankIdCache.put(id, this); ranksCollection.insertOne(this, callback); } public void delete(SingleResultCallback callback) { + rankCache.remove(this); + rankIdCache.remove(id); ranksCollection.deleteOne(new Document("_id", id), callback); } diff --git a/src/main/java/net/frozenorb/apiv3/model/Server.java b/src/main/java/net/frozenorb/apiv3/model/Server.java index ea6a072..b7cca91 100644 --- a/src/main/java/net/frozenorb/apiv3/model/Server.java +++ b/src/main/java/net/frozenorb/apiv3/model/Server.java @@ -89,6 +89,8 @@ public final class Server { } public void insert(SingleResultCallback callback) { + serverCache.add(this); + serverIdCache.put(id, this); serversCollection.insertOne(this, callback); } @@ -97,6 +99,8 @@ public final class Server { } public void delete(SingleResultCallback callback) { + serverCache.remove(this); + serverIdCache.remove(id); serversCollection.deleteOne(new Document("_id", id), callback); } diff --git a/src/main/java/net/frozenorb/apiv3/model/ServerGroup.java b/src/main/java/net/frozenorb/apiv3/model/ServerGroup.java index f1b286a..b64d812 100644 --- a/src/main/java/net/frozenorb/apiv3/model/ServerGroup.java +++ b/src/main/java/net/frozenorb/apiv3/model/ServerGroup.java @@ -77,6 +77,8 @@ public final class ServerGroup { } public void insert(SingleResultCallback callback) { + serverGroupCache.add(this); + serverGroupIdCache.put(id, this); serverGroupsCollection.insertOne(this, callback); } @@ -85,6 +87,8 @@ public final class ServerGroup { } public void delete(SingleResultCallback callback) { + serverGroupCache.remove(this); + serverGroupIdCache.remove(id); serverGroupsCollection.deleteOne(new Document("_id", id), callback); }