-
Notifications
You must be signed in to change notification settings - Fork 57
/
debug.go
46 lines (40 loc) · 850 Bytes
/
debug.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package iotservice
import (
"bytes"
"net/http"
"net/http/httputil"
)
type requestOutDump http.Request
func (r *requestOutDump) String() string {
b, err := httputil.DumpRequestOut((*http.Request)(r), true)
if err != nil {
panic(err)
}
return prefix(b, "> ")
}
type responseDump http.Response
func (r *responseDump) String() string {
b, err := httputil.DumpResponse((*http.Response)(r), true)
if err != nil {
panic(err)
}
return prefix(b, "< ")
}
func prefix(b []byte, prefix string) string {
off := 0
buf := bytes.NewBuffer(make([]byte, 0,
len(b)+(bytes.Count(b, []byte{'\n'})*len(prefix)+len(prefix))),
)
buf.WriteString(prefix)
for {
i := bytes.Index(b[off:], []byte{'\n'})
if i < 0 {
buf.Write(b[off:])
break
}
buf.Write(b[off : off+i+1])
buf.WriteString(prefix)
off += i + 1
}
return buf.String()
}