chore: tweaks, simplification, tests

fileproc/resource_monitor_rate_limiting.go

@@ -0,0 +1,36 @@
+package fileproc
+
+import (
+	"context"
+	"time"
+
+	"github.com/sirupsen/logrus"
+)
+
+// WaitForRateLimit waits for rate limiting if enabled.
+func (rm *ResourceMonitor) WaitForRateLimit(ctx context.Context) error {
+	if !rm.enabled || rm.rateLimitFilesPerSec <= 0 {
+		return nil
+	}
+
+	select {
+	case <-ctx.Done():
+		return ctx.Err()
+	case <-rm.rateLimitChan:
+		return nil
+	case <-time.After(time.Second): // Fallback timeout
+		logrus.Warn("Rate limiting timeout exceeded, continuing without rate limit")
+		return nil
+	}
+}
+
+// rateLimiterRefill refills the rate limiting channel periodically.
+func (rm *ResourceMonitor) rateLimiterRefill() {
+	for range rm.rateLimiter.C {
+		select {
+		case rm.rateLimitChan <- struct{}{}:
+		default:
+			// Channel is full, skip
+		}
+	}
+}