Add queryWithParameter to Cloud Spanner sample. (GoogleCloudPlatform#1417)

jsimonweb · web-flow · commit 34003421a7c6 · 2019-05-14T17:14:40.000-07:00
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
@@ -1013,6 +1013,29 @@ public Void run(TransactionContext transaction) throws Exception {
   }
   // [END spanner_dml_getting_started_insert]
 
+  // [START spanner_query_with_parameter]
+  static void queryWithParameter(DatabaseClient dbClient) {
+    Statement statement =
+        Statement
+            .newBuilder(
+                "SELECT SingerId, FirstName, LastName\n"
+                    + "FROM Singers\n"
+                    + "WHERE LastName = @lastName")
+            .bind("lastName")
+            .to("Garcia")
+            .build();
+
+    ResultSet resultSet = dbClient.singleUse().executeQuery(statement);
+    while (resultSet.next()) {
+      System.out.printf(
+          "%d %s %s\n",
+          resultSet.getLong("SingerId"),
+          resultSet.getString("FirstName"),
+          resultSet.getString("LastName"));
+    }
+  }
+  // [END spanner_query_with_parameter]
+
   // [START spanner_dml_getting_started_update]
   static void writeWithTransactionUsingDml(DatabaseClient dbClient) {
     dbClient
@@ -1229,6 +1252,9 @@ static void run(
       case "writeusingdml":
         writeUsingDml(dbClient);
         break;
+      case "querywithparameter":
+        queryWithParameter(dbClient);
+        break;
       case "writewithtransactionusingdml":
         writeWithTransactionUsingDml(dbClient);
         break;
@@ -1286,6 +1312,7 @@ static void printUsageAndExit() {
     System.err.println("    SpannerExample writeandreadusingdml my-instance example-db");
     System.err.println("    SpannerExample updateusingdmlwithstruct my-instance example-db");
     System.err.println("    SpannerExample writeusingdml my-instance example-db");
+    System.err.println("    SpannerExample queryWithParameter my-instance example-db");
     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");
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
@@ -144,7 +144,7 @@ public void testSample() throws Exception {
     assertThat(out).startsWith("6\n");
 
     out = runSample("querywitharrayofstruct");
-    assertThat(out).startsWith("6\n7");
+    assertThat(out).startsWith("8\n7\n6");
 
     out = runSample("querystructfield");
     assertThat(out).startsWith("6\n");
@@ -181,6 +181,8 @@ public void testSample() throws Exception {
     assertThat(out).contains("Russell Morales");
     assertThat(out).contains("Jacqueline Long");
     assertThat(out).contains("Dylan Shaw");
+    out = runSample("querywithparameter");
+    assertThat(out).contains("12 Melissa Garcia");
 
     runSample("writewithtransactionusingdml");
     out = runSample("querymarketingbudget");    
