You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.2 KiB
95 lines
2.2 KiB
package api
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/mediocregopher/blog.mediocregopher.com/srv/api/apiutil"
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
|
|
)
|
|
|
|
func addResponseHeaders(headers map[string]string, h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
for k, v := range headers {
|
|
rw.Header().Set(k, v)
|
|
}
|
|
h.ServeHTTP(rw, r)
|
|
})
|
|
}
|
|
|
|
func setLoggerMiddleware(logger *mlog.Logger, h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
|
|
type reqInfoKey string
|
|
|
|
ip, _, _ := net.SplitHostPort(r.RemoteAddr)
|
|
|
|
ctx := r.Context()
|
|
ctx = mctx.Annotate(ctx,
|
|
reqInfoKey("remote_ip"), ip,
|
|
reqInfoKey("url"), r.URL,
|
|
reqInfoKey("method"), r.Method,
|
|
)
|
|
|
|
r = r.WithContext(ctx)
|
|
r = apiutil.SetRequestLogger(r, logger)
|
|
h.ServeHTTP(rw, r)
|
|
})
|
|
}
|
|
|
|
type logResponseWriter struct {
|
|
http.ResponseWriter
|
|
http.Hijacker
|
|
statusCode int
|
|
}
|
|
|
|
func newLogResponseWriter(rw http.ResponseWriter) *logResponseWriter {
|
|
h, _ := rw.(http.Hijacker)
|
|
return &logResponseWriter{
|
|
ResponseWriter: rw,
|
|
Hijacker: h,
|
|
statusCode: 200,
|
|
}
|
|
}
|
|
|
|
func (lrw *logResponseWriter) WriteHeader(statusCode int) {
|
|
lrw.statusCode = statusCode
|
|
lrw.ResponseWriter.WriteHeader(statusCode)
|
|
}
|
|
|
|
func logReqMiddleware(h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
|
|
lrw := newLogResponseWriter(rw)
|
|
|
|
started := time.Now()
|
|
h.ServeHTTP(lrw, r)
|
|
took := time.Since(started)
|
|
|
|
type logCtxKey string
|
|
|
|
ctx := r.Context()
|
|
ctx = mctx.Annotate(ctx,
|
|
logCtxKey("took"), took.String(),
|
|
logCtxKey("response_code"), lrw.statusCode,
|
|
)
|
|
|
|
apiutil.GetRequestLogger(r).Info(ctx, "handled HTTP request")
|
|
})
|
|
}
|
|
|
|
func disallowGetMiddleware(h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
|
|
// we allow websockets to be GETs because, well, they must be
|
|
if r.Method != "GET" || r.Header.Get("Upgrade") == "websocket" {
|
|
h.ServeHTTP(rw, r)
|
|
return
|
|
}
|
|
|
|
apiutil.GetRequestLogger(r).WarnString(r.Context(), "method not allowed")
|
|
rw.WriteHeader(405)
|
|
})
|
|
}
|
|
|