From 3bb69430cb22beeb856731ff90c7c3684ac7e4a2 Mon Sep 17 00:00:00 2001
From: xige-16 <xi.ge@zilliz.com>
Date: Thu, 22 Apr 2021 09:23:49 +0800
Subject: [PATCH] delete contex timeout in queryService's load (#4986)

Signed-off-by: xige-16 <xi.ge@zilliz.com>
---
 internal/queryservice/impl.go | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/internal/queryservice/impl.go b/internal/queryservice/impl.go
index ce65310587..0b9c5a7176 100644
--- a/internal/queryservice/impl.go
+++ b/internal/queryservice/impl.go
@@ -178,12 +178,10 @@ func (qs *QueryService) LoadCollection(ctx context.Context, req *querypb.LoadCol
 			return status, err
 		}
 	}
-	loadCtx, cancel := context.WithTimeout(qs.loopCtx, 30*time.Second)
 	loadCollectionTask := &LoadCollectionTask{
 		BaseTask: BaseTask{
-			ctx:       loadCtx,
-			cancel:    cancel,
-			Condition: NewTaskCondition(loadCtx),
+			ctx:       qs.loopCtx,
+			Condition: NewTaskCondition(qs.loopCtx),
 		},
 		LoadCollectionRequest: req,
 		masterService:         qs.masterServiceClient,
@@ -283,12 +281,10 @@ func (qs *QueryService) LoadPartitions(ctx context.Context, req *querypb.LoadPar
 		return status, err
 	}
 
-	releaseCtx, cancel := context.WithTimeout(qs.loopCtx, 30*time.Second)
 	loadPartitionTask := &LoadPartitionTask{
 		BaseTask: BaseTask{
-			ctx:       releaseCtx,
-			cancel:    cancel,
-			Condition: NewTaskCondition(releaseCtx),
+			ctx:       qs.loopCtx,
+			Condition: NewTaskCondition(qs.loopCtx),
 		},
 		LoadPartitionsRequest: req,
 		masterService:         qs.masterServiceClient,