Improve the proposed changed

Author: stIncMale

src/integrationTest/java/com/mongodb/hibernate/jdbc/MongoPreparedStatementIntegrationTests.java

@@ -17,8 +17,7 @@
 package com.mongodb.hibernate.jdbc;
 
 import static com.mongodb.hibernate.internal.MongoConstants.ID_FIELD_NAME;
-import static com.mongodb.hibernate.jdbc.MongoStatementIntegrationTests.doAndTerminateTransaction;
-import static com.mongodb.hibernate.jdbc.MongoStatementIntegrationTests.doWithSpecifiedAutoCommit;
+import static com.mongodb.hibernate.jdbc.MongoStatementIntegrationTests.doWorkWithSpecifiedAutoCommit;
 import static com.mongodb.hibernate.jdbc.MongoStatementIntegrationTests.insertTestData;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertAll;
@@ -30,7 +29,6 @@
 
 import com.mongodb.client.MongoCollection;
 import com.mongodb.client.model.Sorts;
-import com.mongodb.hibernate.jdbc.MongoStatementIntegrationTests.SqlExecutable;
 import com.mongodb.hibernate.junit.InjectMongoCollection;
 import com.mongodb.hibernate.junit.MongoExtension;
 import java.math.BigDecimal;
@@ -409,11 +407,6 @@ private void assertExecuteUpdate(
     }
 
     private void doWorkAwareOfAutoCommit(Work work) {
-        session.doWork(connection -> doAwareOfAutoCommit(connection, () -> work.execute(connection)));
-    }
-
-    void doAwareOfAutoCommit(Connection connection, SqlExecutable work) throws SQLException {
-        doWithSpecifiedAutoCommit(
-                autoCommit, connection, autoCommit ? work : () -> doAndTerminateTransaction(connection, work));
+        doWorkWithSpecifiedAutoCommit(autoCommit, session, work);
     }
 }

src/integrationTest/java/com/mongodb/hibernate/jdbc/MongoStatementIntegrationTests.java

@@ -60,7 +60,7 @@ class MongoStatementIntegrationTests {
     private Session session;
 
     @Parameter
-    boolean autoCommit;
+    private boolean autoCommit;
 
     @BeforeAll
     static void beforeAll() {
@@ -282,15 +282,20 @@ static void insertTestData(Session session, String insertMql) {
     }
 
     private void doWorkAwareOfAutoCommit(Work work) {
-        session.doWork(connection -> doAwareOfAutoCommit(connection, () -> work.execute(connection)));
+        doWorkWithSpecifiedAutoCommit(autoCommit, session, work);
     }
 
-    void doAwareOfAutoCommit(Connection connection, SqlExecutable work) throws SQLException {
-        doWithSpecifiedAutoCommit(
-                autoCommit, connection, autoCommit ? work : () -> doAndTerminateTransaction(connection, work));
+    static void doWorkWithSpecifiedAutoCommit(boolean autoCommit, Session session, Work work) {
+        session.doWork(connection -> {
+            SqlExecutable executable = () -> work.execute(connection);
+            doWithSpecifiedAutoCommit(
+                    autoCommit,
+                    connection,
+                    autoCommit ? executable : () -> doAndTerminateTransaction(connection, executable));
+        });
     }
 
-    static void doWithSpecifiedAutoCommit(boolean autoCommit, Connection connection, SqlExecutable work)
+    private static void doWithSpecifiedAutoCommit(boolean autoCommit, Connection connection, SqlExecutable work)
             throws SQLException {
         var originalAutoCommit = connection.getAutoCommit();
         connection.setAutoCommit(autoCommit);
@@ -301,7 +306,8 @@ static void doWithSpecifiedAutoCommit(boolean autoCommit, Connection connection,
         }
     }
 
-    static void doAndTerminateTransaction(Connection connectionNoAutoCommit, SqlExecutable work) throws SQLException {
+    private static void doAndTerminateTransaction(Connection connectionNoAutoCommit, SqlExecutable work)
+            throws SQLException {
         Throwable primaryException = null;
         try {
             work.execute();
@@ -320,7 +326,7 @@ static void doAndTerminateTransaction(Connection connectionNoAutoCommit, SqlExec
         }
     }
 
-    interface SqlExecutable {
+    private interface SqlExecutable {
         void execute() throws SQLException;
     }
 }