Move to typical go project layout, fix golangci-lint warnings, update deps

2024-10-06 22:10:19 +03:00
parent 0b573aa169
commit 83970650b5
115 changed files with 9402 additions and 4920 deletions
--- a/internal/lib/reniced.go
+++ b/internal/lib/reniced.go
@@ -0,0 +1,116 @@
+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: */