fix: channel close once

strahe · Dec 11, 2024 · 2c8c3e1 · 2c8c3e1
1 parent 42a7b02
commit 2c8c3e1
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/service.go b/service.go
@@ -43,10 +43,11 @@ const (
 )
 
 type watchMessage struct {
-	id       uint
-	db       *gorm.DB
-	started  time.Time
-	cancelCh chan struct{}
+	id         uint
+	db         *gorm.DB
+	started    time.Time
+	cancelOnce sync.Once
+	cancelCh   chan struct{}
 }
 
 func newWatchMessage(db *gorm.DB, id uint) *watchMessage {
@@ -59,7 +60,9 @@ func newWatchMessage(db *gorm.DB, id uint) *watchMessage {
 }
 
 func (w *watchMessage) Cancel() {
-	close(w.cancelCh)
+	w.cancelOnce.Do(func() {
+		close(w.cancelCh)
+	})
 }
 
 func (w *watchMessage) Wait() {