From 1c2ae59ece6b8850a4201876af79430343508495 Mon Sep 17 00:00:00 2001 From: congqixia Date: Mon, 8 Apr 2024 10:35:17 +0800 Subject: [PATCH] fix: [skip e2e] Dedup available ports and retry for integration setup (#31902) See also #31901 Signed-off-by: Congqi Xia --- tests/integration/minicluster_v2.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/integration/minicluster_v2.go b/tests/integration/minicluster_v2.go index da9fa588a0..7f7cbfca7e 100644 --- a/tests/integration/minicluster_v2.go +++ b/tests/integration/minicluster_v2.go @@ -51,6 +51,7 @@ import ( "github.com/milvus-io/milvus/pkg/log" "github.com/milvus-io/milvus/pkg/util/etcd" "github.com/milvus-io/milvus/pkg/util/paramtable" + "github.com/milvus-io/milvus/pkg/util/typeutil" ) var params *paramtable.ComponentParam = paramtable.Get() @@ -427,15 +428,15 @@ func (cluster *MiniClusterV2) GetFactory() dependency.Factory { } func (cluster *MiniClusterV2) GetAvailablePorts(n int) ([]int, error) { - ports := make([]int, n) - for i := range ports { + ports := typeutil.NewSet[int]() + for ports.Len() < n { port, err := cluster.GetAvailablePort() if err != nil { return nil, err } - ports[i] = port + ports.Insert(port) } - return ports, nil + return ports.Collect(), nil } func (cluster *MiniClusterV2) GetAvailablePort() (int, error) {