log.go

package log

import (
	"context"
	"net/http"
	"time"

	"github.com/go-kit/kit/log"
	"github.com/kjk/betterguid"
)

type contextKey string

// Request wraps the given HTTP handler with a log entry that logs how long it has been running.
// It makes a preconfigured logger for the request available as part of the request context.
func Request(next http.Handler, logger log.Logger) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		remote := r.Header.Get("X-Forwarded-For")
		if remote == "" {
			remote = r.RemoteAddr
		}

		logger = log.With(logger,
			"method", r.Method,
			"url", r.URL,
			"remote", remote,
			"id", betterguid.New())

		start := time.Now()
		defer func() {
			d := time.Since(start)

			logger.Log("duration", d, "msg", "request handled")
		}()

		key := contextKey("logger")
		ctx := context.WithValue(r.Context(), key, logger)

		next.ServeHTTP(w, r.WithContext(ctx))
	})
}

// Get returns the logger for the given HTTP request. It panics if r has no logger
func Get(r *http.Request) log.Logger {
	return GetContext(r.Context())
}

// GetContext returns the logger for the given context. If ctx has no logger, it panics
func GetContext(ctx context.Context) log.Logger {
	return ctx.Value(contextKey("logger")).(log.Logger)
}