Johan Brandhorst-Satzkorn
8ed560fc3e
* fix(cmd): Add profiling build imports It seems like this file has been bitrotted since last used, and it was missing some imports. This was discovered by running goimports on the code. * Add build-memprof Makefile target
42 lines
1.0 KiB
Go
42 lines
1.0 KiB
Go
//go:build memprofiler
|
|
// +build memprofiler
|
|
|
|
package base
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"runtime/pprof"
|
|
"time"
|
|
|
|
"github.com/hashicorp/boundary/internal/observability/event"
|
|
)
|
|
|
|
func StartMemProfiler(ctx context.Context) {
|
|
const op = "base.StartMemProfiler"
|
|
profileDir := filepath.Join(os.TempDir(), "boundaryprof")
|
|
if err := os.MkdirAll(profileDir, 0o700); err != nil {
|
|
event.WriteError(ctx, op, err, event.WithInfoMsg("could not create profile directory"))
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
for {
|
|
filename := filepath.Join(profileDir, time.Now().UTC().Format("20060102_150405")) + ".pprof"
|
|
f, err := os.Create(filename)
|
|
if err != nil {
|
|
event.WriteError(ctx, op, err, event.WithInfoMsg("could not create memory profile"))
|
|
}
|
|
runtime.GC()
|
|
if err := pprof.WriteHeapProfile(f); err != nil {
|
|
event.WriteError(ctx, op, err, event.WithInfoMsg("could not write memory profile"))
|
|
}
|
|
f.Close()
|
|
event.WriteSysEvent(ctx, op, "wrote memory profile", "filename", filename)
|
|
time.Sleep(5 * time.Minute)
|
|
}
|
|
}()
|
|
}
|