From 7fedacdc5c924de8b25aac59d00018704e0d5af8 Mon Sep 17 00:00:00 2001
From: Stephanie Wang <stephaniewang526@users.noreply.github.com>
Date: Tue, 14 Jan 2020 11:53:20 -0500
Subject: [PATCH] feat: sample - run a legacy SQL query (#97)

* feat: sample - run a legacy SQL query

* nit: update

* nit: update

* update base on comments

* update base on comments
---
 .../com/example/bigquery/RunLegacyQuery.java  | 52 +++++++++++++++++++
 .../bigquery/CopyMultipleTablesIT.java        |  2 +-
 .../example/bigquery/RunLegacyQueryIT.java    | 48 +++++++++++++++++
 3 files changed, 101 insertions(+), 1 deletion(-)
 create mode 100644 samples/src/main/java/com/example/bigquery/RunLegacyQuery.java
 create mode 100644 samples/src/test/java/com/example/bigquery/RunLegacyQueryIT.java

diff --git a/samples/src/main/java/com/example/bigquery/RunLegacyQuery.java b/samples/src/main/java/com/example/bigquery/RunLegacyQuery.java
new file mode 100644
index 000000000..747772220
--- /dev/null
+++ b/samples/src/main/java/com/example/bigquery/RunLegacyQuery.java
@@ -0,0 +1,52 @@
+/*
+ * Copyright 2020 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.example.bigquery;
+
+// [START bigquery_query_legacy]
+import com.google.cloud.bigquery.BigQuery;
+import com.google.cloud.bigquery.BigQueryException;
+import com.google.cloud.bigquery.BigQueryOptions;
+import com.google.cloud.bigquery.QueryJobConfiguration;
+import com.google.cloud.bigquery.TableResult;
+
+public class RunLegacyQuery {
+
+  public static void runLegacyQuery() {
+    try {
+      // Initialize client that will be used to send requests. This client only needs to be created
+      // once, and can be reused for multiple requests.
+      BigQuery bigquery = BigQueryOptions.getDefaultInstance().getService();
+
+      // To use legacy SQL syntax, set useLegacySql to true.
+      String query =
+          "SELECT corpus FROM [bigquery-public-data:samples.shakespeare] GROUP BY corpus;";
+      QueryJobConfiguration queryConfig =
+          QueryJobConfiguration.newBuilder(query).setUseLegacySql(true).build();
+
+      // Execute the query.
+      TableResult result = bigquery.query(queryConfig);
+
+      // Print the results.
+      result.iterateAll().forEach(rows -> rows.forEach(row -> System.out.println(row.getValue())));
+
+      System.out.println("Legacy query ran successfully");
+    } catch (BigQueryException | InterruptedException e) {
+      System.out.println("Legacy query did not run \n" + e.toString());
+    }
+  }
+}
+// [END bigquery_query_legacy]
diff --git a/samples/src/test/java/com/example/bigquery/CopyMultipleTablesIT.java b/samples/src/test/java/com/example/bigquery/CopyMultipleTablesIT.java
index f78e3ff81..bcf7475c7 100644
--- a/samples/src/test/java/com/example/bigquery/CopyMultipleTablesIT.java
+++ b/samples/src/test/java/com/example/bigquery/CopyMultipleTablesIT.java
@@ -70,7 +70,7 @@ public void testCopyMultipleTables() {
     CopyMultipleTables.copyMultipleTables(BIGQUERY_DATASET_NAME, generatedTableName);
     assertThat(bout.toString()).contains("Table copied successfully.");
 
-    //Clean up
+    // Clean up
     DeleteTable.deleteTable(BIGQUERY_DATASET_NAME, generatedTableName);
   }
 }
diff --git a/samples/src/test/java/com/example/bigquery/RunLegacyQueryIT.java b/samples/src/test/java/com/example/bigquery/RunLegacyQueryIT.java
new file mode 100644
index 000000000..e8a7d16eb
--- /dev/null
+++ b/samples/src/test/java/com/example/bigquery/RunLegacyQueryIT.java
@@ -0,0 +1,48 @@
+/*
+ * Copyright 2020 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.example.bigquery;
+
+import static com.google.common.truth.Truth.assertThat;
+
+import java.io.ByteArrayOutputStream;
+import java.io.PrintStream;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+public class RunLegacyQueryIT {
+  private ByteArrayOutputStream bout;
+  private PrintStream out;
+
+  @Before
+  public void setUp() {
+    bout = new ByteArrayOutputStream();
+    out = new PrintStream(bout);
+    System.setOut(out);
+  }
+
+  @After
+  public void tearDown() {
+    System.setOut(null);
+  }
+
+  @Test
+  public void testRunLegacyQuery() {
+    RunLegacyQuery.runLegacyQuery();
+    assertThat(bout.toString()).contains("Legacy query ran successfully");
+  }
+}