Provide access to the native Transaction object from AsyncTransaction

tomrussello · tomrussello · commit fe2065b74d6b · 2025-10-30T18:31:22.000+01:00
diff --git a/src/main/java/com/googlecode/objectify/cache/CachingAsyncTransaction.java b/src/main/java/com/googlecode/objectify/cache/CachingAsyncTransaction.java
@@ -6,6 +6,7 @@
 import com.google.cloud.datastore.Query;
 import com.google.cloud.datastore.QueryResults;
 import com.google.cloud.datastore.ReadOption;
+import com.google.cloud.datastore.Transaction;
 import com.google.cloud.datastore.Transaction.Response;
 import com.google.protobuf.ByteString;
 import com.googlecode.objectify.Result;
@@ -125,7 +126,12 @@ public void listenForCommit(final Runnable listener) {
 	public ByteString getTransactionHandle() {
 		return raw.getTransactionHandle();
 	}
-
+	
+	@Override
+	public Transaction getRawTransaction() {
+		return raw.getRawTransaction();
+	}
+	
 	@Override
 	public void runCommitListeners() {
 		((PrivateAsyncTransaction)raw).runCommitListeners();
diff --git a/src/main/java/com/googlecode/objectify/impl/AsyncTransaction.java b/src/main/java/com/googlecode/objectify/impl/AsyncTransaction.java
@@ -1,5 +1,6 @@
 package com.googlecode.objectify.impl;
 
+import com.google.cloud.datastore.Transaction;
 import com.google.cloud.datastore.Transaction.Response;
 import com.google.protobuf.ByteString;
 
@@ -19,4 +20,6 @@ public interface AsyncTransaction extends AsyncDatastoreReaderWriter {
 	void listenForCommit(final Runnable listener);
 
 	ByteString getTransactionHandle();
-}
+	
+	Transaction getRawTransaction();
+}
diff --git a/src/main/java/com/googlecode/objectify/impl/AsyncTransactionImpl.java b/src/main/java/com/googlecode/objectify/impl/AsyncTransactionImpl.java
@@ -66,7 +66,12 @@ public void listenForCommit(final Runnable listener) {
 	public ByteString getTransactionHandle() {
 		return transaction.getTransactionId();
 	}
-
+	
+	@Override
+	public Transaction getRawTransaction() {
+		return transaction;
+	}
+	
 	@Override
 	public void runCommitListeners() {
 		for (final Runnable listener : listeners) {
@@ -104,4 +109,4 @@ public boolean isActive() {
 	public void rollback() {
 		transaction.rollback();
 	}
-}
+}
