mirror of
https://github.com/mayswind/ezbookkeeping.git
synced 2026-05-15 23:47:33 +08:00
MaysWind
28 lines
561 B
Go
28 lines
561 B
Go
package middlewares
|
|
|
|
import (
|
|
"github.com/mayswind/lab/pkg/core"
|
|
"github.com/mayswind/lab/pkg/requestid"
|
|
"github.com/mayswind/lab/pkg/settings"
|
|
)
|
|
|
|
const requestIdHeader = "X-Request-ID"
|
|
|
|
func RequestId(config *settings.Config) core.MiddlewareHandlerFunc {
|
|
return func(c *core.Context) {
|
|
if requestid.Container.Current == nil {
|
|
c.Next()
|
|
return
|
|
}
|
|
|
|
requestId := requestid.Container.Current.GenerateRequestId(c.ClientIP())
|
|
c.SetRequestId(requestId)
|
|
|
|
if config.EnableRequestIdHeader {
|
|
c.Header(requestIdHeader, requestId)
|
|
}
|
|
|
|
c.Next()
|
|
}
|
|
}
|