imrplement amd power
This commit is contained in:
+13
-2
@@ -16,11 +16,13 @@ func RenderCPU(console *strings.Builder) {
|
|||||||
console.WriteString("\033[H")
|
console.WriteString("\033[H")
|
||||||
console.WriteString(fmt.Sprintf("=== %s ===\n", proc.Model()))
|
console.WriteString(fmt.Sprintf("=== %s ===\n", proc.Model()))
|
||||||
|
|
||||||
RenderTemperatures(console, proc)
|
RenderTemperatureSensors(console, proc)
|
||||||
|
RenderPowerSensors(console, proc)
|
||||||
}
|
}
|
||||||
|
|
||||||
func RenderTemperatures(console *strings.Builder, proc proc.Proc) {
|
func RenderTemperatureSensors(console *strings.Builder, proc proc.Proc) {
|
||||||
temps, _ := proc.SensorReader.ReadTemperatureSensors()
|
temps, _ := proc.SensorReader.ReadTemperatureSensors()
|
||||||
|
console.WriteString("=== Temperature ===\n")
|
||||||
for _, sensor := range temps {
|
for _, sensor := range temps {
|
||||||
tempC := sensor.TempC
|
tempC := sensor.TempC
|
||||||
tempF := (tempC * 9 / 5) + 32
|
tempF := (tempC * 9 / 5) + 32
|
||||||
@@ -28,3 +30,12 @@ func RenderTemperatures(console *strings.Builder, proc proc.Proc) {
|
|||||||
console.WriteString(fmt.Sprintf("Sensor: %-25s | %5.1f°C (%5.1f°F)\033[K\n", sensor.Name, tempC, tempF))
|
console.WriteString(fmt.Sprintf("Sensor: %-25s | %5.1f°C (%5.1f°F)\033[K\n", sensor.Name, tempC, tempF))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func RenderPowerSensors(console *strings.Builder, proc proc.Proc) {
|
||||||
|
console.WriteString("=== Power ===\n")
|
||||||
|
power, _ := proc.SensorReader.ReadPowerSensors()
|
||||||
|
for _, sensor := range power {
|
||||||
|
// \033[K clears from cursor to end of line (prevents lingering chars)
|
||||||
|
console.WriteString(fmt.Sprintf("Sensor: %-25s | %5.1fW\033[K\n", sensor.Name, sensor.Watts))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ type Proc struct {
|
|||||||
|
|
||||||
type CPUSensorReader interface {
|
type CPUSensorReader interface {
|
||||||
ReadTemperatureSensors() ([]TempSensor, error)
|
ReadTemperatureSensors() ([]TempSensor, error)
|
||||||
|
ReadPowerSensors() ([]PowerSensor, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
type TempSensor struct {
|
type TempSensor struct {
|
||||||
@@ -21,6 +22,11 @@ type TempSensor struct {
|
|||||||
TempC float64
|
TempC float64
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type PowerSensor struct {
|
||||||
|
Name string
|
||||||
|
Watts float64
|
||||||
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
VendorIntel string = "GenuineIntel"
|
VendorIntel string = "GenuineIntel"
|
||||||
VendorAMD string = "AuthenticAMD"
|
VendorAMD string = "AuthenticAMD"
|
||||||
|
|||||||
+154
-1
@@ -12,9 +12,10 @@ import (
|
|||||||
type AMDReader struct {
|
type AMDReader struct {
|
||||||
hwmonDir string
|
hwmonDir string
|
||||||
driver string
|
driver string
|
||||||
|
energyState map[string]EnergyState // tracks (path -> energy state) for energy*_input calculations
|
||||||
}
|
}
|
||||||
|
|
||||||
type PowerState struct {
|
type EnergyState struct {
|
||||||
lastEnergy uint64
|
lastEnergy uint64
|
||||||
lastTime time.Time
|
lastTime time.Time
|
||||||
}
|
}
|
||||||
@@ -78,3 +79,155 @@ func (r *AMDReader) ReadTemperatureSensors() ([]TempSensor, error) {
|
|||||||
|
|
||||||
return temps, nil
|
return temps, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (r *AMDReader) ReadPowerSensors() ([]PowerSensor, error) {
|
||||||
|
hwmonSensors := r.readHwmonPower()
|
||||||
|
if len(hwmonSensors) > 0 {
|
||||||
|
return hwmonSensors, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return r.readRaplPower(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AMDReader) readHwmonPower() []PowerSensor {
|
||||||
|
var powerSensors []PowerSensor
|
||||||
|
|
||||||
|
getLabel := func(inputPath string) string {
|
||||||
|
labelPath := strings.TrimSuffix(inputPath, "_input") + "_label"
|
||||||
|
if lBytes, err := os.ReadFile(labelPath); err == nil {
|
||||||
|
return strings.TrimSpace(string(lBytes))
|
||||||
|
}
|
||||||
|
return filepath.Base(strings.TrimSuffix(inputPath, "_input"))
|
||||||
|
}
|
||||||
|
|
||||||
|
powerInputs, _ := filepath.Glob(filepath.Join(r.hwmonDir, "power*_input"))
|
||||||
|
for _, inputPath := range powerInputs {
|
||||||
|
valBytes, err := os.ReadFile(inputPath)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
microWatts, err := strconv.ParseFloat(strings.TrimSpace(string(valBytes)), 64)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
powerSensors = append(powerSensors, PowerSensor{
|
||||||
|
Name: getLabel(inputPath),
|
||||||
|
Watts: microWatts / 1_000_000.0,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
energyInputs, _ := filepath.Glob(filepath.Join(r.hwmonDir, "energy*_input"))
|
||||||
|
if len(energyInputs) > 0 {
|
||||||
|
r.calculateEnergyDelta(energyInputs, getLabel, &powerSensors)
|
||||||
|
}
|
||||||
|
|
||||||
|
return powerSensors
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AMDReader) readRaplPower() []PowerSensor {
|
||||||
|
var powerSensors []PowerSensor
|
||||||
|
|
||||||
|
// Match both top-level and nested RAPL zone directories
|
||||||
|
patterns := []string{
|
||||||
|
"/sys/class/powercap/intel-rapl*/energy_uj",
|
||||||
|
"/sys/class/powercap/intel-rapl/intel-rapl*/energy_uj",
|
||||||
|
"/sys/class/powercap/intel-rapl/intel-rapl*/*/energy_uj",
|
||||||
|
}
|
||||||
|
|
||||||
|
var energyInputs []string
|
||||||
|
seen := make(map[string]bool)
|
||||||
|
|
||||||
|
for _, pattern := range patterns {
|
||||||
|
matches, _ := filepath.Glob(pattern)
|
||||||
|
for _, match := range matches {
|
||||||
|
// Resolve symlinks to get real sysfs path
|
||||||
|
realPath, err := filepath.EvalSymlinks(match)
|
||||||
|
if err != nil {
|
||||||
|
realPath = match
|
||||||
|
}
|
||||||
|
if !seen[realPath] {
|
||||||
|
seen[realPath] = true
|
||||||
|
energyInputs = append(energyInputs, realPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(energyInputs) == 0 {
|
||||||
|
return powerSensors
|
||||||
|
}
|
||||||
|
|
||||||
|
getLabel := func(inputPath string) string {
|
||||||
|
dir := filepath.Dir(inputPath)
|
||||||
|
if nameBytes, err := os.ReadFile(filepath.Join(dir, "name")); err == nil {
|
||||||
|
return strings.TrimSpace(string(nameBytes))
|
||||||
|
}
|
||||||
|
return filepath.Base(dir)
|
||||||
|
}
|
||||||
|
|
||||||
|
r.calculateEnergyDelta(energyInputs, getLabel, &powerSensors)
|
||||||
|
return powerSensors
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AMDReader) calculateEnergyDelta(inputs []string, getLabel func(string) string, out *[]PowerSensor) {
|
||||||
|
now := time.Now()
|
||||||
|
if r.energyState == nil {
|
||||||
|
r.energyState = make(map[string]EnergyState)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, inputPath := range inputs {
|
||||||
|
// Canonicalize path to resolve symlinks and prevent map key duplication
|
||||||
|
realPath, err := filepath.EvalSymlinks(inputPath)
|
||||||
|
if err != nil {
|
||||||
|
realPath = inputPath
|
||||||
|
}
|
||||||
|
|
||||||
|
valBytes, err := os.ReadFile(realPath)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
microJoules, err := strconv.ParseUint(strings.TrimSpace(string(valBytes)), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
prevState, exists := r.energyState[realPath]
|
||||||
|
r.energyState[realPath] = EnergyState{
|
||||||
|
lastEnergy: microJoules,
|
||||||
|
lastTime: now,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initial seed step: power delta requires two distinct reads
|
||||||
|
if !exists {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
timeDiff := now.Sub(prevState.lastTime).Seconds()
|
||||||
|
if timeDiff <= 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
maxRange := uint64(1<<64 - 1)
|
||||||
|
rangeBytes, err := os.ReadFile(filepath.Join(filepath.Dir(realPath), "max_energy_range_uj"))
|
||||||
|
if err == nil {
|
||||||
|
if parsedMax, err := strconv.ParseUint(strings.TrimSpace(string(rangeBytes)), 10, 64); err == nil {
|
||||||
|
maxRange = parsedMax
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var energyDiff uint64
|
||||||
|
if microJoules >= prevState.lastEnergy {
|
||||||
|
energyDiff = microJoules - prevState.lastEnergy
|
||||||
|
} else {
|
||||||
|
energyDiff = maxRange - prevState.lastEnergy + microJoules
|
||||||
|
}
|
||||||
|
|
||||||
|
watts := (float64(energyDiff) / 1_000_000.0) / timeDiff
|
||||||
|
*out = append(*out, PowerSensor{
|
||||||
|
Name: getLabel(realPath),
|
||||||
|
Watts: watts,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -74,3 +74,7 @@ func (r *IntelReader) ReadTemperatureSensors() ([]TempSensor, error) {
|
|||||||
|
|
||||||
return temps, nil
|
return temps, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (r *IntelReader) ReadPowerSensors() ([]PowerSensor, error) {
|
||||||
|
return []PowerSensor{}, nil
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user