微服务追踪SQL(支持Isto管控下的gorm查询追踪)

SQL的追踪正确插入到微服务的调用链之间

详细记录了SQL的执行内容和消耗时间

搜索SQL的类型

多线程(goroutine)下的追踪效果

在 Kubernetes 中部署微服务后，通过 Istio 管控微服务之间的流量，Istio 默认会将微服务之间的调用关系生成追踪，放入追踪收集器。但他不支持更细化的SQL操作，比如执行了哪些SQL，每条SQL消耗多长时间。为此，我写了一个简单的支持gorm查询并生成追踪的插件：istio-gorm-tracing

在kubernetes上使用Istio管控微服务后，微服务之间的调用会自动上传到Jaeger的收集器。但只能追踪到服务层，这是我不能接受的，我希望能进一步追踪到服务中的所有MySQL查询，记录每个 sql 的耗时，所以，我简单写了这个插件。

在Istio管控下的容器请求之间，会自动携带x-b3-traceid、x-b3-parentspanid、x-b3-spanid、x-b3-sampled等请求头，这些请求头都是与zipkin对齐的。此插件中会根据传递进来的请求头信息，自动解析出父span，并绑定上下服务之间的调用关系。

每次查询都会记录下执行的SQL语句以及执行耗时等信息，作为后期微服务追踪的依据。

package main

import (
    istiogormtracing "github.com/liamhao/istio-gorm-tracing"
    "log"
    "fmt"
    "github.com/gin-gonic/gin"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

func main() {

    router := gin.Default()

    dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", "dbuser", "dbpswd", "dbhost", 3306, "dbname")
    gormDb, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Println("mysql连接出现了问题：", err.Error())
    } else {
        log.Println("mysql连接成功：", dsn)
    }

    // 这一步很关键，一定要加上，为了启用我们的插件
    gormDb.Use(istiogormtracing.NewDefault(
        // 你的微服务名称
        "istiogormtracing-service",
        // 你的 Jaeger 收集器地址
        "http://127.0.0.1:14268/api/traces",
    ))

    router.GET("/", func(c *gin.Context) {

        // 这一步很关键，一定要加上，为了SQL能与上下游服务做关联
        istiogormtracing.H = c.Request.Header

        list := []map[string]interface{}{}
        gormDb.Table("users").Where("name = 'xiaoming'").Find(&list)

        c.JSON(http.StatusOK, map[string]interface{}{
            "istiogormtracing": "ok",
        })
    })

    router.Run(":7000")
}

然后即可在Jaeger面板中看到我们记录的SQL了。