return the datastore service from Batch and Transation

googleapis · Feb 10, 2015 · 0ae2558 · 0ae2558
1 parent beac379
commit 0ae2558
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/src/main/java/com/google/gcloud/datastore/Batch.java b/src/main/java/com/google/gcloud/datastore/Batch.java
@@ -46,4 +46,9 @@ interface Response {
    * @throws DatastoreServiceException if there was any failure or if batch is not longer active
    */
   Response submit();
+
+  /**
+   * Returns the batch associated {@link DatastoreService}.
+   */
+  DatastoreService datastore();
 }
diff --git a/src/main/java/com/google/gcloud/datastore/BatchImpl.java b/src/main/java/com/google/gcloud/datastore/BatchImpl.java
@@ -72,4 +72,9 @@ public Batch.Response submit() {
     deactivate();
     return new ResponseImpl(responsePb);
   }
+
+  @Override
+  public DatastoreService datastore() {
+    return datastore;
+  }
 }
diff --git a/src/main/java/com/google/gcloud/datastore/Transaction.java b/src/main/java/com/google/gcloud/datastore/Transaction.java
@@ -109,4 +109,9 @@ interface Response {
    */
   @Override
   boolean active();
+
+  /**
+   * Returns the transaction associated {@link DatastoreService}.
+   */
+  DatastoreService datastore();
 }
diff --git a/src/main/java/com/google/gcloud/datastore/TransactionImpl.java b/src/main/java/com/google/gcloud/datastore/TransactionImpl.java
@@ -117,4 +117,9 @@ public void rollback() {
     deactivate();
     rolledback = true;
   }
+
+  @Override
+  public DatastoreService datastore() {
+    return datastore;
+  }
 }