Add MagicCollection#insertOneReturnId(T)

srnyx · Sep 29, 2024 · 01894b1 · 01894b1
1 parent 0019c13
commit 01894b1
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/main/java/xyz/srnyx/magicmongo/MagicCollection.java b/src/main/java/xyz/srnyx/magicmongo/MagicCollection.java
@@ -15,6 +15,7 @@
 import org.bson.Document;
 import org.bson.codecs.configuration.CodecRegistry;
 import org.bson.conversions.Bson;
+import org.bson.types.ObjectId;
 
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -85,6 +86,18 @@ public List<T> findMany(@NotNull Bson filter) {
         return find(filter).into(new ArrayList<>());
     }
 
+    /**
+     * Inserts a document in the collection and returns the inserted document's ID
+     *
+     * @param   t   the document to insert
+     *
+     * @return      the ID of the inserted document
+     */
+    @NotNull
+    public ObjectId insertOneReturnId(@NotNull T t) {
+        return Objects.requireNonNull(insertOne(t).getInsertedId()).asObjectId().getValue();
+    }
+
     /**
      * Upserts a document in the collection
      *