golang pprof

golang pprof

package common

import (
	"fmt"
	"net"
	"net/http"
	_ "net/http/pprof"
	"time"
)

func NewPprofServe(ports ...int64) {
	fn := "NewPprofServe"
	addr := ":0"
	if len(ports) > 0 {
		addr = fmt.Sprintf(":%d", ports[0])
	}
	server := &http.Server{
		Addr:    addr,
		Handler: nil,
	}
	actualAddr := make(chan string, 1)
	go func() {
		listener, err := net.Listen("tcp", server.Addr)
		if err != nil {
			fmt.Println(fn, " Listener error:", err)
			return
		}

		actualPort := listener.Addr().(*net.TCPAddr).Port
		actualAddr <- fmt.Sprintf(":%d", actualPort)

		// 启动 HTTP 服务器
		err = server.Serve(listener)
		if err != nil {
			fmt.Println(fn, " Server error:", err)
		}
	}()
	select {
	case adr := <-actualAddr:
		fmt.Println(fn, " pprof addr -> ", adr)
	case <-time.After(time.Second * 3):
		fmt.Println(fn, " pprof server timeout 3s")
		return
	}
}