MaysWind
29 lines
689 B
Go
29 lines
689 B
Go
package middlewares
|
|
|
|
import (
|
|
"github.com/mayswind/ezbookkeeping/pkg/core"
|
|
"github.com/mayswind/ezbookkeeping/pkg/requestid"
|
|
"github.com/mayswind/ezbookkeeping/pkg/settings"
|
|
)
|
|
|
|
const requestIdHeader = "X-Request-ID"
|
|
|
|
// RequestId generates a new request id and add it to context and response header
|
|
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.ClientPort())
|
|
c.SetRequestId(requestId)
|
|
|
|
if config.EnableRequestIdHeader {
|
|
c.Header(requestIdHeader, requestId)
|
|
}
|
|
|
|
c.Next()
|
|
}
|
|
}
|