need be careful to use Timer.Reset

Signed-off-by: lhy1024 <[email protected]>

client/tso_dispatcher.go

@@ -158,6 +158,15 @@ func NewTSDeadline(
 	cancel context.CancelFunc,
 ) *TSDeadline {
 	timer := timerPool.Get().(*time.Timer)
+	// Stop the timer if it's not stopped.
+	if !timer.Stop() {
+		select {
+		case <-timer.C: // try to drain from the channel
+		default:
+		}
+	}
+	// We need be careful here, see more details in the comments of Timer.Reset.
+	// https://pkg.go.dev/time@master#Timer.Reset
 	timer.Reset(timeout)
 	return &TSDeadline{
 		timer:  timer,
@@ -419,6 +428,15 @@ tsoBatchLoop:
 		if maxBatchWaitInterval >= 0 {
 			tbc.adjustBestBatchSize()
 		}
+		// Stop the timer if it's not stopped.
+		if !streamLoopTimer.Stop() {
+			select {
+			case <-streamLoopTimer.C: // try to drain from the channel
+			default:
+			}
+		}
+		// We need be careful here, see more details in the comments of Timer.Reset.
+		// https://pkg.go.dev/time@master#Timer.Reset
 		streamLoopTimer.Reset(c.option.timeout)
 		// Choose a stream to send the TSO gRPC request.
 	streamChoosingLoop:

pkg/utils/tsoutil/tso_dispatcher.go

@@ -215,6 +215,15 @@ func NewTSDeadline(
 	cancel context.CancelFunc,
 ) *TSDeadline {
 	timer := timerPool.Get().(*time.Timer)
+	// Stop the timer if it's not stopped.
+	if !timer.Stop() {
+		select {
+		case <-timer.C: // try to drain from the channel
+		default:
+		}
+	}
+	// We need be careful here, see more details in the comments of Timer.Reset.
+	// https://pkg.go.dev/time@master#Timer.Reset
 	timer.Reset(timeout)
 	return &TSDeadline{
 		timer:  timer,