Mirroristas/baton
2
0
Fork 0
mirror of https://github.com/americanexpress/baton.git synced 2025-11-23 22:15:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3 from madflojo/canada

Browse Source 
Anonymous contribution
This commit is contained in:
Benjamin Cane 2020-05-10 11:14:41 -07:00 committed by GitHub
commit 5763390413
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
worker.go
View File

@ -19,6 +19,7 @@ package main
import ( import (
"github.com/valyala/fasthttp" "github.com/valyala/fasthttp"
"math/rand" "math/rand"
"strings"
"time" "time"
) )
@ -97,6 +98,10 @@ func buildRequest(requests []preLoadedRequest, totalPremadeRequests int) (*fasth
req.Header.SetMethod(currentReq.method) req.Header.SetMethod(currentReq.method)
req.SetBodyString(currentReq.body) req.SetBodyString(currentReq.body)
for i := 0; i < len(currentReq.headers); i++ { for i := 0; i < len(currentReq.headers); i++ {
if strings.EqualFold(currentReq.headers[i][0], "Content-type") {
req.Header.SetContentType(currentReq.headers[i][1])
continue
}
req.Header.Add(currentReq.headers[i][0], currentReq.headers[i][1]) req.Header.Add(currentReq.headers[i][0], currentReq.headers[i][1])
} }
return req, resp return req, resp