add check for WriteRecordBatch

Xinyi Jiang · Xinyi Jiang · commit 8c941b538d47 · 2026-01-16T18:03:48.000-05:00
diff --git a/internal/storagev2/packed/packed_writer.go b/internal/storagev2/packed/packed_writer.go
@@ -128,6 +128,10 @@ func NewPackedWriter(filePaths []string, schema *arrow.Schema, bufferSize int64,
 }
 
 func (pw *PackedWriter) WriteRecordBatch(recordBatch arrow.Record) error {
+	if recordBatch.NumCols() == 0 {
+		return nil
+	}
+
 	cArrays := make([]CArrowArray, recordBatch.NumCols())
 	cSchemas := make([]CArrowSchema, recordBatch.NumCols())
 

Original file line number	Diff line number	Diff line change
`@@ -128,6 +128,10 @@ func NewPackedWriter(filePaths []string, schema *arrow.Schema, bufferSize int64,`
`128`	`128`	`}`
`129`	`129`
`130`	`130`	`func (pw *PackedWriter) WriteRecordBatch(recordBatch arrow.Record) error {`
	`131`	`+ if recordBatch.NumCols() == 0 {`
	`132`	`+ return nil`
	`133`	`+ }`
	`134`	`+`
`131`	`135`	`cArrays := make([]CArrowArray, recordBatch.NumCols())`
`132`	`136`	`cSchemas := make([]CArrowSchema, recordBatch.NumCols())`
`133`	`137`