refactor: move boomer from hrp internal to pkg

hrp/pkg/boomer/utils.go

@@ -0,0 +1,150 @@
+package boomer
+
+import (
+	"crypto/md5"
+	"fmt"
+	"io"
+	"math"
+	"os"
+	"runtime/pprof"
+	"strings"
+	"time"
+
+	"github.com/google/uuid"
+	"github.com/rs/zerolog/log"
+	"github.com/shirou/gopsutil/cpu"
+	"github.com/shirou/gopsutil/mem"
+	"github.com/shirou/gopsutil/process"
+)
+
+func round(val float64, roundOn float64, places int) (newVal float64) {
+	var round float64
+	pow := math.Pow(10, float64(places))
+	digit := pow * val
+	_, div := math.Modf(digit)
+	if div >= roundOn {
+		round = math.Ceil(digit)
+	} else {
+		round = math.Floor(digit)
+	}
+	newVal = round / pow
+	return
+}
+
+// genMD5 returns the md5 hash of strings.
+func genMD5(slice ...string) string {
+	h := md5.New()
+	for _, v := range slice {
+		io.WriteString(h, v)
+	}
+	return fmt.Sprintf("%x", h.Sum(nil))
+}
+
+// startMemoryProfile starts memory profiling and save the results in file.
+func startMemoryProfile(file string, duration time.Duration) (err error) {
+	f, err := os.Create(file)
+	if err != nil {
+		return err
+	}
+
+	log.Info().Dur("duration", duration).Msg("Start memory profiling")
+	time.AfterFunc(duration, func() {
+		err := pprof.WriteHeapProfile(f)
+		if err != nil {
+			log.Error().Err(err).Msg("failed to write memory profile")
+		}
+		f.Close()
+		log.Info().Dur("duration", duration).Msg("Stop memory profiling")
+	})
+	return nil
+}
+
+// startCPUProfile starts cpu profiling and save the results in file.
+func startCPUProfile(file string, duration time.Duration) (err error) {
+	f, err := os.Create(file)
+	if err != nil {
+		return err
+	}
+
+	log.Info().Dur("duration", duration).Msg("Start CPU profiling")
+	err = pprof.StartCPUProfile(f)
+	if err != nil {
+		f.Close()
+		return err
+	}
+
+	time.AfterFunc(duration, func() {
+		pprof.StopCPUProfile()
+		f.Close()
+		log.Info().Dur("duration", duration).Msg("Stop CPU profiling")
+	})
+	return nil
+}
+
+// generate a random nodeID like locust does, using the same algorithm.
+func getNodeID() (nodeID string) {
+	hostname, _ := os.Hostname()
+	id := strings.Replace(uuid.New().String(), "-", "", -1)
+	nodeID = fmt.Sprintf("%s_%s", hostname, id)
+	return
+}
+
+// GetCurrentPidCPUUsage get current pid CPU usage
+func GetCurrentPidCPUUsage() float64 {
+	currentPid := os.Getpid()
+	p, err := process.NewProcess(int32(currentPid))
+	if err != nil {
+		log.Error().Err(err).Msg(fmt.Sprintf("failed to get CPU percent\n"))
+		return 0.0
+	}
+	percent, err := p.CPUPercent()
+	if err != nil {
+		log.Error().Err(err).Msg(fmt.Sprintf("failed to get CPU percent\n"))
+		return 0.0
+	}
+	return percent
+}
+
+// GetCurrentPidCPUPercent get the percentage of current pid cpu used
+func GetCurrentPidCPUPercent() float64 {
+	currentPid := os.Getpid()
+	p, err := process.NewProcess(int32(currentPid))
+	if err != nil {
+		log.Error().Err(err).Msg(fmt.Sprintf("failed to get CPU percent\n"))
+		return 0.0
+	}
+	percent, err := p.Percent(time.Second)
+	if err != nil {
+		log.Error().Err(err).Msg(fmt.Sprintf("failed to get CPU percent\n"))
+		return 0.0
+	}
+	return percent
+}
+
+// GetCurrentCPUPercent get the percentage of current cpu used
+func GetCurrentCPUPercent() float64 {
+	percent, _ := cpu.Percent(time.Second, false)
+	return percent[0]
+}
+
+// GetCurrentMemoryPercent get the percentage of current memory used
+func GetCurrentMemoryPercent() float64 {
+	memInfo, _ := mem.VirtualMemory()
+	return memInfo.UsedPercent
+}
+
+// GetCurrentPidMemoryUsage get current Memory usage
+func GetCurrentPidMemoryUsage() float64 {
+	currentPid := os.Getpid()
+	p, err := process.NewProcess(int32(currentPid))
+	if err != nil {
+		log.Error().Err(err).Msg(fmt.Sprintf("failed to get CPU percent\n"))
+		return 0.0
+	}
+	percent, err := p.MemoryPercent()
+	if err != nil {
+		log.Error().Err(err).Msg(fmt.Sprintf("failed to get CPU percent\n"))
+		return 0.0
+	}
+	return float64(percent)
+}