Go获取协程ID

xiaohai 2021-05-05 21:46:52 3088人围观 标签: Go 
简介最近在做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