fix double close

babylonlabs-io · Dec 19, 2024 · ae10617 · ae10617
1 parent bc643d1
commit ae10617
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/btcstaking-tracker/stakingeventwatcher/stakingeventwatcher.go b/btcstaking-tracker/stakingeventwatcher/stakingeventwatcher.go
@@ -676,7 +676,8 @@ func (sew *StakingEventWatcher) activateBtcDelegation(
 	requiredDepth uint32,
 	done chan struct{},
 ) {
-	defer close(done)
+	var once sync.Once
+	defer once.Do(func() { close(done) })
 	sew.metrics.NumberOfActivationInProgress.Inc()
 	defer sew.metrics.NumberOfActivationInProgress.Dec()
 
@@ -714,7 +715,9 @@ func (sew *StakingEventWatcher) activateBtcDelegation(
 
 		sew.metrics.ReportedActivateDelegationsCounter.Inc()
 
-		close(done)
+		once.Do(func() {
+			close(done)
+		})
 
 		sew.pendingTracker.RemoveDelegation(stakingTxHash)
 		sew.metrics.NumberOfVerifiedDelegations.Dec()