From e68b9d00a6e05b3a941f63ffb696f91e554ac5ec Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 18 Oct 2024 20:33:49 +0200 Subject: Adding upstream version 9.0.3. Signed-off-by: Daniel Baumann --- modules/pprof/pprof.go | 45 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 modules/pprof/pprof.go (limited to 'modules/pprof/pprof.go') diff --git a/modules/pprof/pprof.go b/modules/pprof/pprof.go new file mode 100644 index 0000000..c611c14 --- /dev/null +++ b/modules/pprof/pprof.go @@ -0,0 +1,45 @@ +// Copyright 2018 The Gitea Authors. All rights reserved. +// SPDX-License-Identifier: MIT + +package pprof + +import ( + "fmt" + "os" + "runtime" + "runtime/pprof" + + "code.gitea.io/gitea/modules/log" +) + +// DumpMemProfileForUsername dumps a memory profile at pprofDataPath as memprofile__ +func DumpMemProfileForUsername(pprofDataPath, username string) error { + f, err := os.CreateTemp(pprofDataPath, fmt.Sprintf("memprofile_%s_", username)) + if err != nil { + return err + } + defer f.Close() + runtime.GC() // get up-to-date statistics + return pprof.WriteHeapProfile(f) +} + +// DumpCPUProfileForUsername dumps a CPU profile at pprofDataPath as cpuprofile__ +// the stop function it returns stops, writes and closes the CPU profile file +func DumpCPUProfileForUsername(pprofDataPath, username string) (func(), error) { + f, err := os.CreateTemp(pprofDataPath, fmt.Sprintf("cpuprofile_%s_", username)) + if err != nil { + return nil, err + } + + err = pprof.StartCPUProfile(f) + if err != nil { + log.Fatal("StartCPUProfile: %v", err) + } + return func() { + pprof.StopCPUProfile() + err = f.Close() + if err != nil { + log.Fatal("StopCPUProfile Close: %v", err) + } + }, nil +} -- cgit v1.2.3