117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package lib
|
|
|
|
import (
|
|
"log"
|
|
"math"
|
|
"runtime"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
proc "github.com/shirou/gopsutil/v3/process"
|
|
)
|
|
|
|
// reniced основная логика.
|
|
func (cnf Config) Reniced() {
|
|
for {
|
|
processList, err := proc.Processes()
|
|
|
|
if err != nil {
|
|
log.Fatalf("Unable to get process list: %s", err)
|
|
}
|
|
|
|
// Не самый эффективный, зато работающий способ - простой перебор массивов.
|
|
for _, p := range processList {
|
|
ok := cnf.Pool.TrySubmit(func() {
|
|
if processName, err := p.Name(); err == nil {
|
|
// Для каждого процесса извлекаем его текущий priority.
|
|
if currentPrioLevel, err := syscall.Getpriority(syscall.PRIO_PROCESS, int(p.Pid)); err == nil {
|
|
if niceLevel := cnf.Renice[processName]; niceLevel != 0 {
|
|
prioLevel := niceLevel
|
|
currentNiceLevel := currentPrioLevel
|
|
|
|
if runtime.GOOS == "linux" {
|
|
// Value of PR = 20 + (-20 to +19) is 0 to 39
|
|
switch {
|
|
case niceLevel > 0:
|
|
prioLevel = 20 - niceLevel
|
|
case niceLevel < 0:
|
|
prioLevel = 20 + int(math.Abs(float64(niceLevel)))
|
|
default:
|
|
prioLevel = 20
|
|
}
|
|
|
|
currentNiceLevel = 20 - currentPrioLevel
|
|
}
|
|
|
|
if currentPrioLevel != prioLevel {
|
|
// on Linux Setpriority actually operates niceness value.
|
|
_ = syscall.Setpriority(syscall.PRIO_PROCESS, int(p.Pid), niceLevel)
|
|
|
|
if cnf.Debug {
|
|
log.Printf(
|
|
"Set niceness for %s(%d) to %d (was %d)",
|
|
processName,
|
|
p.Pid,
|
|
niceLevel,
|
|
currentNiceLevel,
|
|
)
|
|
}
|
|
} else if cnf.Debug {
|
|
log.Printf(
|
|
"Niceness for %s(%d) already set to %d",
|
|
processName,
|
|
p.Pid,
|
|
niceLevel,
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
cnf.IORenice(p, processName)
|
|
|
|
// Посылаем процессу сигналы, если таковые есть в конфиге.
|
|
if killSignal := cnf.KillSignal[processName]; killSignal != 0 {
|
|
switch killSignal { //nolint:exhaustive
|
|
case syscall.SIGSTOP:
|
|
_ = p.SendSignal(killSignal)
|
|
|
|
if cnf.Debug {
|
|
log.Printf(
|
|
"Matching processName with killProcessname %s(%d), SIG%s sent",
|
|
processName,
|
|
p.Pid,
|
|
strings.ToUpper(killSignal.String()),
|
|
)
|
|
}
|
|
|
|
default:
|
|
_ = p.SendSignal(killSignal)
|
|
|
|
if cnf.Debug {
|
|
log.Printf(
|
|
"Matching processName with killProcessname %s(%d), SIG%s sent",
|
|
processName,
|
|
p.Pid,
|
|
strings.ToUpper(killSignal.String()),
|
|
)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
if !ok {
|
|
log.Printf("Unable to add task to pool. Increase nb_capacity?")
|
|
}
|
|
}
|
|
|
|
time.Sleep(time.Millisecond * time.Duration(cnf.LoopDelay))
|
|
}
|
|
|
|
// Мы сюда никогда не попадём.
|
|
// pool.StopAndWait()
|
|
} //nolint:wsl
|
|
|
|
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */
|