cleanup project structur
This commit is contained in:
130
main.go
130
main.go
@@ -1,111 +1,59 @@
|
||||
// cmd/sysusage/main.go
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"bufio"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"runtime"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/shirou/gopsutil/v3/cpu"
|
||||
"github.com/shirou/gopsutil/v3/mem"
|
||||
)
|
||||
|
||||
type Result struct {
|
||||
CPUPercent float64 `json:"cpu_percent"`
|
||||
RAMPercent float64 `json:"ram_percent"`
|
||||
}
|
||||
|
||||
func sampleCPU(interval time.Duration, samples int) (float64, error) {
|
||||
if samples <= 0 {
|
||||
samples = 1
|
||||
}
|
||||
var sum float64
|
||||
for i := 0; i < samples; i++ {
|
||||
// cpu.Percent blockiert für "interval" und liefert die Auslastung in diesem Zeitfenster
|
||||
values, err := cpu.Percent(interval, false)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if len(values) == 0 {
|
||||
return 0, fmt.Errorf("no cpu values returned")
|
||||
}
|
||||
sum += values[0]
|
||||
}
|
||||
return sum / float64(samples), nil
|
||||
}
|
||||
|
||||
func readRAM() (float64, error) {
|
||||
vm, err := mem.VirtualMemory()
|
||||
func linuxID() string {
|
||||
f, err := os.Open("/etc/os-release")
|
||||
if err != nil {
|
||||
return 0, err
|
||||
return ""
|
||||
}
|
||||
defer f.Close()
|
||||
sc := bufio.NewScanner(f)
|
||||
for sc.Scan() {
|
||||
line := sc.Text()
|
||||
if strings.HasPrefix(line, "ID=") {
|
||||
val := strings.TrimPrefix(line, "ID=")
|
||||
val = strings.Trim(val, `"`)
|
||||
return strings.ToLower(val)
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func isSupported() bool {
|
||||
switch runtime.GOOS {
|
||||
case "darwin":
|
||||
return true
|
||||
case "linux":
|
||||
id := linuxID()
|
||||
return id == "ubuntu" || id == "arch"
|
||||
default:
|
||||
return false
|
||||
}
|
||||
return vm.UsedPercent, nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
var (
|
||||
intervalStr string
|
||||
samples int
|
||||
jsonOut bool
|
||||
watch bool
|
||||
)
|
||||
|
||||
flag.StringVar(&intervalStr, "interval", "1s", "Messintervall pro Stichprobe (z.B. 200ms, 1s, 2s)")
|
||||
flag.IntVar(&samples, "samples", 1, "Anzahl der Stichproben (über das Intervall gemittelt)")
|
||||
flag.BoolVar(&jsonOut, "json", false, "Ausgabe als JSON")
|
||||
flag.BoolVar(&watch, "watch", false, "Kontinuierlich messen, bis Strg+C")
|
||||
flag.Parse()
|
||||
|
||||
interval, err := time.ParseDuration(intervalStr)
|
||||
if err != nil || interval <= 0 {
|
||||
log.Fatalf("ungültiges -interval: %v", intervalStr)
|
||||
if !isSupported() {
|
||||
return
|
||||
}
|
||||
|
||||
printOnce := func() error {
|
||||
cpuPct, err := sampleCPU(interval, samples)
|
||||
if err != nil {
|
||||
return fmt.Errorf("CPU-Messung fehlgeschlagen: %w", err)
|
||||
}
|
||||
ramPct, err := readRAM()
|
||||
if err != nil {
|
||||
return fmt.Errorf("RAM-Messung fehlgeschlagen: %w", err)
|
||||
}
|
||||
|
||||
if jsonOut {
|
||||
out := Result{CPUPercent: cpuPct, RAMPercent: ramPct}
|
||||
enc := json.NewEncoder(stdoutNoHTMLEscape{})
|
||||
enc.SetIndent("", " ")
|
||||
return enc.Encode(out)
|
||||
}
|
||||
|
||||
fmt.Printf("CPU: %.1f%% | RAM: %.1f%%\n", cpuPct, ramPct)
|
||||
return nil
|
||||
values, err := cpu.Percent(400*time.Millisecond, false)
|
||||
if err != nil || len(values) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
if watch {
|
||||
for {
|
||||
if err := printOnce(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
// Bei watch: zwischen den Zeilen kurz pausieren, damit die Ausgabe lesbar bleibt.
|
||||
time.Sleep(250 * time.Millisecond)
|
||||
}
|
||||
} else {
|
||||
if err := printOnce(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
vm, err := mem.VirtualMemory()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// stdoutNoHTMLEscape verhindert das HTML-Escaping des Standard-Encoders (nur Kosmetik für JSON)
|
||||
type stdoutNoHTMLEscape struct{}
|
||||
|
||||
func (stdoutNoHTMLEscape) Write(p []byte) (int, error) { return fmt.Print(string(p)) }
|
||||
func (stdoutNoHTMLEscape) Encode(v any) error {
|
||||
enc := json.NewEncoder(stdoutNoHTMLEscape{})
|
||||
enc.SetEscapeHTML(false)
|
||||
return enc.Encode(v)
|
||||
fmt.Printf("CPU %.0f%% | RAM %.0f%%", values[0], vm.UsedPercent)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user