mirror of
https://github.com/ivuorinen/gibidify.git
synced 2026-01-26 11:34:03 +00:00
e9bd694685
Add overflow checks before converting uint64 memory values to int64 to prevent potential integer overflow issues identified by gosec (G115). - Add math.MaxInt64 checks in fileproc/backpressure.go - Add math.MaxInt64 checks in fileproc/resource_monitor_validation.go - Add math.MaxInt64 checks in fileproc/resource_monitor_metrics.go - Add math.MaxInt64 check in benchmark/benchmark.go with nosec annotation Co-authored-by: ivuorinen <11024+ivuorinen@users.noreply.github.com>
37 lines
761 B
Go
37 lines
761 B
Go
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
|
|
}
|
|
}
|
|
}
|