From 4e0165ea609585f16c9aca7676e71f5a17452b85 Mon Sep 17 00:00:00 2001
From: bigsheeper <yihao.dai@zilliz.com>
Date: Fri, 27 Dec 2024 00:47:10 +0800
Subject: [PATCH] fix: Fix rootcoord meta mutex contention

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
---
 internal/metastore/model/collection.go | 23 +++++++++++++++++++++++
 internal/rootcoord/meta_table.go       |  6 +++---
 2 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/internal/metastore/model/collection.go b/internal/metastore/model/collection.go
index 66acf68cf248c..08f6cf08ff1c0 100644
--- a/internal/metastore/model/collection.go
+++ b/internal/metastore/model/collection.go
@@ -34,6 +34,29 @@ func (c *Collection) Available() bool {
 	return c.State == pb.CollectionState_CollectionCreated
 }
 
+func (c *Collection) ShadowClone() *Collection {
+	return &Collection{
+		TenantID:             c.TenantID,
+		DBID:                 c.DBID,
+		CollectionID:         c.CollectionID,
+		Name:                 c.Name,
+		Description:          c.Description,
+		AutoID:               c.AutoID,
+		Fields:               c.Fields,
+		Partitions:           c.Partitions,
+		VirtualChannelNames:  c.VirtualChannelNames,
+		PhysicalChannelNames: c.PhysicalChannelNames,
+		ShardsNum:            c.ShardsNum,
+		ConsistencyLevel:     c.ConsistencyLevel,
+		CreateTime:           c.CreateTime,
+		StartPositions:       c.StartPositions,
+		Aliases:              c.Aliases,
+		Properties:           c.Properties,
+		State:                c.State,
+		EnableDynamicField:   c.EnableDynamicField,
+	}
+}
+
 func (c *Collection) Clone() *Collection {
 	return &Collection{
 		TenantID:             c.TenantID,
diff --git a/internal/rootcoord/meta_table.go b/internal/rootcoord/meta_table.go
index 8616ce6dc9092..f84055d5495ec 100644
--- a/internal/rootcoord/meta_table.go
+++ b/internal/rootcoord/meta_table.go
@@ -537,12 +537,12 @@ func (mt *MetaTable) RemoveCollection(ctx context.Context, collectionID UniqueID
 }
 
 func filterUnavailable(coll *model.Collection) *model.Collection {
-	clone := coll.Clone()
+	clone := coll.ShadowClone()
 	// pick available partitions.
-	clone.Partitions = nil
+	clone.Partitions = make([]*model.Partition, 0, len(coll.Partitions))
 	for _, partition := range coll.Partitions {
 		if partition.Available() {
-			clone.Partitions = append(clone.Partitions, partition.Clone())
+			clone.Partitions = append(clone.Partitions, partition)
 		}
 	}
 	return clone