diff --git a/btcd.go b/btcd.go index 4f04657b..2fc3f8cc 100644 --- a/btcd.go +++ b/btcd.go @@ -150,6 +150,8 @@ func btcdMain(serverChan chan<- *server) error { param.SetNetwork(activeNetParams.Params.Net) // prep the claimtrie params + go logMemoryUsage() + // Create server and start it. server, err := newServer(cfg.Listeners, cfg.AgentBlacklist, cfg.AgentWhitelist, db, activeNetParams.Params, interrupt) diff --git a/resourceLogging.go b/resourceLogging.go new file mode 100644 index 00000000..d6e92051 --- /dev/null +++ b/resourceLogging.go @@ -0,0 +1,74 @@ +package main + +import ( + "fmt" + + "github.com/shirou/gopsutil/v3/disk" + "github.com/shirou/gopsutil/v3/mem" + "github.com/shirou/gopsutil/v3/process" + + "os" + "path/filepath" + "time" +) + +func toGB(n uint64) float64 { + return float64(n) / 1024.0 / 1024.0 / 1024.0 +} + +func dirSize(path string) (int64, error) { + var size int64 + err := filepath.Walk(path, func(_ string, info os.FileInfo, err error) error { + if err != nil { + return err + } + if !info.IsDir() { + size += info.Size() + } + return err + }) + return size, err +} + +func logMemoryUsage() { + last := "" + tick := time.NewTicker(40 * time.Second) + for range tick.C { + m, err := mem.VirtualMemory() + if err != nil { + btcdLog.Warnf("When reading memory size: %s", err.Error()) + continue + } + + d, err := disk.Usage(cfg.DataDir) + if err != nil { + btcdLog.Warnf("When reading disk usage: %s", err.Error()) + continue + } + + p, err := process.NewProcess(int32(os.Getpid())) + if err != nil { + btcdLog.Warnf("When reading process: %s", err.Error()) + continue + } + + m2, err := p.MemoryInfo() + if err != nil { + btcdLog.Warnf("When reading memory info: %s", err.Error()) + continue + } + + ds, err := dirSize(cfg.DataDir) + if err != nil { + btcdLog.Debugf("When reading directory: %s", err.Error()) + continue + } + + cur := fmt.Sprintf("RAM: using %.1f GB with %.1f available, DISK: using %.1f GB with %.1f available", + toGB(m2.RSS), toGB(m.Available), toGB(uint64(ds)), toGB(d.Free)) + if cur != last { + btcdLog.Infof(cur) + last = cur + } + } +}