From ce2fa3de01ab1f31e22fe4eb918f9837d0ae7359 Mon Sep 17 00:00:00 2001
From: XuanYang-cn <xuan.yang@zilliz.com>
Date: Sat, 9 Nov 2024 01:42:26 +0800
Subject: [PATCH] fix: Set correct memory size (#37551)

See also: #37549

Signed-off-by: yangxuan <xuan.yang@zilliz.com>
---
 internal/storage/serde.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internal/storage/serde.go b/internal/storage/serde.go
index 7da70f282b..3a5b12e274 100644
--- 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)