Skip to content

Commit 0abdd90

Browse files
author
Praful Makani
authored
docs(samples): add query with timestamp parameters (#604)
1 parent 1eeffe6 commit 0abdd90

File tree

2 files changed

+116
-0
lines changed

2 files changed

+116
-0
lines changed
Lines changed: 67 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,67 @@
1+
/*
2+
* Copyright 2020 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
package com.example.bigquery;
18+
19+
// [START bigquery_query_params_timestamps]
20+
import com.google.cloud.bigquery.BigQuery;
21+
import com.google.cloud.bigquery.BigQueryException;
22+
import com.google.cloud.bigquery.BigQueryOptions;
23+
import com.google.cloud.bigquery.QueryJobConfiguration;
24+
import com.google.cloud.bigquery.QueryParameterValue;
25+
import com.google.cloud.bigquery.TableResult;
26+
import org.threeten.bp.LocalDateTime;
27+
import org.threeten.bp.ZoneOffset;
28+
import org.threeten.bp.ZonedDateTime;
29+
30+
// Sample to running a query with timestamp query parameters.
31+
public class QueryWithTimestampParameters {
32+
33+
public static void runQueryWithTimestampParameters() {
34+
queryWithTimestampParameters();
35+
}
36+
37+
public static void queryWithTimestampParameters() {
38+
try {
39+
// Initialize client that will be used to send requests. This client only needs to be created
40+
// once, and can be reused for multiple requests.
41+
BigQuery bigquery = BigQueryOptions.getDefaultInstance().getService();
42+
43+
ZonedDateTime timestamp = LocalDateTime.of(2016, 12, 7, 8, 0, 0).atZone(ZoneOffset.UTC);
44+
String query = "SELECT TIMESTAMP_ADD(@ts_value, INTERVAL 1 HOUR);";
45+
// Note: Standard SQL is required to use query parameters.
46+
QueryJobConfiguration queryConfig =
47+
QueryJobConfiguration.newBuilder(query)
48+
.addNamedParameter(
49+
"ts_value",
50+
QueryParameterValue.timestamp(
51+
// Timestamp takes microseconds since 1970-01-01T00:00:00 UTC
52+
timestamp.toInstant().toEpochMilli() * 1000))
53+
.build();
54+
55+
TableResult results = bigquery.query(queryConfig);
56+
57+
results
58+
.iterateAll()
59+
.forEach(row -> row.forEach(val -> System.out.printf("%s", val.toString())));
60+
61+
System.out.println("Query with timestamp parameter performed successfully.");
62+
} catch (BigQueryException | InterruptedException e) {
63+
System.out.println("Query not performed \n" + e.toString());
64+
}
65+
}
66+
}
67+
// [END bigquery_query_params_timestamps]
Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
1+
/*
2+
* Copyright 2020 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
package com.example.bigquery;
18+
19+
import static com.google.common.truth.Truth.assertThat;
20+
21+
import java.io.ByteArrayOutputStream;
22+
import java.io.PrintStream;
23+
import org.junit.After;
24+
import org.junit.Before;
25+
import org.junit.Test;
26+
27+
public class QueryWithTimestampParametersIT {
28+
29+
private ByteArrayOutputStream bout;
30+
private PrintStream out;
31+
32+
@Before
33+
public void setUp() {
34+
bout = new ByteArrayOutputStream();
35+
out = new PrintStream(bout);
36+
System.setOut(out);
37+
}
38+
39+
@After
40+
public void tearDown() {
41+
System.setOut(null);
42+
}
43+
44+
@Test
45+
public void testQueryWithTimestampParameters() {
46+
QueryWithTimestampParameters.queryWithTimestampParameters();
47+
assertThat(bout.toString()).contains("Query with timestamp parameter performed successfully.");
48+
}
49+
}

0 commit comments

Comments
 (0)