Allow accessing the ClientSession programmatively

quarkusio · Jan 2, 2023 · fa29a6e · fa29a6e
1 parent bd66b86
commit fa29a6e
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/...mmon/runtime/src/main/java/io/quarkus/mongodb/panache/common/runtime/MongoOperations.java b/...mmon/runtime/src/main/java/io/quarkus/mongodb/panache/common/runtime/MongoOperations.java
@@ -320,23 +320,27 @@ ClientSession getSession(Object entity) {
         return getSession(entity.getClass());
     }
 
-    ClientSession getSession(Class<?> entityClass) {
+    public ClientSession getSession(Class<?> entityClass) {
         ClientSession clientSession = null;
-        MongoEntity mongoEntity = entityClass.getAnnotation(MongoEntity.class);
         InstanceHandle<TransactionSynchronizationRegistry> instance = Arc.container()
                 .instance(TransactionSynchronizationRegistry.class);
         if (instance.isAvailable()) {
             TransactionSynchronizationRegistry registry = instance.get();
             if (registry.getTransactionStatus() == Status.STATUS_ACTIVE) {
                 clientSession = (ClientSession) registry.getResource(SESSION_KEY);
                 if (clientSession == null) {
+                    MongoEntity mongoEntity = entityClass == null ? null : entityClass.getAnnotation(MongoEntity.class);
                     return registerClientSession(mongoEntity, registry);
                 }
             }
         }
         return clientSession;
     }
 
+    public ClientSession getSession() {
+        return getSession(null);
+    }
+
     private ClientSession registerClientSession(MongoEntity mongoEntity,
             TransactionSynchronizationRegistry registry) {
         TransactionManager transactionManager = Arc.container().instance(TransactionManager.class).get();

diff --git a/...ons/panache/mongodb-panache/runtime/src/main/java/io/quarkus/mongodb/panache/Panache.java b/...ons/panache/mongodb-panache/runtime/src/main/java/io/quarkus/mongodb/panache/Panache.java
@@ -0,0 +1,15 @@
+package io.quarkus.mongodb.panache;
+
+import com.mongodb.session.ClientSession;
+import io.quarkus.mongodb.panache.runtime.JavaMongoOperations;
+
+public class Panache {
+
+    public static ClientSession getSession() {
+        return JavaMongoOperations.INSTANCE.getSession();
+    }
+
+    public static ClientSession getSession(Class entityClass) {
+        return JavaMongoOperations.INSTANCE.getSession(entityClass);
+    }
+}