bigquery: support BYTES data type

Change-Id: I30d78682a67773f58ed8d07cd0cec0175d21410b Reviewed-on: https://code-review.googlesource.com/9556 Reviewed-by: Michael Darakananda <[email protected]> Reviewed-by: Sarah Adams <[email protected]>
googleapis · Nov 28, 2016 · 648bc87 · 648bc87
1 parent 7bb67ec
commit 648bc87
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/bigquery/schema.go b/bigquery/schema.go
@@ -106,6 +106,7 @@ type FieldType string
 
 const (
 	StringFieldType    FieldType = "STRING"
+	BytesFieldType     FieldType = "BYTES"
 	IntegerFieldType   FieldType = "INTEGER"
 	FloatFieldType     FieldType = "FLOAT"
 	BooleanFieldType   FieldType = "BOOLEAN"

diff --git a/bigquery/value.go b/bigquery/value.go
@@ -15,6 +15,7 @@
 package bigquery
 
 import (
+	"encoding/base64"
 	"errors"
 	"fmt"
 	"strconv"
@@ -194,6 +195,8 @@ func convertBasicType(val string, typ FieldType) (Value, error) {
 	switch typ {
 	case StringFieldType:
 		return val, nil
+	case BytesFieldType:
+		return base64.StdEncoding.DecodeString(val)
 	case IntegerFieldType:
 		return strconv.ParseInt(val, 10, 64)
 	case FloatFieldType:

diff --git a/bigquery/value_test.go b/bigquery/value_test.go
@@ -15,6 +15,7 @@
 package bigquery
 
 import (
+	"encoding/base64"
 	"fmt"
 	"reflect"
 	"testing"
@@ -29,20 +30,22 @@ func TestConvertBasicValues(t *testing.T) {
 		{Type: IntegerFieldType},
 		{Type: FloatFieldType},
 		{Type: BooleanFieldType},
+		{Type: BytesFieldType},
 	}
 	row := &bq.TableRow{
 		F: []*bq.TableCell{
 			{V: "a"},
 			{V: "1"},
 			{V: "1.2"},
 			{V: "true"},
+			{V: base64.StdEncoding.EncodeToString([]byte("foo"))},
 		},
 	}
 	got, err := convertRow(row, schema)
 	if err != nil {
 		t.Fatalf("error converting: %v", err)
 	}
-	want := []Value{"a", int64(1), 1.2, true}
+	want := []Value{"a", int64(1), 1.2, true, []byte("foo")}
 	if !reflect.DeepEqual(got, want) {
 		t.Errorf("converting basic values: got:\n%v\nwant:\n%v", got, want)
 	}