Files
podman/pkg/api/server/handler_rid.go
Chris Evich d968f3fe09 Replace deprecated ioutil
Package `io/ioutil` was deprecated in golang 1.16, preventing podman from
building under Fedora 37.  Fortunately, functionality identical
replacements are provided by the packages `io` and `os`.  Replace all
usage of all `io/ioutil` symbols with appropriate substitutions
according to the golang docs.

Signed-off-by: Chris Evich <cevich@redhat.com>
2022-09-20 15:34:27 -04:00

42 lines
1.0 KiB
Go

package server
import (
"fmt"
"io"
"net/http"
"github.com/containers/podman/v4/pkg/api/types"
"github.com/google/uuid"
"github.com/gorilla/handlers"
"github.com/gorilla/mux"
"github.com/sirupsen/logrus"
)
// referenceIDHandler adds X-Reference-Id Header allowing event correlation
// and Apache style request logging
func referenceIDHandler() mux.MiddlewareFunc {
return func(h http.Handler) http.Handler {
// Only log Apache access_log-like entries at Info level or below
out := io.Discard
if logrus.IsLevelEnabled(logrus.InfoLevel) {
out = logrus.StandardLogger().Out
}
return handlers.CombinedLoggingHandler(out,
http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
rid := r.Header.Get("X-Reference-Id")
if rid == "" {
if c := r.Context().Value(types.ConnKey); c == nil {
rid = uuid.New().String()
} else {
rid = fmt.Sprintf("%p", c)
}
}
r.Header.Set("X-Reference-Id", rid)
w.Header().Set("X-Reference-Id", rid)
h.ServeHTTP(w, r)
}))
}
}