微服務追蹤SQL(支援Isto管控下的gorm查詢追蹤)

LiamHao 發表於 2022-07-01
微服務 Go SQL

效果圖

SQL的追蹤正確插入到微服務的呼叫鏈之間

圖片

詳細記錄了SQL的執行內容和消耗時間

圖片

搜尋SQL的型別

圖片

多執行緒(goroutine)下的追蹤效果

圖片

在 Kubernetes 中部署微服務後,通過 Istio 管控微服務之間的流量,Istio 預設會將微服務之間的呼叫關係生成追蹤,放入追蹤收集器。但他不支援更細化的SQL操作,比如執行了哪些SQL,每條SQL消耗多長時間。為此,我寫了一個簡單的支援gorm查詢並生成追蹤的外掛:istio-gorm-tracing

微服務追蹤 gorm 外掛

kubernetes上使用Istio管控微服務後,微服務之間的呼叫會自動上傳到Jaeger的收集器。但只能追蹤到服務層,這是我不能接受的,我希望能進一步追蹤到服務中的所有MySQL查詢,記錄每個 sql 的耗時,所以,我簡單寫了這個外掛。

特性

支援Istio

Istio管控下的容器請求之間,會自動攜帶x-b3-traceidx-b3-parentspanidx-b3-spanidx-b3-sampled等請求頭,這些請求頭都是與zipkin對齊的。此外掛中會根據傳遞進來的請求頭資訊,自動解析出父span,並繫結上下服務之間的呼叫關係。

記錄SQL資訊

每次查詢都會記錄下執行的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:%[email protected](%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了。

相關文章