From 12b8a14090daa5cddedfbc059f5d6f5ffe95494c Mon Sep 17 00:00:00 2001
From: evanchooly <jlee@antwerkz.com>
Date: Tue, 15 Oct 2024 00:03:01 -0400
Subject: [PATCH] bump the driver version to 5.2.0 add new method

---
 .../dev/morphia/transactions/SessionDatastore.java   | 12 ++++++++++++
 pom.xml                                              |  4 ++--
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/core/src/main/java/dev/morphia/transactions/SessionDatastore.java b/core/src/main/java/dev/morphia/transactions/SessionDatastore.java
index 13dec49e78a..7be97fa065b 100644
--- a/core/src/main/java/dev/morphia/transactions/SessionDatastore.java
+++ b/core/src/main/java/dev/morphia/transactions/SessionDatastore.java
@@ -13,6 +13,7 @@
 import com.mongodb.client.result.InsertManyResult;
 import com.mongodb.client.result.InsertOneResult;
 import com.mongodb.client.result.UpdateResult;
+import com.mongodb.internal.TimeoutContext;
 import com.mongodb.lang.NonNull;
 import com.mongodb.lang.Nullable;
 import com.mongodb.session.ServerSession;
@@ -44,6 +45,8 @@ public class SessionDatastore extends MorphiaDatastore implements MorphiaSession
 
     private final ClientSession session;
 
+    private TimeoutContext timeoutContext;
+
     /**
      * Creates a new session.
      *
@@ -275,6 +278,15 @@ public BsonDocument getClusterTime() {
         return session.getClusterTime();
     }
 
+    @Override
+    public TimeoutContext getTimeoutContext() {
+        return timeoutContext;
+    }
+
+    public void setTimeoutContext(TimeoutContext timeoutContext) {
+        this.timeoutContext = timeoutContext;
+    }
+
     @Override
     public void close() {
         session.close();
diff --git a/pom.xml b/pom.xml
index dc0d5b88552..3e341d53623 100644
--- a/pom.xml
+++ b/pom.xml
@@ -44,8 +44,8 @@
     <properties>
         <kotlin.version>2.0.21</kotlin.version>
         <awaitility.version>1.7.0</awaitility.version>
-        <driver.version>5.1.4</driver.version>
-        <driver.minor.version>5.1</driver.minor.version>
+        <driver.version>5.2.0</driver.version>
+        <driver.minor.version>5.2</driver.minor.version>
         <jackson.version>2.18.0</jackson.version>
         <json.assert.version>1.5.3</json.assert.version>
         <javapoet.version>1.13.0</javapoet.version>