From ccd79551d0960333ec175b32e0db616bf9d083ad Mon Sep 17 00:00:00 2001
From: Lazar <lukic.lazar@yahoo.com>
Date: Fri, 6 Dec 2024 16:14:27 +0100
Subject: [PATCH 1/2] add semaphore to rate limit activations

---
 .../stakingeventwatcher/stakingeventwatcher.go | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/btcstaking-tracker/stakingeventwatcher/stakingeventwatcher.go b/btcstaking-tracker/stakingeventwatcher/stakingeventwatcher.go
index 613a4ab..d16b694 100644
--- a/btcstaking-tracker/stakingeventwatcher/stakingeventwatcher.go
+++ b/btcstaking-tracker/stakingeventwatcher/stakingeventwatcher.go
@@ -5,6 +5,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"golang.org/x/sync/semaphore"
 	"sync"
 	"sync/atomic"
 	"time"
@@ -26,8 +27,9 @@ import (
 )
 
 var (
-	fixedDelyTypeWithJitter = retry.DelayType(retry.CombineDelay(retry.FixedDelay, retry.RandomDelay))
-	retryForever            = retry.Attempts(0)
+	fixedDelyTypeWithJitter  = retry.DelayType(retry.CombineDelay(retry.FixedDelay, retry.RandomDelay))
+	retryForever             = retry.Attempts(0)
+	maxConcurrentActivations = int64(1000)
 )
 
 func (sew *StakingEventWatcher) quitContext() (context.Context, func()) {
@@ -83,6 +85,7 @@ type StakingEventWatcher struct {
 	unbondingDelegationChan chan *newDelegation
 	unbondingRemovalChan    chan *delegationInactive
 	currentBestBlockHeight  atomic.Uint32
+	activationLimiter       *semaphore.Weighted
 }
 
 func NewStakingEventWatcher(
@@ -106,6 +109,7 @@ func NewStakingEventWatcher(
 		inProgressTracker:       NewTrackedDelegations(),
 		unbondingDelegationChan: make(chan *newDelegation),
 		unbondingRemovalChan:    make(chan *delegationInactive),
+		activationLimiter:       semaphore.NewWeighted(maxConcurrentActivations), // todo(lazar): this should be in config
 	}
 }
 
@@ -613,6 +617,11 @@ func (sew *StakingEventWatcher) checkBtcForStakingTx() {
 			continue
 		}
 
+		if err := sew.activationLimiter.Acquire(context.Background(), 1); err != nil {
+			sew.logger.Warnf("error acquiring a activation semaphore %s", err)
+			continue
+		}
+
 		if _, err = sew.inProgressTracker.AddDelegation(
 			del.StakingTx,
 			del.StakingOutputIdx,
@@ -624,7 +633,10 @@ func (sew *StakingEventWatcher) checkBtcForStakingTx() {
 			continue
 		}
 
-		go sew.activateBtcDelegation(txHash, proof, details.Block.BlockHash(), params.ConfirmationTimeBlocks)
+		go func() {
+			defer sew.activationLimiter.Release(1)
+			sew.activateBtcDelegation(txHash, proof, details.Block.BlockHash(), params.ConfirmationTimeBlocks)
+		}()
 	}
 }
 

From 16dc34eb75808f415b0232f093310deed8e81071 Mon Sep 17 00:00:00 2001
From: Lazar <lukic.lazar@yahoo.com>
Date: Fri, 6 Dec 2024 16:16:12 +0100
Subject: [PATCH 2/2] changelog

---
 CHANGELOG.md | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 7a34d22..1a8fded 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -37,6 +37,10 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 
 ## Unreleased
 
+### Improvements
+
+* [#136](https://github.com/babylonlabs-io/vigilante/pull/136) rate limit activations
+
 ## v0.18.0
 
 ### Improvements