deblan/capture
deblan/capture
1
1
Fork 0
Code Issues Pull requests Packages Projects Releases 6 Wiki Activity

change ReadAll to TeeReader

Browse source
This commit is contained in:
Fabricio 2019-11-16 13:50:13 -03:00
parent d907a55566
commit 85d0a71d13
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
main.go
View file

@ -6,6 +6,7 @@ import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"html/template" "html/template"
"io"
"io/ioutil" "io/ioutil"
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
@ -167,9 +168,9 @@ func NewRecorderHandler(srv *CaptureService, next http.HandlerFunc) http.Handler
return func(rw http.ResponseWriter, r *http.Request) { return func(rw http.ResponseWriter, r *http.Request) {
// Save req body for later. // Save req body for later.
reqBody, _ := ioutil.ReadAll(r.Body)
r.Body.Close() reqBody := &bytes.Buffer{}
r.Body = ioutil.NopCloser(bytes.NewReader(reqBody)) r.Body = ioutil.NopCloser(io.TeeReader(r.Body, reqBody))
rec := httptest.NewRecorder() rec := httptest.NewRecorder()
@ -199,7 +200,7 @@ func NewRecorderHandler(srv *CaptureService, next http.HandlerFunc) http.Handler
Url: r.URL.String(), Url: r.URL.String(),
Path: r.URL.Path, Path: r.URL.Path,
Header: r.Header, Header: r.Header,
Body: reqBody, Body: reqBody.Bytes(),
} }
res := Res{ res := Res{
Proto: rec.Result().Proto, Proto: rec.Result().Proto,