f6df7126b6
* feat(admin): add profiling options for debugging high memory/CPU usage Add -debug, -debug.port, -cpuprofile, and -memprofile flags to the admin command, matching the profiling support already available in master, volume, and other server commands. This enables investigation of resource usage issues like #8919. * refactor(admin): move profiling flags into AdminOptions struct Move cpuprofile and memprofile flags from global variables into the AdminOptions struct and init() function for consistency with other flags. * fix(debug): bind pprof server to localhost only and document profiling flags StartDebugServer was binding to all interfaces (0.0.0.0), exposing runtime profiling data to the network. Restrict to 127.0.0.1 since this is a development/debugging tool. Also add a "Debugging and Profiling" section to the admin command's help text documenting the new flags.
71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package grace
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
"os"
|
|
"runtime"
|
|
"runtime/pprof"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
)
|
|
|
|
// StartDebugServer starts an HTTP server for pprof debugging on the specified port.
|
|
// The server runs in a goroutine and serves pprof endpoints at /debug/pprof/*.
|
|
func StartDebugServer(debugPort int) {
|
|
go func() {
|
|
addr := fmt.Sprintf("127.0.0.1:%d", debugPort)
|
|
glog.V(0).Infof("Starting debug server for pprof at http://%s/debug/pprof/", addr)
|
|
if err := http.ListenAndServe(addr, nil); err != nil && err != http.ErrServerClosed {
|
|
glog.Errorf("Failed to start debug server on %s: %v", addr, err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func SetupProfiling(cpuProfile, memProfile string) {
|
|
if cpuProfile != "" {
|
|
f, err := os.Create(cpuProfile)
|
|
if err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
runtime.SetBlockProfileRate(1)
|
|
runtime.SetMutexProfileFraction(1)
|
|
pprof.StartCPUProfile(f)
|
|
OnInterrupt(func() {
|
|
pprof.StopCPUProfile()
|
|
|
|
// write block pprof
|
|
blockF, err := os.Create(cpuProfile + ".block")
|
|
if err != nil {
|
|
return
|
|
}
|
|
p := pprof.Lookup("block")
|
|
p.WriteTo(blockF, 0)
|
|
blockF.Close()
|
|
|
|
// write mutex pprof
|
|
mutexF, err := os.Create(cpuProfile + ".mutex")
|
|
if err != nil {
|
|
return
|
|
}
|
|
p = pprof.Lookup("mutex")
|
|
p.WriteTo(mutexF, 0)
|
|
mutexF.Close()
|
|
|
|
})
|
|
}
|
|
if memProfile != "" {
|
|
runtime.MemProfileRate = 1
|
|
f, err := os.Create(memProfile)
|
|
if err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
OnInterrupt(func() {
|
|
pprof.WriteHeapProfile(f)
|
|
f.Close()
|
|
})
|
|
}
|
|
|
|
}
|