Mirroristas/baton
2
0
Fork 0
mirror of https://github.com/americanexpress/baton.git synced 2025-07-06 12:22:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Respect noCopy in fasthttp.RequestHeader

Browse Source
This commit is contained in:
Panayiotis Pastos 2018-02-20 12:55:16 +00:00
parent 100dfcd352
commit dd97c7ec82
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
baton_test.go
View File

@ -32,7 +32,7 @@ type HTTPTestHandler struct {
lastBodyReceived string lastBodyReceived string
lastMethodReceived string lastMethodReceived string
lastURIReceived string lastURIReceived string
lastHeadersReceived fasthttp.RequestHeader lastHeadersReceived *fasthttp.RequestHeader
lastTimestamp int64 lastTimestamp int64
} }
@ -42,7 +42,10 @@ func (h *HTTPTestHandler) HandleRequest(ctx *fasthttp.RequestCtx) {
h.lastBodyReceived = hex.EncodeToString(ctx.Request.Body()) h.lastBodyReceived = hex.EncodeToString(ctx.Request.Body())
h.lastMethodReceived = string(ctx.Request.Header.Method()) h.lastMethodReceived = string(ctx.Request.Header.Method())
h.lastURIReceived = ctx.Request.URI().String() h.lastURIReceived = ctx.Request.URI().String()
h.lastHeadersReceived = ctx.Request.Header newHeader := fasthttp.RequestHeader{}
ctx.Request.Header.CopyTo(&newHeader)
h.lastHeadersReceived = &newHeader
} }
func (h *HTTPTestHandler) reset() { func (h *HTTPTestHandler) reset() {
@ -59,7 +62,7 @@ var port = "8888"
func startServer() *HTTPTestHandler { func startServer() *HTTPTestHandler {
if !serverRunning { if !serverRunning {
internalHandlerRef = &HTTPTestHandler{0, "", "", "", fasthttp.RequestHeader{}, 0} internalHandlerRef = &HTTPTestHandler{0, "", "", "", &fasthttp.RequestHeader{}, 0}
serverRunning = true serverRunning = true
go func() { go func() {
err := fasthttp.ListenAndServe(":"+port, internalHandlerRef.HandleRequest) err := fasthttp.ListenAndServe(":"+port, internalHandlerRef.HandleRequest)