extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-06-19 08:17:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

min/max pacing (latency between requests) (#40)

Browse Source
This commit is contained in:
Michael Quigley 2022-10-12 12:40:17 -04:00
parent f17a1f1446
commit 01f056784a
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/zrok/loop.go
View File

@ -36,6 +36,8 @@ type loopCmd struct {
timeoutSeconds int timeoutSeconds int
minPayload int minPayload int
maxPayload int maxPayload int
minPacingMs int
maxPacingMs int
} }
func newLoopCmd() *loopCmd { func newLoopCmd() *loopCmd {
@ -53,6 +55,8 @@ func newLoopCmd() *loopCmd {
cmd.Flags().IntVarP(&r.timeoutSeconds, "timeout-seconds", "T", 30, "Time out after # seconds when sending http requests") cmd.Flags().IntVarP(&r.timeoutSeconds, "timeout-seconds", "T", 30, "Time out after # seconds when sending http requests")
cmd.Flags().IntVar(&r.minPayload, "min-payload", 64, "Minimum payload size in bytes") cmd.Flags().IntVar(&r.minPayload, "min-payload", 64, "Minimum payload size in bytes")
cmd.Flags().IntVar(&r.maxPayload, "max-payload", 10240, "Maximum payload size in bytes") cmd.Flags().IntVar(&r.maxPayload, "max-payload", 10240, "Maximum payload size in bytes")
cmd.Flags().IntVar(&r.minPacingMs, "min-pacing-ms", 0, "Minimum pacing in milliseconds")
cmd.Flags().IntVar(&r.maxPacingMs, "max-pacing-ms", 0, "Maximum pacing in milliseconds")
return r return r
} }
@ -211,6 +215,11 @@ func (l *looper) iterate() {
} else { } else {
logrus.Errorf("looper #%d error creating request: %v", l.id, err) logrus.Errorf("looper #%d error creating request: %v", l.id, err)
} }
pacingMs := l.cmd.maxPayload
if l.cmd.maxPacingMs-l.cmd.minPacingMs > 0 {
pacingMs = rand.Intn(l.cmd.maxPacingMs-l.cmd.minPacingMs) + l.cmd.minPacingMs
time.Sleep(time.Duration(pacingMs) * time.Millisecond)
}
} }
} }