Go 웹 서비스 강화: 맞춤형 프로파일러 구축-Golang-php.cn

Supercharge Your Go Web Service: Building a Custom Profiler

소개

Go 개발자로서 우리는 애플리케이션을 최적화할 때 내장된 프로파일링 도구를 자주 사용합니다. 하지만 애플리케이션의 언어를 말하는 프로파일러를 만들 수 있다면 어떨까요? 이 가이드에서는 요청 처리, 데이터베이스 작업 및 메모리 사용량에 중점을 두고 Go 웹 서비스용 사용자 정의 프로파일러를 구성합니다.

맞춤형 프로파일링 사례

Go의 표준 프로파일러는 강력하지만 웹 서비스와 관련된 모든 것을 캡처하지 못할 수도 있습니다.

다양한 엔드포인트에서의 웹 요청 처리 패턴
다양한 작업에 대한 데이터베이스 쿼리 성능
최대 로드 시 메모리 사용량 변동

이러한 요구 사항을 정확히 해결하는 프로파일러를 구축해 보겠습니다.

샘플 웹 서비스

먼저 프로파일링할 기본 웹 서비스를 설정해 보겠습니다.

package main

import (
    "database/sql"
    "encoding/json"
    "log"
    "net/http"

    _ "github.com/lib/pq"
)

type User struct {
    ID   int    `json:"id"`
    Name string `json:"name"`
}

var db *sql.DB

func main() {
    // Initialize database connection
    var err error
    db, err = sql.Open("postgres", "postgres://username:password@localhost/database?sslmode=disable")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // Set up routes
    http.HandleFunc("/user", handleUser)

    // Start the server
    log.Println("Server starting on :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleUser(w http.ResponseWriter, r *http.Request) {
    // Handle GET and POST requests for users
    // Implementation omitted for brevity
}

로그인 후 복사

이제 이 서비스에 대한 깊은 통찰력을 얻기 위해 맞춤형 프로파일러를 구축해 보겠습니다.

커스텀 프로파일러 구현

1. 요청 기간 추적

각 요청에 걸리는 시간을 측정하는 것부터 시작하겠습니다.

import (
    "time"
    "sync"
)

var (
    requestDurations = make(map[string]time.Duration)
    requestMutex     sync.RWMutex
)

func trackRequestDuration(handler http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        handler(w, r)
        duration := time.Since(start)

        requestMutex.Lock()
        requestDurations[r.URL.Path] += duration
        requestMutex.Unlock()
    }
}

// In main(), wrap your handlers:
http.HandleFunc("/user", trackRequestDuration(handleUser))

로그인 후 복사

2. 데이터베이스 쿼리 프로파일링

다음으로 데이터베이스 성능을 살펴보겠습니다.

type QueryStats struct {
    Count    int
    Duration time.Duration
}

var (
    queryStats = make(map[string]QueryStats)
    queryMutex sync.RWMutex
)

func trackQuery(query string, duration time.Duration) {
    queryMutex.Lock()
    defer queryMutex.Unlock()

    stats := queryStats[query]
    stats.Count++
    stats.Duration += duration
    queryStats[query] = stats
}

// Use this function to wrap your database queries:
func profiledQuery(query string, args ...interface{}) (*sql.Rows, error) {
    start := time.Now()
    rows, err := db.Query(query, args...)
    duration := time.Since(start)
    trackQuery(query, duration)
    return rows, err
}

로그인 후 복사

3. 메모리 사용량 추적

프로파일러를 완성하기 위해 메모리 사용량 추적을 추가해 보겠습니다.

import "runtime"

func getMemStats() runtime.MemStats {
    var m runtime.MemStats
    runtime.ReadMemStats(&m)
    return m
}

func logMemStats() {
    stats := getMemStats()
    log.Printf("Alloc = %v MiB", bToMb(stats.Alloc))
    log.Printf("TotalAlloc = %v MiB", bToMb(stats.TotalAlloc))
    log.Printf("Sys = %v MiB", bToMb(stats.Sys))
    log.Printf("NumGC = %v", stats.NumGC)
}

func bToMb(b uint64) uint64 {
    return b / 1024 / 1024
}

// Call this periodically in a goroutine:
go func() {
    ticker := time.NewTicker(1 * time.Minute)
    for range ticker.C {
        logMemStats()
    }
}()

로그인 후 복사

4. 프로파일러 API 엔드포인트

마지막으로 프로파일링 데이터를 노출할 엔드포인트를 만들어 보겠습니다.

func handleProfile(w http.ResponseWriter, r *http.Request) {
    requestMutex.RLock()
    queryMutex.RLock()
    defer requestMutex.RUnlock()
    defer queryMutex.RUnlock()

    profile := map[string]interface{}{
        "requestDurations": requestDurations,
        "queryStats":       queryStats,
        "memStats":         getMemStats(),
    }

    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(profile)
}

// In main():
http.HandleFunc("/debug/profile", handleProfile)

로그인 후 복사

모든 것을 하나로 합치기

이제 프로파일러 구성요소가 있으므로 이를 기본 애플리케이션에 통합해 보겠습니다.

func main() {
    // ... (previous database initialization code) ...

    // Set up profiled routes
    http.HandleFunc("/user", trackRequestDuration(handleUser))
    http.HandleFunc("/debug/profile", handleProfile)

    // Start memory stats logging
    go func() {
        ticker := time.NewTicker(1 * time.Minute)
        for range ticker.C {
            logMemStats()
        }
    }()

    // Start the server
    log.Println("Server starting on :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

로그인 후 복사