extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-01-11 08:29:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

dwell and timeout (#40)

Browse Source
This commit is contained in:
Michael Quigley 2022-10-04 13:28:16 -04:00
parent 35c9733668
commit 9f567513f5
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
cmd/zrokloop/loop.go
View File

@ -25,10 +25,12 @@ func init() {
} }
type run struct { type run struct {
cmd *cobra.Command cmd *cobra.Command
loopers int loopers int
iterations int iterations int
statusEvery int statusEvery int
dwellSeconds int
timeoutSeconds int
} }
func newRun() *run { func newRun() *run {
@ -42,6 +44,8 @@ func newRun() *run {
cmd.Flags().IntVarP(&r.loopers, "loopers", "l", 1, "Number of current loopers to start") cmd.Flags().IntVarP(&r.loopers, "loopers", "l", 1, "Number of current loopers to start")
cmd.Flags().IntVarP(&r.iterations, "iterations", "i", 1, "Number of iterations per looper") cmd.Flags().IntVarP(&r.iterations, "iterations", "i", 1, "Number of iterations per looper")
cmd.Flags().IntVarP(&r.statusEvery, "status-every", "E", 100, "Show status every # iterations") cmd.Flags().IntVarP(&r.statusEvery, "status-every", "E", 100, "Show status every # iterations")
cmd.Flags().IntVarP(&r.dwellSeconds, "dwell-seconds", "D", 1, "Dwell # seconds before starting iterations")
cmd.Flags().IntVarP(&r.timeoutSeconds, "timeout-seconds", "T", 30, "Time out after # seconds when sending http requests")
return r return r
} }
@ -59,15 +63,15 @@ func (r *run) run(_ *cobra.Command, _ []string) {
type looper struct { type looper struct {
id int id int
runi *run r *run
done chan struct{} done chan struct{}
listener edge.Listener listener edge.Listener
} }
func newLooper(id int, runi *run) *looper { func newLooper(id int, r *run) *looper {
return &looper{ return &looper{
id: id, id: id,
runi: runi, r: r,
done: make(chan struct{}), done: make(chan struct{}),
} }
} }
@ -104,17 +108,17 @@ func (l *looper) run() {
logrus.Infof("looper #%d, service: %v, frontend: %v", l.id, tunnelResp.Payload.Service, tunnelResp.Payload.ProxyEndpoint) logrus.Infof("looper #%d, service: %v, frontend: %v", l.id, tunnelResp.Payload.Service, tunnelResp.Payload.ProxyEndpoint)
go l.serviceListener(zif, tunnelResp.Payload.Service) go l.serviceListener(zif, tunnelResp.Payload.Service)
time.Sleep(1 * time.Second) time.Sleep(time.Duration(l.r.dwellSeconds) * time.Second)
for i := 0; i < l.runi.iterations; i++ { for i := 0; i < l.r.iterations; i++ {
if i > 0 && i%l.runi.statusEvery == 0 { if i > 0 && i%l.r.statusEvery == 0 {
logrus.Infof("looper #%d: iteration #%d", l.id, i) logrus.Infof("looper #%d: iteration #%d", l.id, i)
} }
outpayload := make([]byte, 10240) outpayload := make([]byte, 10240)
outbase64 := base64.StdEncoding.EncodeToString(outpayload) outbase64 := base64.StdEncoding.EncodeToString(outpayload)
rand.Read(outpayload) rand.Read(outpayload)
if req, err := http.NewRequest("POST", tunnelResp.Payload.ProxyEndpoint, bytes.NewBufferString(outbase64)); err == nil { if req, err := http.NewRequest("POST", tunnelResp.Payload.ProxyEndpoint, bytes.NewBufferString(outbase64)); err == nil {
client := &http.Client{Timeout: time.Second * 10} client := &http.Client{Timeout: time.Second * time.Duration(l.r.timeoutSeconds)}
if resp, err := client.Do(req); err == nil { if resp, err := client.Do(req); err == nil {
inpayload := new(bytes.Buffer) inpayload := new(bytes.Buffer)
io.Copy(inpayload, resp.Body) io.Copy(inpayload, resp.Body)