add timerPool.Put

Signed-off-by: lhy1024 <admin@liudos.us>
tikv · Jun 29, 2023 · b5c5c68 · b5c5c68
1 parent 1ae3f52
commit b5c5c68
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/client/tso_dispatcher.go b/client/tso_dispatcher.go
@@ -211,6 +211,8 @@ func (c *tsoClient) watchTSDeadline(ctx context.Context, dcLocation string) {
 						timerPool.Put(d.timer)
 						continue
 					case <-ctx.Done():
+						d.timer.Stop() // not received from timer.C, so we need to stop the timer
+						timerPool.Put(d.timer)
 						return
 					}
 				case <-ctx.Done():

diff --git a/pkg/utils/tsoutil/tso_dispatcher.go b/pkg/utils/tsoutil/tso_dispatcher.go
@@ -243,6 +243,8 @@ func WatchTSDeadline(ctx context.Context, tsDeadlineCh <-chan *TSDeadline) {
 				timerPool.Put(d.timer)
 				continue
 			case <-ctx.Done():
+				d.timer.Stop() // not received from timer.C, so we need to stop the timer
+				timerPool.Put(d.timer)
 				return
 			}
 		case <-ctx.Done():