27 lines
730 B
Go
27 lines
730 B
Go
|
package ginutils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/google/uuid"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RequestID is a middleware that will get the request_id from the incoming request if exist and
|
||
|
|
// add it to the response or generate a new ID.
|
||
|
|
func RequestID() gin.HandlerFunc {
|
||
|
|
return func(c *gin.Context) {
|
||
|
|
|
||
|
|
requestID := c.GetHeader(string(HeaderKey_RequestID))
|
||
|
|
if _, err := uuid.Parse(requestID); err != nil {
|
||
|
|
// no request ID or invalid. let's generate a new one
|
||
|
|
requestID = uuid.New().String()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Adding to context
|
||
|
|
c.Set(string(ContextKey_RequestID), requestID)
|
||
|
|
// Add request-id to the header before sending the response
|
||
|
|
c.Header(string(HeaderKey_RequestID), requestID)
|
||
|
|
// Let's the next part run
|
||
|
|
c.Next()
|
||
|
|
}
|
||
|
|
}
|