Add withDocumentClass() to the ReactiveMongoCollection

Author: loicmathieu

extensions/mongodb-client/runtime/src/main/java/io/quarkus/mongodb/impl/ReactiveMongoCollectionImpl.java

@@ -820,4 +820,9 @@ public Uni<Void> renameCollection(ClientSession clientSession, MongoNamespace ne
     public CodecRegistry getCodecRegistry() {
         return collection.getCodecRegistry();
     }
+
+    @Override
+    public <NewTDocument> ReactiveMongoCollection<NewTDocument> withDocumentClass(Class<NewTDocument> clazz) {
+        return new ReactiveMongoCollectionImpl<>(this.collection.withDocumentClass(clazz));
+    }
 }

extensions/mongodb-client/runtime/src/main/java/io/quarkus/mongodb/reactive/ReactiveMongoCollection.java

@@ -2,6 +2,7 @@
 
 import java.util.List;
 
+import com.mongodb.client.MongoCollection;
 import org.bson.Document;
 import org.bson.codecs.configuration.CodecRegistry;
 import org.bson.conversions.Bson;
@@ -1489,4 +1490,13 @@ Uni<Void> renameCollection(ClientSession clientSession, MongoNamespace newCollec
      * @return the codec registry
      */
     CodecRegistry getCodecRegistry();
+
+    /**
+     * Create a new ReactiveMongoCollection instance with a different default class to cast any documents returned from the database into..
+     *
+     * @param clazz the default class to cast any documents returned from the database into.
+     * @param <NewTDocument> The type that the new collection will encode documents from and decode documents to
+     * @return a new MongoCollection instance with the different default class
+     */
+    <NewTDocument> ReactiveMongoCollection<NewTDocument> withDocumentClass(Class<NewTDocument> clazz);
 }