开篇
golang在http.Request中提供了一个Context用于存储kv对，我们可以通过这个来存储请求相关的数据。在请求入口，我们把唯一的requstID存储到context中，在后续需要调用的地方把值取出来打印。如果日志是在controller中打印，这个很好处理，http.Request是作为入参的。但如果是在更底层呢？比如说是在model甚至是一些工具类中。我们当然可以给每个方法都提供一个参数，由调用方把context一层一层传下来，但这种方式明显不够优雅。想想java里面是怎么做的--ThreadLocal。虽然golang官方不太认可这种方式，但是我们今天就是要基于goroutine id实现它。
We wouldn't even be having this discussion if thread local storage wasn't useful. But every feature comes at a cost, and in my opinion the cost of threadlocals far outweighs their benefits. They're just not a good fit for Go.
思路
每个goroutine有一个唯一的id，但是被隐藏了，我们首先把它暴露出来，然后建立一个map，用id作为key，goroutineLocal存储的实际数据作为value。
获取goroutine id
1.修改 $GOROOT/src/runtime/proc.go 文件，添加 GetGoroutineId() 函数
func GetGoroutineId() int64 {
return getg().goid
}其中getg()函数是获取当前执行的g对象，g对象包含了栈，cgo信息，GC信息，goid等相关数据，goid就是我们想要的。
2.重新编译源码
cd ~/go/src
GOROOT_BOOTSTRAP='/Users/qiuxudong/go1.9' ./all.bash实现 GoroutineLocal
package goroutine_local
import (
"sync"
"runtime"
)
type goroutineLocal struct {
initfun func() interface{}
m *sync.Map
}
func NewGoroutineLocal(initfun func() interface{}) *goroutineLocal {
return &goroutineLocal{initfun:initfun, m:&sync.Map{}}
}
func (gl *goroutineLocal)Get() interface{} {
value, ok := gl.m.Load(runtime.GetGoroutineId())
if !ok && gl.initfun != nil {
value = gl.initfun()
}
return value
}
func (gl *goroutineLocal)Set(v interface{}) {
gl.m.Store(runtime.GetGoroutineId(), v)
}
func (gl *goroutineLocal)Remove() {
gl.m.Delete(runtime.GetGoroutineId())
}简单测试一下
package goroutine_local
import (
"testing"
"fmt"
"time"
"runtime"
)
var gl = NewGoroutineLocal(func() interface{} {
return "default"
})
func TestGoroutineLocal(t *testing.T) {
gl.Set("test0")
fmt.Println(runtime.GetGoroutineId(), gl.Get())
go func() {
gl.Set("test1")
fmt.Println(runtime.GetGoroutineId(), gl.Get())
gl.Remove()
fmt.Println(runtime.GetGoroutineId(), gl.Get())
}()
time.Sleep(2 * time.Second)
}可以看到结果
5 test0
6 test1
6 default内存泄露问题
由于跟goroutine绑定的数据放在goroutineLocal的map里面，即使goroutine销毁了数据还在，可能存在内存泄露，因此不使用时要记得调用Remove清除数据
有疑问加站长微信联系（非本文作者）
