pki/vendor/github.com/labstack/echo/v4/middleware/request_id.go

76 lines
1.9 KiB
Go
Raw Normal View History

2024-04-19 16:50:48 +02:00
// SPDX-License-Identifier: MIT
// SPDX-FileCopyrightText: © 2015 LabStack LLC and Echo contributors
2020-11-25 20:36:07 +01:00
package middleware
import (
"github.com/labstack/echo/v4"
)
2024-04-19 16:50:48 +02:00
// RequestIDConfig defines the config for RequestID middleware.
type RequestIDConfig struct {
// Skipper defines a function to skip middleware.
Skipper Skipper
2020-11-25 20:36:07 +01:00
2024-04-19 16:50:48 +02:00
// Generator defines a function to generate an ID.
// Optional. Defaults to generator for random string of length 32.
Generator func() string
2021-08-30 20:04:15 +02:00
2024-04-19 16:50:48 +02:00
// RequestIDHandler defines a function which is executed for a request id.
RequestIDHandler func(echo.Context, string)
2022-11-02 17:49:03 +01:00
2024-04-19 16:50:48 +02:00
// TargetHeader defines what header to look for to populate the id
TargetHeader string
}
2020-11-25 20:36:07 +01:00
2024-04-19 16:50:48 +02:00
// DefaultRequestIDConfig is the default RequestID middleware config.
var DefaultRequestIDConfig = RequestIDConfig{
Skipper: DefaultSkipper,
Generator: generator,
TargetHeader: echo.HeaderXRequestID,
}
2020-11-25 20:36:07 +01:00
// RequestID returns a X-Request-ID middleware.
func RequestID() echo.MiddlewareFunc {
return RequestIDWithConfig(DefaultRequestIDConfig)
}
// RequestIDWithConfig returns a X-Request-ID middleware with config.
func RequestIDWithConfig(config RequestIDConfig) echo.MiddlewareFunc {
// Defaults
if config.Skipper == nil {
config.Skipper = DefaultRequestIDConfig.Skipper
}
if config.Generator == nil {
config.Generator = generator
}
2022-11-02 17:49:03 +01:00
if config.TargetHeader == "" {
config.TargetHeader = echo.HeaderXRequestID
}
2020-11-25 20:36:07 +01:00
return func(next echo.HandlerFunc) echo.HandlerFunc {
return func(c echo.Context) error {
if config.Skipper(c) {
return next(c)
}
req := c.Request()
res := c.Response()
2022-11-02 17:49:03 +01:00
rid := req.Header.Get(config.TargetHeader)
2020-11-25 20:36:07 +01:00
if rid == "" {
rid = config.Generator()
}
2022-11-02 17:49:03 +01:00
res.Header().Set(config.TargetHeader, rid)
2021-08-30 20:04:15 +02:00
if config.RequestIDHandler != nil {
config.RequestIDHandler(c, rid)
}
2020-11-25 20:36:07 +01:00
return next(c)
}
}
}
func generator() string {
2024-04-19 16:50:48 +02:00
return randomString(32)
2020-11-25 20:36:07 +01:00
}