详解如何在Go服务中做链路追踪
使用 Go 语言开发微服务的时候，需要追踪每一个请求的访问链路，这块在 Go 中目前没有很好的解决方案。
在 Java 中解决这个问题比较简单，可以使用 MDC，在一个进程内共享一个请求的 RequestId。
在 Go 中实现链路追踪有两种思路：一种是在项目中使用一个全局的 map， key 是 goroutine 的唯一 Id，value 是 RequestId，另一种思路可以使用 context.Context 来实现。
下面的代码基于 gin 框架来实现。
1. 使用全局 map 来实现
使用 map 方案需要在全局维护一个 map，在一个请求进来的时候，会为每一个请求生成 RequestId，然后在每次在打印日志的时候，从这个 Map 中通过 goid 获取到 RequestId，打印到日志中。
代码的实现很简单：
var requestIdMap = make(map[int64]string) // 全局的 Map
func main() {
r := gin.Default()
r.Use(Logger()) // 使用中间件
r.GET("/index", func(c *gin.Context) {
Info("main goroutine") // 打印日志
c.JSON(200, gin.H{
"message": "index",
})
})
r.Run()
}
func Logger() gin.HandlerFunc {
return func(c *gin.Context) {
requestIdMap[goid.Get()] = uuid.New().String() // 在日志中间件中为每个请求设定
c.Next()
}
}
func Info(msg string) {
now := time.Now()
nowStr := now.Format("2006-01-02 15:04:05")
fmt.Printf("%s [%s] %s\n", nowStr, requestIdMap[goid.Get()], msg) // 打印日志
}
这样的实现很简单，但是问题也很多。
第一个问题就是，在 Go 程序中，一次请求可能会涉及到多个 goroutine，用这种方式很难在多个 gotoutine 之间传递 RequestId。
在下面的代码中，如果新启动了一个 goroutine，就会导致日志中获取不到 RequestId：
func main() {
r := gin.Default()
r.Use(Logger())
r.GET("/index", func(c *gin.Context) {
Info("main goroutine")
go func() { // 这里新启动了一个一个 goroutine
Info("goroutine1")
}()
c.JSON(200, gin.H{
"message": "index",
})
})
r.Run()
}
获取 goroutine id 也不是一种常规的做法，一般要通过 hack 的方式来获取，这种做法已经不推荐了。而且这个全局的 map 为了并发安全，在实际的使用中，可以还需要用到锁，在高并发的情况下必然会影响性能。
在每个请求结束的时候，还需要手动的把 requestId 从 map 中删除，否则就会造成内存泄漏。
总的来说，使用 map 这种方式来实现并不是很好。
2. 使用 Context 来实现
在上面的代码中，我们使用一个 hack 的方式去获取 goroutine id，这种方式早就不推荐使用，更推荐使用 Context，关于 Context 内容，可以去看我之前的文章，在这里就不多说了。
在传递 RequestId 的场景中，同样也可以使用 Context 来实现，使用 Context 好处很明显，Context 生命周期与请求相同，不需要手动销毁。而且Context 是每个请求独享的，也不用担心并发安全的问题，Context 还可以在 goroutine 之间传递。
使用 Context 实现的代码如下：
func main() {
r := gin.Default()
r.Use(Logger())
r.GET("/index", func(c *gin.Context) {
ctx, _ := c.Get("ctx")
Info(ctx.(context.Context) , "main goroutine")
go func() {
Info(ctx.(context.Context), "goroutine1")
}()
c.JSON(200, gin.H{
"message": "index",
})
})
r.Run()
}
func Logger() gin.HandlerFunc {
return func(c *gin.Context) {
valueCtx := context.WithValue(c.Request.Context(), "RequestId", uuid.New().String())
c.Set("ctx", valueCtx)
c.Next()
}
}
func Info(ctx context.Context, msg string) {
now := time.Now()
nowStr := now.Format("2006-01-02 15:04:05")
fmt.Printf("%s [%s] %s\n", nowStr, ctx.Value("RequestId"), msg)
}
这样在一个请求中，所有的 gotroutine 都可以获取到同一个 RequestId，而且不用担心内存泄漏和并发安全。
但是使用 Context 也有个问题就是需要每次传递 Context，很多人还不习惯使用这种方式。其实 Go 官方早就推荐使用 Context了，通常会把 Context 作为函数的第一个参数。如果函数使用结构体作为参数，也可以直接把 Context 作为结构体的一个字段。
Context 除了使用可以同来传递 RequestId 之外，还可以用来控制 goroutine 的生命周期，这些内容在之前的 Context 文章中详细说明了，感兴趣的可以去看看。
3. 小结
获取 goroutine id 这种方式应该被抛弃，而是应该使用 Context, Go 官方也早就推荐使用这种方式，在上文中，我们使用 Context 来传递 RequestId，除此之外还可以用来传递单个请求范围的值，比如认证的 token 之类的，应该习惯在代码中使用 Context。
[1] https://blog.golang.org/context
到此这篇关于详解如何在Go 服务中做链路追踪的文章就介绍到这了,更多相关Go 服务中做链路追踪内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
Viper 是一个功能齐全的 Go 应用程序配置库，支持很多场景。在本文中，我们将深入探讨 Viper 的各种用法和使用场景，以帮助读者更好地了解和使用 Viper 来管理应用程序配置，感兴趣的同学可以参考阅读2023-05-05
这篇文章主要介绍了使用go实现删除sql里面的注释和字符串功能,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-11-11
这篇文章主要为大家详细介绍了如何利用Go语言实现聊天小工具（UDP版），文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-03-03
在当今的软件开发领域,多种编程语言常常需要协同工作,以充分利用各自的优势来构建复杂的应用系统,Go 语言作为一门高效、简洁的编程语言,也经常需要与其他语言进行交互,接下来,我们将详细探讨 Go 语言如何与其他语言进行交互,需要的朋友可以参考下2024-06-06
panic()和recover()是Go语言中用于处理错误的两个重要函数,本文主要介绍了Go语言panic和recover的用法实例,panic()用于中止程序并引发panic,而recover()用于捕获panic并恢复程序的执行,感兴趣的可以了解一下2024-01-01
这篇文章主要为大家详细介绍了Go语言多人聊天室项目实战，实现单撩或多撩等多种功能，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-08-08
这篇文章主要为大家详细介绍一下Golang切片拷贝和数组拷贝，文中有详细的代码示例供大家参考，需要的可以参考一下2023-04-04
这篇文章主要为大家详细介绍了Go语言中接口应用模式或惯例介绍的相关知识,文中的示例代码讲解详细,有需要的小伙伴可以跟随小编一起学习一下2023-11-11
Go语言动态并发控制sync.WaitGroup的灵活运用示例详解
本文将讲解 sync.WaitGroup 的使用方法、原理以及在实际项目中的应用场景,用清晰的代码示例和详细的注释,助力读者掌握并发编程中等待组的使用技巧2023-11-11
这篇文章主要为大家详细介绍了Go语言中内置RPC的原理。说起 RPC 大家想到的一般是框架，Go 作为编程语言竟然还内置了 RPC，着实让我有些吃鲸，本文就来一起聊聊吧2023-03-03
最新评论