package main

import (
    "fmt"
    "runtime"
    "time"
)

func worker(id int) {
    for {
        // 模擬一些工作
        time.Sleep(time.Millisecond * 100)
        fmt.Printf("Worker %d is running\n", id)
    }
}

func main() {
    // 啟動(dòng)幾個(gè)worker goroutine
    for i := 0; i < 3; i++ {
        go worker(i)
    }

    // 定期采集goroutine profile
    ticker := time.NewTicker(time.Second)
    defer ticker.Stop()

    for i := 0; i < 3; i++ {
        <-ticker.C

        // 獲取當(dāng)前的goroutine profile
        profiles := make([]runtime.StackRecord, 10)
        n, ok := runtime.GoroutineProfile(profiles)
        if !ok {
            profiles = make([]runtime.StackRecord, n)
            n, _ = runtime.GoroutineProfile(profiles)
        }

        fmt.Printf("=== Sample %d ===\n", i+1)
        for j, profile := range profiles[:n] {
            fmt.Printf("Goroutine %d: %d stack frames\n", j, len(profile.Stack0))
            // 問(wèn)題：無(wú)法知道這個(gè)goroutine的ID，無(wú)法跨采樣關(guān)聯(lián)
        }
    }
}

// 假設(shè)的新結(jié)構(gòu)體
type GoroutineRecord struct {
    ID    int64           // goroutine ID
    GoPC  uintptr        // goroutine起始PC
    Stack []uintptr      // 調(diào)用棧
    // 其他相關(guān)字段
}

// 對(duì)應(yīng)的新API
func GoroutineProfileWithID() []GoroutineRecord {
    // 實(shí)現(xiàn)細(xì)節(jié)...
}

func trackGoroutines() {
    // 第一次采樣
    sample1 := runtime.GoroutineProfileWithID()

    time.Sleep(time.Second)

    // 第二次采樣
    sample2 := runtime.GoroutineProfileWithID()

    // 現(xiàn)在可以通過(guò)ID關(guān)聯(lián)同一個(gè)goroutine了
    for _, g1 := range sample1 {
        for _, g2 := range sample2 {
            if g1.ID == g2.ID {
                fmt.Printf("Goroutine %d 在兩次采樣中都存在\n", g1.ID)
                // 可以分析這個(gè)goroutine的行為變化
            }
        }
    }
}

// 當(dāng)前的困境：無(wú)法跟蹤特定請(qǐng)求的goroutine
func analyzeSlowRequests() {
    // 采樣1：請(qǐng)求剛開(kāi)始
    profiles1 := getGoroutineProfile()

    // 等待一段時(shí)間
    time.Sleep(5 * time.Second)

    // 采樣2：請(qǐng)求可能仍在處理
    profiles2 := getGoroutineProfile()

    // 問(wèn)題：無(wú)法確定哪個(gè)goroutine處理的是同一個(gè)請(qǐng)求
    // 只能看到所有g(shù)oroutine的快照，但無(wú)法關(guān)聯(lián)
}

// 有了新功能后的可能性
func analyzeSlowRequestsWithID() {
    profiles1 := getGoroutineProfileWithID()
    time.Sleep(5 * time.Second)
    profiles2 := getGoroutineProfileWithID()

    // 可以精確跟蹤特定goroutine的執(zhí)行軌跡
    for _, g1 := range profiles1 {
        for _, g2 := range profiles2 {
            if g1.ID == g2.ID {
                // 分析同一個(gè)goroutine在不同時(shí)間點(diǎn)的棧信息
                // 找出性能瓶頸所在
            }
        }
    }
}