You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.3 KiB
56 lines
1.3 KiB
6 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/shirou/gopsutil/process"
|
||
|
|
"github.com/simple-rules/harmony-benchmark/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
type profilerSetting struct {
|
||
|
|
pid int32
|
||
|
|
shardID string
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
setting profilerSetting
|
||
|
|
)
|
||
|
|
|
||
|
|
func logPerf() {
|
||
|
|
p, _ := process.NewProcess(setting.pid)
|
||
|
|
for {
|
||
|
|
// log mem usage
|
||
|
|
info, _ := p.MemoryInfo()
|
||
|
|
memMap, _ := p.MemoryMaps(false)
|
||
|
|
log.Info("Mem Report", "info", info, "map", memMap, "shardID", setting.shardID)
|
||
|
|
|
||
|
|
// log cpu usage
|
||
|
|
percent, _ := p.CPUPercent()
|
||
|
|
times, _ := p.Times()
|
||
|
|
log.Info("CPU Report", "percent", percent, "times", times, "shardID", setting.shardID)
|
||
|
|
|
||
|
|
time.Sleep(3 * time.Second)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
pid := flag.Int("pid", 0, "process id of the node")
|
||
|
|
shardID := flag.String("shard_id", "0", "the shard id of this node")
|
||
|
|
logFolder := flag.String("log_folder", "latest", "the folder collecting the logs of this execution")
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
setting.pid = int32(*pid)
|
||
|
|
setting.shardID = *shardID
|
||
|
|
logFileName := fmt.Sprintf("./%v/profiler-%v.log", *logFolder, *shardID)
|
||
|
|
h := log.MultiHandler(
|
||
|
|
log.StdoutHandler,
|
||
|
|
log.Must.FileHandler(logFileName, log.JSONFormat()), // Log to file
|
||
|
|
// log.Must.NetHandler("tcp", ":3000", log.JSONFormat()) // Log to remote
|
||
|
|
)
|
||
|
|
log.Root().SetHandler(h)
|
||
|
|
|
||
|
|
logPerf()
|
||
|
|
}
|