Handle parquet files that don't have any rows

parsyl · Feb 13, 2019 · 641ebd1 · 641ebd1
1 parent 8bcb5a2
commit 641ebd1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/cmd/parquetgen/main.go b/cmd/parquetgen/main.go
@@ -753,6 +753,13 @@ func (p *ParquetReader) Error() error {
 }
 
 func (p *ParquetReader) readRowGroup() error {
+	p.rowGroupCursor = 0
+
+	if len(p.rowGroups) == 0 {
+		p.rowGroupCount = 0
+		return nil
+	}
+
 	rg := p.rowGroups[0]
 	p.fields = getFields(Fields())
 	p.rowGroupCount = rg.Rows