goid_go_test.go

package unsafe_test

import (
    "fmt"
    "github.com/mzzsfy/go-util/unsafe"
    "runtime"
    "strings"
    "sync"
    "testing"
    _ "unsafe"
)

func Test_GoID(t *testing.T) {
    var wg sync.WaitGroup
    for i := 0; i < 5000; i++ {
        wg.Add(1)
        go func() {
            testGoID(t)
            wg.Done()
        }()
    }
    wg.Wait()
}

func testGoID(t *testing.T) {
    id := unsafe.GoID()
    lines := strings.Split(stackTrace(), "\n")
    for i, line := range lines {
        if !strings.HasPrefix(line, fmt.Sprintf("goroutine %d ", id)) {
            continue
        }
        if i+1 == len(lines) {
            break
        }
        if !strings.Contains(lines[i+1], ".stackTrace") {
            t.Errorf("there are goroutine id %d but it is not me: %s", id, lines[i+1])
        }
        return
    }
    t.Errorf("there are no goroutine %d", id)
}

func stackTrace() string {
    var n int
    for n = 4096; n < 16777216; n *= 2 {
        buf := make([]byte, n)
        ret := runtime.Stack(buf, false)
        if ret != n {
            return string(buf[:ret])
        }
    }
    panic(n)
}