28 lines
605 B
Go
28 lines
605 B
Go
package requestid
|
||
|
||
import (
|
||
"github.com/jtolds/gls"
|
||
)
|
||
|
||
// requestIDKey 是 gls 中存储 request_id 的 key
|
||
var requestIDKey = gls.GenSym()
|
||
|
||
// glsMgr 是 gls 管理器
|
||
var glsMgr = gls.NewContextManager()
|
||
|
||
// SetRequestID 在 gls 中设置 request_id，并在 fn 执行期间保持有效
|
||
func SetRequestID(requestID string, fn func()) {
|
||
glsMgr.SetValues(gls.Values{requestIDKey: requestID}, fn)
|
||
}
|
||
|
||
// GetRequestID 从 gls 中获取当前 goroutine 的 request_id
|
||
func GetRequestID() string {
|
||
val, ok := glsMgr.GetValue(requestIDKey)
|
||
if !ok {
|
||
return ""
|
||
}
|
||
reqID, _ := val.(string)
|
||
return reqID
|
||
}
|
||
|