Go 程序運行時數據統計的可視化工具 Statsviz

作者：Go 夜讀 2021-09-18 08:02:49

開發后端大數據

今天跟大家介紹一款實時可視化 Go 程序運行時數據統計的工具 statsviz。

[[424685]]

本文轉載自微信公眾號「 Go 夜讀」，作者 Go 夜讀。轉載本文請聯系 Go 夜讀公眾號。

今天跟大家介紹一款實時可視化 Go 程序運行時數據統計的工具 statsviz

https://github.com/arl/statsviz

它的圖形化展現對于我們了解 Go 程序的 GC 行為，以及內存開銷等很有用!

使用也很簡單：

1. go get github.com/arl/statsviz

2. 在你的 http.ServeMux 上注冊

mux := http.NewServeMux() 
statsviz.Register(mux)

或者使用默認 http 注冊：

statsviz.RegisterDefault()

如果你的程序不是一個 http 應用程序，那么你可以添加以下代碼來啟動

go func() { 
    log.Println(http.ListenAndServe("localhost:6060", nil)) 
}()

這段代碼，我相信大家都非常熟悉了吧~

當我們將啟動之后，我們可以直接在瀏覽器中打開：

http://localhost:6060/debug/statsviz/

看起來酷炫高大上，其實主要還是依賴于 Go 為我們提供的 runtime stats

具體我們來看看它的一些代碼：

1. websocket handler

// NewWsHandler returns a handler that upgrades the HTTP server connection to the WebSocket 
// protocol and sends application statistics at the given frequency. 
// 
// If the upgrade fails, an HTTP error response is sent to the client. 
func NewWsHandler(frequency time.Duration) http.HandlerFunc { 
 return func(w http.ResponseWriter, r *http.Request) { 
  var upgrader = websocket.Upgrader{ 
   ReadBufferSize:  1024, 
   WriteBufferSize: 1024, 
  } 
 
  ws, err := upgrader.Upgrade(w, r, nil) 
  if err != nil { 
   return 
  } 
  defer ws.Close() 
 
  // Explicitly ignore this error. We don't want to spam standard output 
  // each time the other end of the websocket connection closes. 
  _ = sendStats(ws, frequency) 
 } 
}

2. sendStats

// sendStats indefinitely send runtime statistics on the websocket connection. 
func sendStats(conn *websocket.Conn, frequency time.Duration) error { 
 tick := time.NewTicker(frequency) 
 defer tick.Stop() 
 
 var ( 
  stats stats 
  err   error 
 ) 
 for range tick.C { 
  runtime.ReadMemStats(&stats.Mem) 
  stats.NumGoroutine = runtime.NumGoroutine() 
  if err = conn.WriteJSON(stats); err != nil { 
   break 
  } 
 } 
 
 return err 
}

3. 其實這個項目比較核心的代碼是前端 JavaScript 代碼：

m.pushData = function (ts, allStats) { 
        data.times.push(ts); // timestamp 
 
        const memStats = allStats.Mem; 
 
        data.gcfraction.push(memStats.GCCPUFraction); 
        data.goroutines.push(allStats.NumGoroutine); 
 
        data.heap[idxHeapAlloc].push(memStats.HeapAlloc); 
        data.heap[idxHeapSys].push(memStats.HeapSys); 
        data.heap[idxHeapIdle].push(memStats.HeapIdle); 
        data.heap[idxHeapInuse].push(memStats.HeapInuse); 
        data.heap[idxHeapNextGC].push(memStats.NextGC); 
 
        data.mspanMCache[idxMSpanMCacheMSpanInUse].push(memStats.MSpanInuse); 
        data.mspanMCache[idxMSpanMCacheMSpanSys].push(memStats.MSpanSys); 
        data.mspanMCache[idxMSpanMSpanMSCacheInUse].push(memStats.MCacheInuse); 
        data.mspanMCache[idxMSpanMSpanMSCacheSys].push(memStats.MCacheSys); 
 
        data.objects[idxObjectsLive].push(memStats.Mallocs - memStats.Frees); 
        data.objects[idxObjectsLookups].push(memStats.Lookups); 
        data.objects[idxObjectsHeap].push(memStats.HeapObjects); 
 
        for (let i = 0; i < memStats.BySize.length; i++) { 
            const size = memStats.BySize[i]; 
            data.bySize[i].push(size.Mallocs - size.Frees); 
        } 
 
        updateLastGC(memStats); 
    }