fix(bigquery): handle null RANGE (#11058)

Fixes #11047

bigquery/value.go

@@ -1021,7 +1021,11 @@ var unboundedRangeSentinel = "UNBOUNDED"
 
 // convertRangeValue aids in parsing the compound RANGE api data representation.
 // The format for a range value is: "[startval, endval)"
-func convertRangeValue(val string, elementType FieldType) (Value, error) {
+func convertRangeValue(cellVal interface{}, elementType FieldType) (Value, error) {
+	if cellVal == nil {
+		return nil, nil
+	}
+	val := cellVal.(string)
 	supported := false
 	for _, t := range []FieldType{DateFieldType, DateTimeFieldType, TimestampFieldType} {
 		if elementType == t {
@@ -1066,13 +1070,13 @@ func convertRangeTableCell(cell *bq.TableCell, fs *FieldSchema) (Value, error) {
 		rangeValues := []Value{}
 		for _, val := range cell.V.([]interface{}) {
 			rawRangeValue := val.(map[string]interface{})["v"]
-			rangeVal, err := convertRangeValue(rawRangeValue.(string), fs.RangeElementType.Type)
+			rangeVal, err := convertRangeValue(rawRangeValue, fs.RangeElementType.Type)
 			if err != nil {
 				return nil, err
 			}
 			rangeValues = append(rangeValues, rangeVal)
 		}
 		return rangeValues, nil
 	}
-	return convertRangeValue(cell.V.(string), fs.RangeElementType.Type)
+	return convertRangeValue(cell.V, fs.RangeElementType.Type)
 }

bigquery/value_test.go

@@ -104,6 +104,42 @@ func TestConvertTime(t *testing.T) {
 	}
 }
 
+func TestConvertRange(t *testing.T) {
+	schema := Schema{
+		{Type: RangeFieldType, RangeElementType: &RangeElementType{Type: TimestampFieldType}},
+		{Type: RangeFieldType, RangeElementType: &RangeElementType{Type: DateTimeFieldType}},
+		{Type: RangeFieldType, RangeElementType: &RangeElementType{Type: DateFieldType}},
+		// Test null value
+		{Type: RangeFieldType, RangeElementType: &RangeElementType{Type: TimestampFieldType}},
+	}
+
+	ts := testTimestamp.Round(time.Millisecond)
+	row := &bq.TableRow{
+		F: []*bq.TableCell{
+			{V: fmt.Sprintf("[%d, UNBOUNDED)", ts.UnixMicro())},
+			{V: fmt.Sprintf("[UNBOUNDED, %s)", testDateTime.String())},
+			{V: fmt.Sprintf("[%s, %s)", testDate.String(), testDate.String())},
+			{V: nil}, // NULL RANGE
+		},
+	}
+	got, err := convertRow(row, schema)
+	if err != nil {
+		t.Fatalf("error converting: %v", err)
+	}
+	want := []Value{
+		&RangeValue{Start: ts},
+		&RangeValue{End: testDateTime},
+		&RangeValue{Start: testDate, End: testDate},
+		nil,
+	}
+	for i, g := range got {
+		w := want[i]
+		if !testutil.Equal(g, w) {
+			t.Errorf("#%d: got:\n%v\nwant:\n%v", i, g, w)
+		}
+	}
+}
+
 func TestConvertSmallTimes(t *testing.T) {
 	for _, year := range []int{1600, 1066, 1} {
 		want := time.Date(year, time.January, 1, 0, 0, 0, 0, time.UTC)