feat: add stream write samples for range (#780)

* feat: add stream write samples for range

* lint

samples/snippets/append_rows_proto2.py

@@ -214,6 +214,14 @@ def append_rows_proto2(project_id: str, dataset_id: str, table_id: str):
  row.struct_list.append(sub_message)
  proto_rows.serialized_rows.append(row.SerializeToString())
 
+ row = sample_data_pb2.SampleData()
+ row.row_num = 16
+ date_value = datetime.date(2021, 8, 8)
+ epoch_value = datetime.date(1970, 1, 1)
+ delta = date_value - epoch_value
+ row.range_date.start = delta.days
+ proto_rows.serialized_rows.append(row.SerializeToString())
+
  request = types.AppendRowsRequest()
  request.offset = 12
  proto_data = types.AppendRowsRequest.ProtoData()

samples/snippets/sample_data.proto

@@ -29,6 +29,11 @@ message SampleData {
  optional int64 sub_int_col = 1;
  }
 
+ message RangeValue {
+ optional int32 start = 1;
+ optional int32 end = 2;
+ }
+
  // The following types map directly between protocol buffers and their
  // corresponding BigQuery data types.
  optional bool bool_col = 1;
@@ -55,7 +60,11 @@ message SampleData {
  optional SampleStruct struct_col = 14;
  repeated SampleStruct struct_list = 15;
 
+ // Range types, see:
+ // https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#range_type
+ optional RangeValue range_date = 16;
+
  // Use the required keyword for client-side validation of required fields.
- required int64 row_num = 16;
+ required int64 row_num = 17;
 }
 // [END bigquerystorage_append_rows_raw_proto2_definition]

samples/snippets/sample_data_schema.json

@@ -72,5 +72,10 @@
  {"name": "sub_int_col", "type": "INTEGER"}
  ],
  "mode": "REPEATED"
+ },
+ {
+ "name": "range_date",
+ "type": "RANGE",
+ "rangeElementType": {"type": "DATE"}
  }
  ]