2021-12-08 18:55:30 -08:00
|
|
|
package middleware
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
|
|
|
|
"github.com/labstack/echo/v4"
|
|
|
|
)
|
|
|
|
|
|
|
|
// LogRequestID includes the request ID in all logs for the given request
|
2021-12-21 18:02:25 -08:00
|
|
|
// This requires that middleware that includes the request ID first execute
|
2021-12-08 18:55:30 -08:00
|
|
|
func LogRequestID() echo.MiddlewareFunc {
|
|
|
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
|
|
return func(c echo.Context) error {
|
2021-12-08 19:21:07 -08:00
|
|
|
rID := c.Response().Header().Get(echo.HeaderXRequestID)
|
|
|
|
format := `{"time":"${time_rfc3339_nano}","id":"%s","level":"${level}","prefix":"${prefix}","file":"${short_file}","line":"${line}"}`
|
|
|
|
c.Logger().SetHeader(fmt.Sprintf(format, rID))
|
2021-12-08 18:55:30 -08:00
|
|
|
return next(c)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|