Shutdown writer after transaction commit

Author: pnv1

ydb-cookbook/src/main/java/tech/ydb/examples/topic/transactions/TransactionWriteSync.java

@@ -1,5 +1,6 @@
 package tech.ydb.examples.topic.transactions;
 
+import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
@@ -8,6 +9,7 @@
 import org.slf4j.LoggerFactory;
 import tech.ydb.common.transaction.TxMode;
 import tech.ydb.core.Result;
+import tech.ydb.core.Status;
 import tech.ydb.core.grpc.GrpcTransport;
 import tech.ydb.examples.SimpleExample;
 import tech.ydb.query.QueryClient;
@@ -108,6 +110,9 @@ private void tableAndTopicWithinTransaction(TopicClient topicClient, QueryClient
             // flush to wait until the message will reach the server before committing transaction
             writer.flush();
 
+            // Commit transaction
+            CompletableFuture<Status> commitStatus = transaction.commit().thenApply(Result::getStatus);
+
             // Shutdown writer
             try {
                 writer.shutdown(SHUTDOWN_TIMEOUT_SECONDS, TimeUnit.SECONDS);
@@ -119,7 +124,7 @@ private void tableAndTopicWithinTransaction(TopicClient topicClient, QueryClient
             }
 
             // Return commit status to SessionRetryContext function
-            return transaction.commit().thenApply(Result::getStatus);
+            return commitStatus;
         }).join().expectSuccess("Couldn't read from table and write to topic in transaction");
     }