oc-discovery -> conf

daemons/node/indexer/behavior.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"oc-discovery/conf"
+	"oc-discovery/daemons/node/common"
 
 	pp "github.com/libp2p/go-libp2p/core/peer"
 )
@@ -13,62 +14,14 @@ import (
 // ── defaults ──────────────────────────────────────────────────────────────────
 
 const (
-	defaultMaxConnPerWindow  = 20
-	defaultConnWindowSecs    = 30
-	defaultMaxHBPerMinute    = 5
-	defaultMaxPublishPerMin  = 10
-	defaultMaxGetPerMin      = 50
-	strikeThreshold          = 3
-	banDuration              = 10 * time.Minute
-	behaviorWindowDur        = 60 * time.Second
+	defaultMaxHBPerMinute   = 5
+	defaultMaxPublishPerMin = 10
+	defaultMaxGetPerMin     = 50
+	strikeThreshold         = 3
+	banDuration             = 10 * time.Minute
+	behaviorWindowDur       = 60 * time.Second
 )
 
-func cfgOr(v, def int) int {
-	if v > 0 {
-		return v
-	}
-	return def
-}
-
-// ── ConnectionRateGuard ───────────────────────────────────────────────────────
-
-// ConnectionRateGuard limits the number of NEW incoming connections accepted
-// within a sliding time window.  It protects public indexers against coordinated
-// registration floods (Sybil bursts).
-type ConnectionRateGuard struct {
-	mu          sync.Mutex
-	window      []time.Time
-	maxInWindow int
-	windowDur   time.Duration
-}
-
-func newConnectionRateGuard() *ConnectionRateGuard {
-	cfg := conf.GetConfig()
-	return &ConnectionRateGuard{
-		maxInWindow: cfgOr(cfg.MaxConnPerWindow, defaultMaxConnPerWindow),
-		windowDur:   time.Duration(cfgOr(cfg.ConnWindowSecs, defaultConnWindowSecs)) * time.Second,
-	}
-}
-
-// Allow returns true if a new connection may be accepted.
-// The internal window is pruned on each call so memory stays bounded.
-func (g *ConnectionRateGuard) Allow() bool {
-	g.mu.Lock()
-	defer g.mu.Unlock()
-	now := time.Now()
-	cutoff := now.Add(-g.windowDur)
-	i := 0
-	for i < len(g.window) && g.window[i].Before(cutoff) {
-		i++
-	}
-	g.window = g.window[i:]
-	if len(g.window) >= g.maxInWindow {
-		return false
-	}
-	g.window = append(g.window, now)
-	return true
-}
-
 // ── per-node state ────────────────────────────────────────────────────────────
 
 type nodeBehavior struct {
@@ -130,9 +83,9 @@ func newNodeBehaviorTracker() *NodeBehaviorTracker {
 	cfg := conf.GetConfig()
 	return &NodeBehaviorTracker{
 		nodes:  make(map[pp.ID]*nodeBehavior),
-		maxHB:  cfgOr(cfg.MaxHBPerMinute, defaultMaxHBPerMinute),
-		maxPub: cfgOr(cfg.MaxPublishPerMinute, defaultMaxPublishPerMin),
-		maxGet: cfgOr(cfg.MaxGetPerMinute, defaultMaxGetPerMin),
+		maxHB:  common.CfgOr(cfg.MaxHBPerMinute, defaultMaxHBPerMinute),
+		maxPub: common.CfgOr(cfg.MaxPublishPerMinute, defaultMaxPublishPerMin),
+		maxGet: common.CfgOr(cfg.MaxGetPerMinute, defaultMaxGetPerMin),
 	}
 }