cpu.go

package profiling

import (
	"errors"
	"os"
	"path/filepath"
	"runtime/pprof"
)

// CPUFilename is a filename in which the CPU profiling is stored.
const CPUFilename = "status_cpu.prof"

var cpuFile *os.File

// StartCPUProfile enables CPU profiling for the current process. While profiling,
// the profile will be buffered and written to the file in folder dataDir.
func StartCPUProfile(dataDir string) error {
	if cpuFile != nil {
		return errors.New("cpu profiling is already started")
	}

	var err error
	cpuFile, err = os.Create(filepath.Join(dataDir, CPUFilename))
	if err != nil {
		return err
	}

	return pprof.StartCPUProfile(cpuFile)
}

// StopCPUProfile stops the current CPU profile, if any, and closes the file.
func StopCPUProfile() error {
	if cpuFile == nil {
		return nil
	}
	pprof.StopCPUProfile()
	err := cpuFile.Close()
	cpuFile = nil
	return err
}