美文网首页golang
修改golang源代码获取goroutine id实现Threa

修改golang源代码获取goroutine id实现Threa

作者: 左手java右手go | 来源:发表于2018-12-27 15:05 被阅读0次

    开篇

    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清除数据

    欢迎您扫一扫上面的二维码关注个人微信公众号

    相关文章

      网友评论

        本文标题:修改golang源代码获取goroutine id实现Threa

        本文链接:https://www.haomeiwen.com/subject/ezfklqtx.html