Fixes Bigquery Timestamp

Author: Praful Makani

google-cloud-clients/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/QueryParameterValue.java

@@ -30,7 +30,9 @@
 import org.threeten.bp.Instant;
 import org.threeten.bp.ZoneOffset;
 import org.threeten.bp.format.DateTimeFormatter;
+import org.threeten.bp.format.DateTimeFormatterBuilder;
 import org.threeten.bp.format.DateTimeParseException;
+import org.threeten.bp.temporal.ChronoField;
 
 /**
  * A value for a QueryParameter along with its type.
@@ -60,6 +62,21 @@
 @AutoValue
 public abstract class QueryParameterValue implements Serializable {
 
+  private static final DateTimeFormatter timestampValidator =
+      new DateTimeFormatterBuilder()
+          .parseLenient()
+          .append(DateTimeFormatter.ISO_LOCAL_DATE)
+          .appendLiteral(' ')
+          .append(DateTimeFormatter.ISO_LOCAL_TIME)
+          .appendFraction(ChronoField.NANO_OF_SECOND, 0, 9, true)
+          .optionalStart()
+          .appendOffsetId()
+          .optionalEnd()
+          .optionalStart()
+          .appendOffset("+HHMM", "Z")
+          .optionalEnd()
+          .toFormatter()
+          .withZone(ZoneOffset.UTC);
   private static final DateTimeFormatter timestampFormatter =
       DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss.SSSSSSxxx").withZone(ZoneOffset.UTC);
   private static final DateTimeFormatter dateFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
@@ -301,7 +318,7 @@ private static <T> String valueToStringOrNull(T value, StandardSQLTypeName type)
           return timestampFormatter.format(Instant.ofEpochMilli(((Long) value) / 1000));
         } else if (value instanceof String) {
           // verify that the String is in the right format
-          checkFormat(value, timestampFormatter);
+          checkFormat(value, timestampValidator);
           return (String) value;
         }
         break;
@@ -335,6 +352,7 @@ private static <T> String valueToStringOrNull(T value, StandardSQLTypeName type)
 
   private static void checkFormat(Object value, DateTimeFormatter formatter) {
     try {
+
       formatter.parse((String) value);
     } catch (DateTimeParseException e) {
       throw new IllegalArgumentException(e.getMessage(), e);

google-cloud-clients/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/QueryParameterValueTest.java

@@ -189,6 +189,25 @@ public void testTimestamp() {
     assertThat(value.getArrayValues()).isNull();
   }
 
+  @Test
+  public void testTimestampWithDateTimeFormatterBuilder() {
+    QueryParameterValue value = QueryParameterValue.timestamp("2019-02-14 12:34:45.938993Z");
+    assertThat(value.getValue()).isEqualTo("2019-02-14 12:34:45.938993Z");
+    assertThat(value.getType()).isEqualTo(StandardSQLTypeName.TIMESTAMP);
+    assertThat(value.getArrayType()).isNull();
+    assertThat(value.getArrayValues()).isNull();
+    QueryParameterValue value1 = QueryParameterValue.timestamp("2019-02-14 12:34:45.938993+0000");
+    assertThat(value1.getValue()).isEqualTo("2019-02-14 12:34:45.938993+0000");
+    assertThat(value1.getType()).isEqualTo(StandardSQLTypeName.TIMESTAMP);
+    assertThat(value1.getArrayType()).isNull();
+    assertThat(value1.getArrayValues()).isNull();
+    QueryParameterValue value2 = QueryParameterValue.timestamp("2019-02-14 12:34:45.102+00:00");
+    assertThat(value2.getValue()).isEqualTo("2019-02-14 12:34:45.102+00:00");
+    assertThat(value2.getType()).isEqualTo(StandardSQLTypeName.TIMESTAMP);
+    assertThat(value2.getArrayType()).isNull();
+    assertThat(value2.getArrayValues()).isNull();
+  }
+
   @Test(expected = IllegalArgumentException.class)
   public void testInvalidTimestamp() {
     // missing the time
@@ -269,6 +288,29 @@ public void testTimestampArray() {
         value.getArrayValues());
   }
 
+  @Test
+  public void testTimestampArrayWithDateTimeFormatterBuilder() {
+    QueryParameterValue value =
+        QueryParameterValue.array(
+            new String[] {
+              "2019-02-14 12:34:45.938993Z",
+              "2019-02-14 12:34:45.938993+0000",
+              "2019-02-14 12:34:45.102+00:00"
+            },
+            StandardSQLTypeName.TIMESTAMP);
+    assertThat(value.getValue()).isNull();
+    assertThat(value.getType()).isEqualTo(StandardSQLTypeName.ARRAY);
+    assertThat(value.getArrayType()).isEqualTo(StandardSQLTypeName.TIMESTAMP);
+    assertArrayDataEquals(
+        new String[] {
+          "2019-02-14 12:34:45.938993Z",
+          "2019-02-14 12:34:45.938993+0000",
+          "2019-02-14 12:34:45.102+00:00"
+        },
+        StandardSQLTypeName.TIMESTAMP,
+        value.getArrayValues());
+  }
+
   @Test
   public void testFromEmptyArray() {
     QueryParameterType typePb =