From 9801c6dfc9d5c20650f994678606d900ede5bfec Mon Sep 17 00:00:00 2001
From: yah01 <yang.cen@zilliz.com>
Date: Tue, 27 Sep 2022 15:16:55 +0800
Subject: [PATCH] Fix data race in DataNode (#19470)

Signed-off-by: yah01 <yang.cen@zilliz.com>

Signed-off-by: yah01 <yang.cen@zilliz.com>
---
 internal/datanode/segment_replica.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internal/datanode/segment_replica.go b/internal/datanode/segment_replica.go
index d7943692be..dc445579cd 100644
--- a/internal/datanode/segment_replica.go
+++ b/internal/datanode/segment_replica.go
@@ -907,8 +907,8 @@ func (replica *SegmentReplica) listNotFlushedSegmentIDs() []UniqueID {
 // getSegmentStatslog returns the segment statslog for the provided segment id.
 func (replica *SegmentReplica) getSegmentStatslog(segID UniqueID) ([]byte, error) {
 	replica.segMu.RLock()
+	defer replica.segMu.RUnlock()
 	colID := replica.getCollectionID()
-	replica.segMu.RUnlock()
 
 	schema, err := replica.getCollectionSchema(colID, 0)
 	if err != nil {