fix: add check for WriteRecordBatch (#47137)

fix for https://github.com/milvus-io/milvus/issues/47073

---------

Signed-off-by: Xinyi Jiang <xinyi.jiang@reddit.com>
Co-authored-by: Xinyi Jiang <xinyi.jiang@reddit.com>
This commit is contained in:
Xinyi7 2026-01-25 12:11:32 -05:00 committed by GitHub
parent e092884822
commit 610bc045da
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -128,6 +128,10 @@ func NewPackedWriter(filePaths []string, schema *arrow.Schema, bufferSize int64,
}
func (pw *PackedWriter) WriteRecordBatch(recordBatch arrow.Record) error {
if recordBatch == nil || recordBatch.NumCols() == 0 {
return nil
}
cArrays := make([]CArrowArray, recordBatch.NumCols())
cSchemas := make([]CArrowSchema, recordBatch.NumCols())