From b92aa0358a12b13165b3c468212284f862b65575 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=A3=8E=E6=89=87=E6=BB=91=E7=BF=94=E7=BF=BC?=
 <Fangliding.fshxy@outlook.com>
Date: Sun, 17 May 2026 13:21:49 +0000
Subject: [PATCH] More

---
 common/signal/timer.go | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/common/signal/timer.go b/common/signal/timer.go
index 86bdd892f..365a05dc2 100644
--- a/common/signal/timer.go
+++ b/common/signal/timer.go
@@ -3,7 +3,6 @@ package signal
 import (
 	"context"
 	"sync"
-	"sync/atomic"
 	"time"
 )
 
@@ -12,24 +11,31 @@ type ActivityUpdater interface {
 }
 
 type ActivityTimer struct {
-	mu        sync.Mutex
-	timer     atomic.Pointer[time.Timer]
+	mu sync.Mutex
+	// timer will be nil if this timer is already finished
+	timer     *time.Timer
 	timeout   time.Duration
 	onTimeout func()
 }
 
 func (t *ActivityTimer) Update() {
-	t.mu.Lock()
+	// someone already called Update or closing, just return
+	if !t.mu.TryLock() {
+		return
+	}
 	defer t.mu.Unlock()
-	if timer := t.timer.Load(); timer != nil {
-		timer.Reset(t.timeout)
+	if t.timer != nil {
+		t.timer.Reset(t.timeout)
 	}
 }
 
 func (t *ActivityTimer) finish() {
-	if timer := t.timer.Swap(nil); timer != nil {
-		timer.Stop()
+	t.mu.Lock()
+	defer t.mu.Unlock()
+	if t.timer != nil {
+		t.timer.Stop()
 		t.onTimeout()
+		t.timer = nil
 	}
 }
 
@@ -41,9 +47,9 @@ func (t *ActivityTimer) SetTimeout(timeout time.Duration) {
 
 	t.mu.Lock()
 	defer t.mu.Unlock()
-	if timer := t.timer.Load(); timer != nil {
+	if t.timer != nil {
 		t.timeout = timeout
-		timer.Reset(timeout)
+		t.timer.Reset(timeout)
 	}
 }
 
@@ -52,10 +58,11 @@ func CancelAfterInactivity(ctx context.Context, cancel context.CancelFunc, timeo
 		timeout:   timeout,
 		onTimeout: cancel,
 	}
+	// strange situation
 	if timeout == 0 {
 		cancel()
 		return activityTimer
 	}
-	activityTimer.timer.Store(time.AfterFunc(timeout, activityTimer.finish))
+	activityTimer.timer = time.AfterFunc(timeout, activityTimer.finish)
 	return activityTimer
 }