golang：获取协程ID，从此走上一条曲折的不归路

最新推荐文章于 2024-03-16 14:49:19 发布

OceanStar的学习笔记

最新推荐文章于 2024-03-16 14:49:19 发布

阅读量3.8k

点赞数

CC 4.0 BY-SA版权

分类专栏： golang

原文链接：https://blog.csdn.net/wyansai/article/details/101004226

golang 专栏收录该内容

206 篇文章

订阅专栏

本文探讨了Go语言中协程ID的使用及潜在问题，介绍了如何利用全局Map结合协程ID来管理请求生命周期的变量，同时强调了谷歌开发者关于避免滥用协程ID的建议。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

类似于其他语言的线程ID，C++当中，每个请求过来，开启一个单独的线程处理它，那么线程的ID是当前请求的唯一标识符。有了唯一标识符以后，就可以创建一个线程安全的全局map来管理每个请求的生命周期的变量。

在go语言中，谷歌开发者不建议大家获取协程ID，主要是为了GC更好的工作，滥用协程ID会导致GC不能及时回收内存。如果一个第三方库使用了协程ID，那么使用该库的人将会莫名中招。

package gomap

import (
	"bytes"
	"fmt"
	"runtime"
	"strconv"
	"sync"
)

var goGlobalMap sync.Map

func Set(k, v interface{}) {
	goGlobalMap.Store(fmt.Sprintf("%d_%v", goID(), k), v)
}

func Get(k interface{}) interface{} {
	if v, ok := goGlobalMap.Load(fmt.Sprintf("%d_%v", goID(), k)); ok {
		return v
	}
	return nil
}

func Del(k interface{}) {
	goGlobalMap.Delete(fmt.Sprintf("%d_%v", goID(), k))
}

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
}