-
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,236 additions
and
1 deletion.
There are no files selected for viewing
207 changes: 207 additions & 0 deletions
207
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,207 @@ | ||
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 get 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 an {@link String}. | ||
*/ | ||
@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}). | ||
*/ | ||
@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}). | ||
*/ | ||
@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.