Go 框架性能监控的自动化工具

import (
    "net/http/pprof"
    "runtime"
)

func init() {
    // 注册 /debug/pprof 端点以启用性能概要。
    http.HandleFunc("/debug/pprof/", pprof.Index)
    http.HandleFunc("/debug/pprof/cpu", pprof.Index)
}

func main() {
    // 运行 Go 程序。
    // 使用命令 "go tool pprof http://localhost:8080/debug/pprof/cpu" 生成 CPU 概要。
    runtime.MemProfileRate = 2
    http.ListenAndServe(":8080", nil)
}

package main

import (
    "fmt"
    "net/http"

    "expvar"
)

var memoryAllocated = expvar.Int{Key: "bytes_allocated"}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

func handler(w http.ResponseWriter, r *http.Request) {
    // 分配 1GB 内存。
    memory := make([]byte, 1<<30)
    atomic.StoreUint64(&memoryAllocated, uint64(len(memory)))
    fmt.Fprintf(w, "Memory allocated: %d bytes", memoryAllocated.Value())
}

import (
    "net/http"

    "github.com/prometheus/client_golang/prometheus"
    "github.com/prometheus/client_golang/prometheus/promhttp"
)

var (
    requestDuration = prometheus.NewHistogramVec(
        prometheus.HistogramOpts{
            Name:    "request_duration_seconds",
            Help:    "Duration of HTTP requests.",
            Buckets: prometheus.DefBuckets,
        },
        []string{"method", "path"},
    )
)

func init() {
    // 注册 Prometheus 遥测收集器。
    prometheus.MustRegister(requestDuration)
    http.Handle("/metrics", promhttp.Handler())
}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

func handler(w http.ResponseWriter, r *http.Request) {
    // 记录请求延迟。
    timer := prometheus.NewTimer(requestDuration.WithLabelValues(r.Method, r.URL.Path))
    defer timer.ObserveDuration()
    // 处理 HTTP 请求。
}