feat: Begin transactions before each transaction read/write

package.json

@@ -43,6 +43,7 @@
   "dependencies": {
     "@google-cloud/promisify": "^4.0.0",
     "arrify": "^2.0.1",
+    "async-mutex": "^0.4.0",
     "concat-stream": "^2.0.0",
     "extend": "^3.0.2",
     "google-gax": "^4.0.5",

src/request.ts

@@ -44,7 +44,7 @@
   KeyProto,
   ResponseResult,
   Entities,
   ValueProto,
 } from './entity';
 import {
   Query,
@@ -79,7 +79,7 @@
  * @class
  */
 class DatastoreRequest {
-  id: string | undefined;
+  id: string | undefined | Uint8Array | null;
   requests_:
     | Entity
     | {
@@ -546,6 +546,19 @@
       );
   }
 
+  /**
+   * Datastore allows you to run aggregate queries by supplying aggregate fields
+   * which will determine the type of aggregation that is performed.
+   *
+   * The query is run, and the results are returned in the second argument of
+   * the callback provided.
+   *
+   * @param {AggregateQuery} query AggregateQuery object.
+   * @param {RunQueryOptions} options Optional configuration
+   * @param {function} [callback] The callback function. If omitted, a promise is
+   * returned.
+   *
+   **/
   runAggregationQuery(
     query: AggregateQuery,
     options?: RunQueryOptions
@@ -596,9 +609,9 @@
           const results = res.batch.aggregationResults;
           const finalResults = results
             .map(
               (aggregationResult: any) => aggregationResult.aggregateProperties
             )
             .map((aggregateProperties: any) =>
               Object.fromEntries(
                 new Map(
                   Object.keys(aggregateProperties).map(key => [
@@ -933,7 +946,7 @@
     callback?: SaveCallback
   ): void | Promise<CommitResponse> {
     const transaction = this.datastore.transaction();
     transaction.run(async (err: any) => {
       if (err) {
         try {
           await transaction.rollback();
@@ -1157,7 +1170,7 @@
   partitionId?: google.datastore.v1.IPartitionId | null;
   readOptions?: {
     readConsistency?: number;
-    transaction?: string;
+    transaction?: string | Uint8Array | null;
     readTime?: ITimestamp;
   };
 }
@@ -1166,9 +1179,9 @@
   keys?: Entity;
   transactionOptions?: {
     readOnly?: {};
-    readWrite?: {previousTransaction?: string};
+    readWrite?: {previousTransaction?: string | Uint8Array | null};
   } | null;
-  transaction?: string | null;
+  transaction?: string | null | Uint8Array;
   mode?: string;
   query?: QueryProto;
   filter?: string;