From 5317c8ababd4ccb9975a68172dd09cf08cc8fb79 Mon Sep 17 00:00:00 2001
From: congqixia <congqi.xia@zilliz.com>
Date: Wed, 27 Oct 2021 14:22:22 +0800
Subject: [PATCH] Fix waitNewTSafe not waiting (#10724)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
---
 internal/querynode/query_collection.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/internal/querynode/query_collection.go b/internal/querynode/query_collection.go
index a91d17f02b..4c643dc38a 100644
--- a/internal/querynode/query_collection.go
+++ b/internal/querynode/query_collection.go
@@ -221,6 +221,8 @@ func (q *queryCollection) waitNewTSafe() (Timestamp, error) {
 	for !q.tSafeUpdate {
 		q.watcherCond.Wait()
 	}
+	q.tSafeUpdate = false
+	q.watcherCond.Broadcast()
 	q.watcherCond.L.Unlock()
 	q.tSafeWatchersMu.RLock()
 	defer q.tSafeWatchersMu.RUnlock()