git.iko.soy/typeable/oauth2-proxy
on 37c415b help sign in
typeable/oauth2-proxy/logging_handler.go
104 lines 2.7 kB history raw
1// largely adapted from https://github.com/gorilla/handlers/blob/master/handlers.go
2// to add logging of request duration as last value (and drop referrer)
4package main
6import (
7 "net/http"
8 "time"
10 "github.com/pusher/oauth2_proxy/logger"
13// responseLogger is wrapper of http.ResponseWriter that keeps track of its HTTP status
14// code and body size
15type responseLogger struct {
16 w http.ResponseWriter
17 status int
18 size int
19 upstream string
20 authInfo string
23// Header returns the ResponseWriter's Header
24func (l *responseLogger) Header() http.Header {
25 return l.w.Header()
28// Support Websocket
29func (l *responseLogger) Hijack() (rwc net.Conn, buf *bufio.ReadWriter, err error) {
30 if hj, ok := l.w.(http.Hijacker); ok {
31 return hj.Hijack()
32 }
33 return nil, nil, errors.New("http.Hijacker is not available on writer")
36// ExtractGAPMetadata extracts and removes GAP headers from the ResponseWriter's
37// Header
38func (l *responseLogger) ExtractGAPMetadata() {
39 upstream := l.w.Header().Get("GAP-Upstream-Address")
40 if upstream != "" {
41 l.upstream = upstream
42 l.w.Header().Del("GAP-Upstream-Address")
43 }
44 authInfo := l.w.Header().Get("GAP-Auth")
45 if authInfo != "" {
46 l.authInfo = authInfo
47 l.w.Header().Del("GAP-Auth")
48 }
51// Write writes the response using the ResponseWriter
52func (l *responseLogger) Write(b []byte) (int, error) {
53 if l.status == 0 {
54 // The status will be StatusOK if WriteHeader has not been called yet
55 l.status = http.StatusOK
56 }
57 l.ExtractGAPMetadata()
58 size, err := l.w.Write(b)
59 l.size += size
60 return size, err
63// WriteHeader writes the status code for the Response
64func (l *responseLogger) WriteHeader(s int) {
65 l.ExtractGAPMetadata()
66 l.w.WriteHeader(s)
67 l.status = s
70// Status returns the response status code
71func (l *responseLogger) Status() int {
72 return l.status
75// Size returns teh response size
76func (l *responseLogger) Size() int {
77 return l.size
80func (l *responseLogger) Flush() {
81 if flusher, ok := l.w.(http.Flusher); ok {
82 flusher.Flush()
83 }
86// loggingHandler is the http.Handler implementation for LoggingHandlerTo and its friends
87type loggingHandler struct {
88 handler http.Handler
91// LoggingHandler provides an http.Handler which logs requests to the HTTP server
92func LoggingHandler(h http.Handler) http.Handler {
93 return loggingHandler{
94 handler: h,
95 }
98func (h loggingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
99 t := time.Now()
100 url := *req.URL
101 responseLogger := &responseLogger{w: w}
102 h.handler.ServeHTTP(responseLogger, req)
103 logger.PrintReq(responseLogger.authInfo, responseLogger.upstream, req, url, t, responseLogger.Status(), responseLogger.Size())