1
0
forked from ALHP/ALHP.GO

switch result pointer with channel

This commit is contained in:
2025-03-22 23:13:15 +01:00
parent f4f64e1999
commit bcfaccfec5
2 changed files with 12 additions and 11 deletions

View File

@@ -849,10 +849,13 @@ func getMemoryStats(pid int) (MemStats, error) {
return stats, nil
}
func pollMemoryUsage(pid int, interval time.Duration, done chan bool, peakMem *int64) {
func pollMemoryUsage(pid int, interval time.Duration, done chan bool, result chan int64) {
var totalMemory int64
for {
select {
case <-done:
result <- totalMemory
return
default:
var totalRSS, totalSwap int64
@@ -878,9 +881,9 @@ func pollMemoryUsage(pid int, interval time.Duration, done chan bool, peakMem *i
}
}
totalMemory := totalRSS + totalSwap
if peakMem == nil || totalMemory > *peakMem {
peakMem = &totalMemory
newMemory := totalRSS + totalSwap
if newMemory > totalMemory {
totalMemory = newMemory
}
time.Sleep(interval)