Increase timeout of activate task (#19330)

Signed-off-by: Nxige-16 <xi.ge@zilliz.com>
Signed-off-by: Nxige-16 <xi.ge@zilliz.com>

configs/milvus.yaml

@@ -181,7 +181,7 @@ queryCoord:
   memoryUsageMaxDifferencePercentage: 30
   checkInterval: 1000
   channelTaskTimeout: 60000 # 1 minute
-  segmentTaskTimeout: 15000 # 15 seconds
+  segmentTaskTimeout: 120000 # 2 minute
   distPullInterval: 500
   loadTimeoutSeconds: 600
   checkHandoffInterval: 5000

internal/querycoordv2/handlers.go

@@ -5,7 +5,9 @@ import (
 	"errors"
 	"fmt"
 	"sync"
-	"time"
+
+	"github.com/samber/lo"
+	"go.uber.org/zap"
 
 	"github.com/milvus-io/milvus/api/commonpb"
 	"github.com/milvus-io/milvus/api/milvuspb"
@@ -19,8 +21,6 @@ import (
 	"github.com/milvus-io/milvus/internal/util/metricsinfo"
 	"github.com/milvus-io/milvus/internal/util/typeutil"
 	"github.com/milvus-io/milvus/internal/util/uniquegenerator"
-	"github.com/samber/lo"
-	"go.uber.org/zap"
 )
 
 // checkAnyReplicaAvailable checks if the collection has enough distinct available shards. These shards
@@ -60,9 +60,7 @@ func (s *Server) getCollectionSegmentInfo(collection int64) []*querypb.SegmentIn
 // parseBalanceRequest parses the load balance request,
 // returns the collection, replica, and segments
 func (s *Server) balanceSegments(ctx context.Context, req *querypb.LoadBalanceRequest, replica *meta.Replica) error {
-	const (
-		manualBalanceTimeout = 10 * time.Second
-	)
+	manualBalanceTimeout := Params.QueryCoordCfg.SegmentTaskTimeout
 
 	srcNode := req.GetSourceNodeIDs()[0]
 	dstNodeSet := typeutil.NewUniqueSet(req.GetDstNodeIDs()...)

internal/util/paramtable/component_param.go

@@ -723,7 +723,7 @@ func (p *queryCoordConfig) initChannelTaskTimeout() {
 }
 
 func (p *queryCoordConfig) initSegmentTaskTimeout() {
-	timeout := p.Base.LoadWithDefault("queryCoord.segmentTaskTimeout", "15000")
+	timeout := p.Base.LoadWithDefault("queryCoord.segmentTaskTimeout", "120000")
 	taskTimeout, err := strconv.ParseInt(timeout, 10, 64)
 	if err != nil {
 		panic(err)