Add Cloud Spanner Batch DML sample (GoogleCloudPlatform#1361)

jsimonweb · web-flow · commit ac1faf4e417f · 2019-03-25T15:30:10.000-07:00
diff --git a/spanner/cloud-client/pom.xml b/spanner/cloud-client/pom.xml
@@ -42,7 +42,7 @@ limitations under the License.
       <dependency>
         <groupId>com.google.cloud</groupId>
         <artifactId>google-cloud-bom</artifactId>
-        <version>0.73.0-alpha</version>
+        <version>0.84.0-alpha</version>
         <type>pom</type>
         <scope>import</scope>
       </dependency>
diff --git a/spanner/cloud-client/src/main/java/com/example/spanner/SpannerSample.java b/spanner/cloud-client/src/main/java/com/example/spanner/SpannerSample.java
@@ -30,6 +30,7 @@
 import com.google.cloud.spanner.ReadOnlyTransaction;
 import com.google.cloud.spanner.ResultSet;
 import com.google.cloud.spanner.Spanner;
+import com.google.cloud.spanner.SpannerBatchUpdateException;
 import com.google.cloud.spanner.SpannerException;
 import com.google.cloud.spanner.SpannerExceptionFactory;
 import com.google.cloud.spanner.SpannerOptions;
@@ -1084,6 +1085,39 @@ static void deleteUsingPartitionedDml(DatabaseClient dbClient) {
   }
   // [END spanner_dml_partitioned_delete]
 
+  // [START spanner_dml_batch_update]  
+  static void updateUsingBatchDml(DatabaseClient dbClient) {
+    dbClient
+        .readWriteTransaction()
+        .run(
+            new TransactionCallable<Void>() {
+              @Override
+              public Void run(TransactionContext transaction) throws Exception {
+                List<Statement> stmts = new ArrayList<Statement>();
+                String sql = "INSERT INTO Albums "
+                            + "(SingerId, AlbumId, AlbumTitle, MarketingBudget) "
+                            + "VALUES (1, 3, 'Test Album Title', 10000) ";
+                stmts.add(Statement.of(sql));
+                sql = "UPDATE Albums "
+                        + "SET MarketingBudget = MarketingBudget * 2 "
+                        + "WHERE SingerId = 1 and AlbumId = 3";
+                stmts.add(Statement.of(sql));
+                long [] rowCounts;
+                try {
+                  rowCounts = transaction.batchUpdate(stmts);
+                } catch (SpannerBatchUpdateException e) {
+                  rowCounts = e.getUpdateCounts();
+                }
+                for (int i = 0; i < rowCounts.length; i++) {
+                  System.out.printf(
+                      "%d record updated by stmt %d.\n", rowCounts[i], i);
+                }
+                return null;
+              }
+            });
+  }
+  // [END spanner_dml_batch_update]
+
   static void run(
       DatabaseClient dbClient,
       DatabaseAdminClient dbAdminClient,
@@ -1203,7 +1237,10 @@ static void run(
         break;
       case "deleteusingpartitioneddml":
         deleteUsingPartitionedDml(dbClient);
-        break;        
+        break;
+      case "updateusingbatchdml":
+        updateUsingBatchDml(dbClient);
+        break;
       default:
         printUsageAndExit();
     }
@@ -1252,6 +1289,7 @@ static void printUsageAndExit() {
     System.err.println("    SpannerExample writewithtransactionusingdml my-instance example-db");
     System.err.println("    SpannerExample updateusingpartitioneddml my-instance example-db");
     System.err.println("    SpannerExample deleteusingpartitioneddml my-instance example-db");
+    System.err.println("    SpannerExample updateusingbatchdml my-instance example-db");
     System.exit(1);
   }
 
diff --git a/spanner/cloud-client/src/test/java/com/example/spanner/SpannerSampleIT.java b/spanner/cloud-client/src/test/java/com/example/spanner/SpannerSampleIT.java
@@ -199,6 +199,10 @@ public void testSample() throws Exception {
     assertThat(out).doesNotContain("Russell Morales");
     assertThat(out).doesNotContain("Jacqueline Long");
     assertThat(out).doesNotContain("Dylan Shaw");
+
+    out = runSample("updateusingbatchdml");
+    assertThat(out).contains("1 record updated by stmt 0");
+    assertThat(out).contains("1 record updated by stmt 1");
   }
 
   private String formatForTest(String name) {
