From 39e93518ce1e3b79fd177742eb1087bdfe82973a Mon Sep 17 00:00:00 2001
From: vunamtien <vunamtien16@gmail.com>
Date: Tue, 5 Mar 2024 20:56:24 +0700
Subject: [PATCH] [JAVA-31501] Upgrade spring-data-mongodb to Spring Boot 3
 (#16037)

---
 persistence-modules/spring-data-mongodb/pom.xml |  5 +++--
 .../baeldung/config/MongoReactiveConfig.java    |  8 ++++++++
 .../MongoTransactionReactiveLiveTest.java       | 17 ++++++++++++++---
 3 files changed, 25 insertions(+), 5 deletions(-)

diff --git a/persistence-modules/spring-data-mongodb/pom.xml b/persistence-modules/spring-data-mongodb/pom.xml
index eec1fa31dc..df16be1396 100644
--- a/persistence-modules/spring-data-mongodb/pom.xml
+++ b/persistence-modules/spring-data-mongodb/pom.xml
@@ -8,9 +8,9 @@
 
     <parent>
         <groupId>com.baeldung</groupId>
-        <artifactId>parent-boot-2</artifactId>
+        <artifactId>parent-boot-3</artifactId>
         <version>0.0.1-SNAPSHOT</version>
-        <relativePath>../../parent-boot-2</relativePath>
+        <relativePath>../../parent-boot-3</relativePath>
     </parent>
 
     <dependencies>
@@ -78,6 +78,7 @@
         <mysema.maven.version>1.1.3</mysema.maven.version>
         <projectreactor.version>3.5.4</projectreactor.version>
         <embed.mongo.version>4.6.3</embed.mongo.version>
+        <start-class>com.baeldung.Main</start-class>
     </properties>
 
 </project>
\ No newline at end of file
diff --git a/persistence-modules/spring-data-mongodb/src/main/java/com/baeldung/config/MongoReactiveConfig.java b/persistence-modules/spring-data-mongodb/src/main/java/com/baeldung/config/MongoReactiveConfig.java
index b4042b5550..6ec84fca1f 100644
--- a/persistence-modules/spring-data-mongodb/src/main/java/com/baeldung/config/MongoReactiveConfig.java
+++ b/persistence-modules/spring-data-mongodb/src/main/java/com/baeldung/config/MongoReactiveConfig.java
@@ -1,11 +1,14 @@
 package com.baeldung.config;
 
+import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.data.mongodb.config.AbstractReactiveMongoConfiguration;
 import org.springframework.data.mongodb.repository.config.EnableReactiveMongoRepositories;
 
 import com.mongodb.reactivestreams.client.MongoClient;
 import com.mongodb.reactivestreams.client.MongoClients;
+import org.springframework.transaction.ReactiveTransactionManager;
+import org.springframework.transaction.reactive.TransactionalOperator;
 
 @Configuration
 @EnableReactiveMongoRepositories(basePackages = "com.baeldung.reactive.repository")
@@ -20,4 +23,9 @@ public class MongoReactiveConfig extends AbstractReactiveMongoConfiguration {
     protected String getDatabaseName() {
         return "reactive";
     }
+
+    @Bean
+    public TransactionalOperator transactionalOperator(ReactiveTransactionManager reactiveTransactionManager) {
+        return TransactionalOperator.create(reactiveTransactionManager);
+    }
 }
diff --git a/persistence-modules/spring-data-mongodb/src/test/java/com/baeldung/transaction/MongoTransactionReactiveLiveTest.java b/persistence-modules/spring-data-mongodb/src/test/java/com/baeldung/transaction/MongoTransactionReactiveLiveTest.java
index 3fc8dcf977..209f5cbbee 100644
--- a/persistence-modules/spring-data-mongodb/src/test/java/com/baeldung/transaction/MongoTransactionReactiveLiveTest.java
+++ b/persistence-modules/spring-data-mongodb/src/test/java/com/baeldung/transaction/MongoTransactionReactiveLiveTest.java
@@ -6,11 +6,14 @@ import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.mongodb.core.ReactiveMongoOperations;
+import org.springframework.data.mongodb.core.ReactiveMongoTemplate;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 
 import com.baeldung.config.MongoReactiveConfig;
 import com.baeldung.model.User;
+import org.springframework.transaction.reactive.TransactionalOperator;
+import reactor.core.publisher.Mono;
 
 /**
  * 
@@ -25,6 +28,12 @@ public class MongoTransactionReactiveLiveTest {
     @Autowired
     private ReactiveMongoOperations reactiveOps;
 
+    @Autowired
+    private TransactionalOperator transactionalOperator;
+
+    @Autowired
+    private ReactiveMongoTemplate mongoTemplate;
+
     @Before
     public void testSetup() {
         if (!reactiveOps.collectionExists(User.class)
@@ -45,9 +54,11 @@ public class MongoTransactionReactiveLiveTest {
     public void whenPerformTransaction_thenSuccess() {
         User user1 = new User("Jane", 23);
         User user2 = new User("John", 34);
-        reactiveOps.inTransaction()
-            .execute(action -> action.insert(user1)
-                .then(action.insert(user2)));
+
+        Mono<User> saveEntity1 = mongoTemplate.save(user1);
+        Mono<User> saveEntity2 = mongoTemplate.save(user2);
+
+        saveEntity1.then(saveEntity2).then().as(transactionalOperator::transactional);
     }
     
 }