This repository was archived by the owner on Oct 2, 2020. It is now read-only.
File tree 2 files changed +6
-16
lines changed
2 files changed +6
-16
lines changed Original file line number Diff line number Diff line change @@ -49,15 +49,12 @@ type reqInfo struct {
49
49
}
50
50
51
51
func (rl * reqLogger ) ServeHTTP (w http.ResponseWriter , r * http.Request ) {
52
- if watcher := rl .watcher ; watcher != nil {
53
- info := reqInfo {
52
+ if watcher := rl .watcher ; watcher . Active () {
53
+ watcher . HandleItem ( reqInfo {
54
54
Method : r .Method ,
55
55
Path : r .URL .Path ,
56
56
Query : r .URL .RawQuery ,
57
- }
58
- if ! watcher .HandleItem (info ) {
59
- rl .watcher = nil
60
- }
57
+ })
61
58
}
62
59
rl .handler .ServeHTTP (w , r )
63
60
}
Original file line number Diff line number Diff line change @@ -44,23 +44,16 @@ var resLogTextTemplate = template.Must(template.New("res_logger_text").Parse(`
44
44
` ))
45
45
46
46
func (rl * resLogger ) ServeHTTP (w http.ResponseWriter , r * http.Request ) {
47
- watcher := rl .watcher
48
- if watcher == nil {
49
- rl .handler .ServeHTTP (w , r )
50
- return
51
- }
47
+ rl .watcher .Active ()
52
48
53
49
rec := httptest .NewRecorder ()
54
50
rl .handler .ServeHTTP (rec , r )
55
51
56
- info := resInfo {
52
+ rl . watcher . HandleItem ( resInfo {
57
53
Code : rec .Code ,
58
54
Bytes : rec .Body .Len (),
59
55
ContentType : rec .HeaderMap .Get ("Content-Type" ),
60
- }
61
- if ! watcher .HandleItem (info ) {
62
- rl .watcher = nil
63
- }
56
+ })
64
57
65
58
hdr := w .Header ()
66
59
for key , vals := range rec .HeaderMap {
You can’t perform that action at this time.
0 commit comments