lzc256
/
tailscale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

types/logger: add TS_DEBUG_LOG_RATE knob to easily turn off rate limiting

reviewable/pr423/r1
Brad Fitzpatrick 2020-05-29 12:30:43 -07:00
parent cf5d25e15b
commit dd6b96ba68
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
types/logger/logger.go
View File

@ -14,6 +14,7 @@ import (
"io" "io"
"io/ioutil" "io/ioutil"
"log" "log"
"os"
"sync" "sync"
"time" "time"
@ -60,10 +61,15 @@ type limitData struct {
ele *list.Element // list element used to access this string in the cache ele *list.Element // list element used to access this string in the cache
} }
var disableRateLimit = os.Getenv("TS_DEBUG_LOG_RATE") == "all"
// RateLimitedFn returns a rate-limiting Logf wrapping the given logf. // RateLimitedFn returns a rate-limiting Logf wrapping the given logf.
// Messages are allowed through at a maximum of one message every f (where f is a time.Duration), in // Messages are allowed through at a maximum of one message every f (where f is a time.Duration), in
// bursts of up to burst messages at a time. Up to maxCache strings will be held at a time. // bursts of up to burst messages at a time. Up to maxCache strings will be held at a time.
func RateLimitedFn(logf Logf, f time.Duration, burst int, maxCache int) Logf { func RateLimitedFn(logf Logf, f time.Duration, burst int, maxCache int) Logf {
if disableRateLimit {
return logf
}
r := rate.Every(f) r := rate.Every(f)
var ( var (
mu sync.Mutex mu sync.Mutex