tiglog/golib
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

referactor: change gin to gweb

Browse Source
This commit is contained in:
tiglog 2023-10-17 17:56:05 +08:00
parent 64679cfdca
commit f5956750db
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gweb/requestid/options.go
View File

@ -7,14 +7,14 @@
package requestid
import "github.com/gin-gonic/gin"
import "git.hexq.cn/tiglog/golib/gweb"
// Option for queue system
type Option func(*config)
type (
Generator func() string
Handler func(c *gin.Context, requestID string)
Handler func(c *gweb.Context, requestID string)
)
type HeaderStrKey string

8
gweb/requestid/reqid.go
View File

@ -8,7 +8,7 @@
package requestid
import (
"github.com/gin-gonic/gin"
"git.hexq.cn/tiglog/golib/gweb"
"github.com/rs/xid"
)
@ -26,7 +26,7 @@ type config struct {
}
// New initializes the RequestID middleware.
func New(opts ...Option) gin.HandlerFunc {
func New(opts ...Option) gweb.HandlerFunc {
cfg := &config{
generator: func() string {
return xid.New().String()
@ -40,7 +40,7 @@ func New(opts ...Option) gin.HandlerFunc {
headerXRequestID = string(cfg.headerKey)
return func(c *gin.Context) {
return func(c *gweb.Context) {
// Get id from request
rid := c.GetHeader(headerXRequestID)
if rid == "" {
@ -57,6 +57,6 @@ func New(opts ...Option) gin.HandlerFunc {
}
// Get returns the request identifier
func Get(c *gin.Context) string {
func Get(c *gweb.Context) string {
return c.GetHeader(headerXRequestID)
}