Go获取协程ID—HI技术分享

最近在做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