forked from everytv/test2doc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
responsewriter.go
77 lines (62 loc) · 1.3 KB
/
responsewriter.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package test
import (
"log"
"net/http"
"net/http/httptest"
"runtime"
"github.com/everytv/test2doc/doc/parse"
)
type ResponseWriter struct {
HandlerInfo HandlerInfo
URLVars map[string]string
W *httptest.ResponseRecorder
}
type HandlerInfo struct {
FileName string
FuncName string
}
func NewResponseWriter(w *httptest.ResponseRecorder) *ResponseWriter {
return &ResponseWriter{
W: w,
}
}
func (rw *ResponseWriter) Header() http.Header {
return rw.W.Header()
}
func (rw *ResponseWriter) Write(b []byte) (int, error) {
rw.setHandlerInfo()
return rw.W.Write(b)
}
func (rw *ResponseWriter) WriteHeader(c int) {
rw.W.WriteHeader(c)
}
func (rw *ResponseWriter) setHandlerInfo() {
i := 1
max := 15
var pc uintptr
var file, fnName string
var ok, fnInPkg, sawPkg bool
// iterate until we find the top level func in this pkg (the handler)
for i < max {
pc, file, _, ok = runtime.Caller(i)
if !ok {
log.Println("test2doc: setHandlerInfo: !ok")
return
}
fn := runtime.FuncForPC(pc)
fnName = fn.Name()
fnInPkg = parse.IsFuncInPkg(fnName)
if sawPkg && !fnInPkg {
pc, file, _, ok = runtime.Caller(i - 1)
fn := runtime.FuncForPC(pc)
fnName = fn.Name()
break
}
sawPkg = fnInPkg
i++
}
rw.HandlerInfo = HandlerInfo{
FileName: file,
FuncName: fnName,
}
}