Fixed deleteKeyInAll logic

Signed-off-by: DevDrizzy <[email protected]>
RefineDevelopment · Mar 17, 2024 · df4f1c5 · df4f1c5
1 parent 28754fb
commit df4f1c5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/java/xyz/refinedev/api/storage/MongoStorage.java b/src/main/java/xyz/refinedev/api/storage/MongoStorage.java
@@ -123,8 +123,11 @@ public CompletableFuture<Integer> deleteKeyInAll(String key) {
             for ( Document document : collection.find() ) {
                 if (document == null) continue;
 
-                document.remove(key);
-                deleteCount++;
+                if (document.get(key) != null) {
+                    document.remove(key);
+                    deleteCount++;
+                }
+                collection.replaceOne(Filters.eq("_id", document.get("_id")), document);
             }
             return deleteCount;
         });