From 871e093dc8296858fb4e5cd388cd9ce51f6397f6 Mon Sep 17 00:00:00 2001 From: "zhenshan.cao" Date: Mon, 17 Jan 2022 14:43:38 +0800 Subject: [PATCH] Call release memory in case error occured (#15237) Signed-off-by: zhenshan.cao --- internal/indexnode/task.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/internal/indexnode/task.go b/internal/indexnode/task.go index 57e002acf2..bc83963185 100644 --- a/internal/indexnode/task.go +++ b/internal/indexnode/task.go @@ -505,6 +505,8 @@ func (it *IndexBuildTask) Execute(ctx context.Context) error { return err } + defer it.releaseMemory() + var err error it.index, err = NewCIndex(it.newTypeParams, it.newIndexParams) if err != nil { @@ -538,7 +540,5 @@ func (it *IndexBuildTask) Execute(ctx context.Context) error { log.Info("IndexNode CreateIndex successfully ", zap.Int64("collect", it.collectionID), zap.Int64("partition", it.partitionID), zap.Int64("segment", it.segmentID)) - it.releaseMemory() - return nil }