fix: Set correct memory size (#37551)

See also: #37549 Signed-off-by: yangxuan <[email protected]>
milvus-io · Nov 8, 2024 · ce2fa3d · ce2fa3d
1 parent ebc3c82
commit ce2fa3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/storage/serde.go b/internal/storage/serde.go
@@ -705,7 +705,7 @@ func (sfw *singleFieldRecordWriter) Write(r Record) error {
 	sfw.numRows += r.Len()
 	a := r.Column(sfw.fieldId)
 
-	sfw.writtenUncompressed += uint64(a.Data().Buffers()[0].Len())
+	sfw.writtenUncompressed += uint64(calculateArraySize(a))
 	rec := array.NewRecord(sfw.schema, []arrow.Array{a}, int64(r.Len()))
 	defer rec.Release()
 	return sfw.fw.WriteBuffered(rec)