Go获取协程ID
简介最近在做Golang的web框架日志,想将一个请求的所有日志上都绑定一个请求ID,由于最初框架的搭建没有解耦好,所以很难在日志中输出请求ID。于是想到了能否用协程ID来记录一个对应的请求ID,然后进行获取。
最近在做Golang的web框架日志,想将一个请求的所有日志上都绑定一个请求ID,由于最初框架的搭建没有解耦好,所以很难在日志中输出请求ID。于是想到了能否用协程ID来记录一个对应的请求ID,然后进行获取。
协程ID作用
类似于其他语言的线程ID,C++当中,每个请求过来,开启一个单独的线程处理它,那么线程的ID是当前请求的唯一标识符。有了唯一标识符以后,就可以创建一个线程安全的全局map来管理每个请求的生命周期的变量。
在go语言中,谷歌开发者不建议大家获取协程ID,主要是为了GC更好的工作,滥用协程ID会导致GC不能及时回收内存。如果一个第三方库使用了协程ID,那么使用该库的人将会莫名中招。
协程ID使用方法
package utils
import (
"bytes"
"fmt"
"runtime"
"strconv"
"sync"
)
//这里存在并发读取,所以采用sync.Map
var globalMap sync.Map
//设置值
func Set(k, v interface{}) {
globalMap.Store(getKey(k), v)
}
//获取值
func Get(k interface{}) interface{} {
if v, ok := globalMap.Load(getKey(k)); ok {
return v
}
return nil
}
//删除值
func Del(k interface{}) {
globalMap.Delete(getKey(k))
}
//获取KEY
func getKey(k interface{}) string {
return fmt.Sprintf("%d_%v", goID(), k)
}
//获取协程ID
func goID() uint64 {
b := make([]byte, 64)
b = b[:runtime.Stack(b, false)]
b = bytes.TrimPrefix(b, []byte("goroutine "))
b = b[:bytes.IndexByte(b, ' ')]
n, _ := strconv.ParseUint(string(b), 10, 64)
return n
}
使用方法:
#在中间件中添加
requestId := uuid.NewV4().String() //生成请求ID
fmt.Println("生成的request_id:", requestId)
utils.Set("request_id", requestId)//设置进去
defer func() {//执行完后删除
utils.Del("request_id")
}()
#任意地方获取
fmt.Println("获取的request_id:",utils.Get("request_id"))
输出的日志:
生成的request_id: ff9e9e81-643f-49e6-8afc-ca6bdec6afaf
获取的request_id: ff9e9e81-643f-49e6-8afc-ca6bdec6afaf