-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add InsertAllRequest and InsertAllResponse. Add model classes for tab…
…le data
- Loading branch information
Showing
9 changed files
with
1,340 additions
and
1 deletion.
There are no files selected for viewing
232 changes: 232 additions & 0 deletions
232
gcloud-java-bigquery/src/main/java/com/google/gcloud/bigquery/FieldValue.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,232 @@ | ||
/* | ||
* Copyright 2015 Google Inc. All Rights Reserved. | ||
* | ||
* 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.google.gcloud.bigquery; | ||
|
||
import static com.google.common.base.Preconditions.checkState; | ||
|
||
import com.google.api.client.util.Data; | ||
import com.google.api.client.util.Lists; | ||
import com.google.common.base.MoreObjects; | ||
|
||
import java.io.Serializable; | ||
import java.util.List; | ||
import java.util.Map; | ||
import java.util.Objects; | ||
|
||
/** | ||
* Google BigQuery Table Field Value class. Objects of this class represent values of a BigQuery | ||
* Table Field. A list of values forms a {@link TableRow}. Tables rows can be gotten as the result | ||
* of a query or when listing table data. | ||
*/ | ||
public class FieldValue implements Serializable { | ||
|
||
private static final int MICROSECONDS = 1000000; | ||
private static final long serialVersionUID = 469098630191710061L; | ||
|
||
private final Kind kind; | ||
private final Object value; | ||
|
||
/** | ||
* The field value's kind, giving information on the field's content type. | ||
*/ | ||
public enum Kind { | ||
/** | ||
* A primitive field value. A {@code FieldValue} has type {@link #PRIMITIVE} when the | ||
* corresponding field has type {@link Field.Type#bool()}, {@link Field.Type#string()} | ||
* {@link Field.Type#floatingPoint()}, {@link Field.Type#integer()}, | ||
* {@link Field.Type#timestamp()} or the value is set to {@code null}. | ||
*/ | ||
PRIMITIVE, | ||
|
||
/** | ||
* A {@code FieldValue} for a field with {@link Field.Mode#REPEATED} mode. | ||
*/ | ||
REPEATED, | ||
|
||
/** | ||
* A {@code FieldValue} for a field of type {@link Field.Type#record(Field...)}. | ||
*/ | ||
RECORD | ||
} | ||
|
||
FieldValue(Kind kind, Object value) { | ||
this.kind = kind; | ||
this.value = value; | ||
} | ||
|
||
/** | ||
* Returns the kind of this Field Value. | ||
* | ||
* @return {@link Kind#PRIMITIVE} if the value is of primitive type ({@link Field.Type#bool()}, | ||
* {@link Field.Type#string()}, {@link Field.Type#floatingPoint()}, | ||
* {@link Field.Type#integer()}, {@link Field.Type#timestamp()}) or is {@code null}. Returns | ||
* {@link Kind#REPEATED} if the corresponding field has ({@link Field.Mode#REPEATED}) mode. | ||
* Returns {@link Kind#RECORD} if the corresponding field has | ||
* {@link Field.Type#record(Field...)} type. | ||
*/ | ||
public Kind kind() { | ||
return kind; | ||
} | ||
|
||
/** | ||
* Return this field's value as an {@link Object}. | ||
*/ | ||
public Object value() { | ||
return value; | ||
} | ||
|
||
/** | ||
* Return this field's value as a {@link String}. This method should only be used if the | ||
* corresponding field has primitive type ({@link Field.Type#bool()}, {@link Field.Type#string()}, | ||
* {@link Field.Type#floatingPoint()}, {@link Field.Type#integer()}, | ||
* {@link Field.Type#timestamp()}). | ||
* | ||
* @throws ClassCastException if the field has not primitive type | ||
*/ | ||
@SuppressWarnings("unchecked") | ||
public String stringValue() { | ||
return (String) value; | ||
} | ||
|
||
/** | ||
* Returns this field's value as a {@link Long}. This method should only be used if the | ||
* corresponding field has {@link Field.Type#integer()} type. | ||
* | ||
* @throws NumberFormatException if the field's value could not be converted to {@link Integer} | ||
*/ | ||
@SuppressWarnings("unchecked") | ||
public long longValue() { | ||
return Long.valueOf(stringValue()); | ||
} | ||
|
||
/** | ||
* Returns this field's value as a {@link Double}. This method should only be used if the | ||
* corresponding field has {@link Field.Type#floatingPoint()} type. | ||
* | ||
* @throws NumberFormatException if the field's value could not be converted to {@link Double} | ||
*/ | ||
@SuppressWarnings("unchecked") | ||
public double doubleValue() { | ||
return Double.valueOf(stringValue()); | ||
} | ||
|
||
/** | ||
* Returns this field's value as a {@link Boolean}. This method should only be used if the | ||
* corresponding field has {@link Field.Type#bool()} type. | ||
* | ||
* @throws IllegalStateException if the field's value could not be converted to {@link Boolean} | ||
*/ | ||
@SuppressWarnings("unchecked") | ||
public boolean booleanValue() { | ||
String stringValue = stringValue(); | ||
checkState(stringValue.equalsIgnoreCase("true") || stringValue.equalsIgnoreCase("false"), | ||
"Field value is not of boolean type"); | ||
return Boolean.parseBoolean(stringValue); | ||
} | ||
|
||
/** | ||
* Returns this field's value as a {@link Long}, representing a timestamp in microseconds. This | ||
* method should only be used if the corresponding field has {@link Field.Type#timestamp()} type. | ||
* | ||
* @throws NumberFormatException if the field's value could not be converted to {@link Long} | ||
*/ | ||
@SuppressWarnings("unchecked") | ||
public long timestampValue() { | ||
return new Double(((Double.valueOf(stringValue())) * MICROSECONDS)).longValue(); | ||
} | ||
|
||
/** | ||
* Returns this field's value as a list of {@link FieldValue}. This method should only be used if | ||
* the corresponding field has {@link Field.Mode#REPEATED} mode (i.e. {@link #kind()} is | ||
* {@link Kind#REPEATED}). | ||
* | ||
* @throws ClassCastException if the field has not {@link Field.Mode#REPEATED} mode | ||
*/ | ||
@SuppressWarnings("unchecked") | ||
public List<FieldValue> repeatedValue() { | ||
return (List<FieldValue>) value; | ||
} | ||
|
||
/** | ||
* Returns this field's value as a list of {@link FieldValue}. This method should only be used if | ||
* the corresponding field has {@link Field.Type#record(Field...)} type (i.e. {@link #kind()} is | ||
* {@link Kind#RECORD}). | ||
* | ||
* @throws ClassCastException if the field has not {@link Field.Type#record(Field...)} type | ||
*/ | ||
@SuppressWarnings("unchecked") | ||
public List<FieldValue> recordValue() { | ||
return (List<FieldValue>) value; | ||
} | ||
|
||
@Override | ||
public String toString() { | ||
return MoreObjects.toStringHelper(this) | ||
.add("kind", kind) | ||
.add("value", value) | ||
.toString(); | ||
} | ||
|
||
@Override | ||
public int hashCode() { | ||
return Objects.hash(kind, value); | ||
} | ||
|
||
@Override | ||
public boolean equals(Object obj) { | ||
if (!(obj instanceof FieldValue)) { | ||
return false; | ||
} | ||
FieldValue other = (FieldValue) obj; | ||
return kind == other.kind && Objects.equals(value, other.value); | ||
} | ||
|
||
@SuppressWarnings("unchecked") | ||
static FieldValue fromPb(Object cellPb) { | ||
if (Data.isNull(cellPb)) { | ||
return new FieldValue(Kind.PRIMITIVE, null); | ||
} | ||
if (cellPb instanceof String) { | ||
return new FieldValue(Kind.PRIMITIVE, cellPb); | ||
} | ||
if (cellPb instanceof List) { | ||
List<Object> cellsListPb = (List<Object>) cellPb; | ||
List<FieldValue> repeatedCells = Lists.newArrayListWithCapacity(cellsListPb.size()); | ||
for (Object repeatedCellPb : cellsListPb) { | ||
repeatedCells.add(FieldValue.fromPb(repeatedCellPb)); | ||
} | ||
return new FieldValue(Kind.REPEATED, repeatedCells); | ||
} | ||
if (cellPb instanceof Map) { | ||
Map<String, Object> cellMapPb = (Map<String, Object>) cellPb; | ||
if (cellMapPb.containsKey("f")) { | ||
List<Object> cellsListPb = (List<Object>) cellMapPb.get("f"); | ||
List<FieldValue> recordCells = Lists.newArrayListWithCapacity(cellsListPb.size()); | ||
for (Object repeatedCellPb : cellsListPb) { | ||
recordCells.add(FieldValue.fromPb(repeatedCellPb)); | ||
} | ||
return new FieldValue(Kind.RECORD, recordCells); | ||
} | ||
// This should never be the case when we are processing a first level table field (i.e. a | ||
// row's field, not a record sub-field) | ||
if (cellMapPb.containsKey("v")) { | ||
return FieldValue.fromPb(cellMapPb.get("v")); | ||
} | ||
} | ||
throw new AssertionError("Unexpected table cell format"); | ||
} | ||
} |
Oops, something went wrong.