From f5956750db0f0d46783e14aa553fe5ccddbfaaac Mon Sep 17 00:00:00 2001 From: tiglog Date: Tue, 17 Oct 2023 17:56:05 +0800 Subject: [PATCH] referactor: change gin to gweb --- gweb/requestid/options.go | 4 ++-- gweb/requestid/reqid.go | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/gweb/requestid/options.go b/gweb/requestid/options.go index 92afe13..b36b7b1 100644 --- a/gweb/requestid/options.go +++ b/gweb/requestid/options.go @@ -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 diff --git a/gweb/requestid/reqid.go b/gweb/requestid/reqid.go index 821aa0e..ab4f336 100644 --- a/gweb/requestid/reqid.go +++ b/gweb/requestid/reqid.go @@ -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) }