Compare commits

...

3 Commits

Author SHA1 Message Date
henrygd
aab5725d82 Use gpu temp as primary sensor if no other sensors 2025-04-18 18:00:39 -04:00
henrygd
e94a1cd421 brew install - change env var from PORT to LISTEN 2025-04-18 17:59:59 -04:00
henrygd
73c1a1b208 Refactor sensor configuration handling in tests and implementation
- Add skipCollection propery
- Ensure that sensors are initialized as an empty map
2025-04-18 17:59:25 -04:00
4 changed files with 85 additions and 40 deletions

View File

@@ -13,26 +13,31 @@ import (
) )
type SensorConfig struct { type SensorConfig struct {
context context.Context context context.Context
sensors map[string]struct{} sensors map[string]struct{}
primarySensor string primarySensor string
isBlacklist bool isBlacklist bool
hasWildcards bool hasWildcards bool
skipCollection bool
} }
func (a *Agent) newSensorConfig() *SensorConfig { func (a *Agent) newSensorConfig() *SensorConfig {
primarySensor, _ := GetEnv("PRIMARY_SENSOR") primarySensor, _ := GetEnv("PRIMARY_SENSOR")
sysSensors, _ := GetEnv("SYS_SENSORS") sysSensors, _ := GetEnv("SYS_SENSORS")
sensors, _ := GetEnv("SENSORS") sensorsEnvVal, sensorsSet := GetEnv("SENSORS")
skipCollection := sensorsSet && sensorsEnvVal == ""
return a.newSensorConfigWithEnv(primarySensor, sysSensors, sensors) return a.newSensorConfigWithEnv(primarySensor, sysSensors, sensorsEnvVal, skipCollection)
} }
// newSensorConfigWithEnv creates a SensorConfig with the provided environment variables // newSensorConfigWithEnv creates a SensorConfig with the provided environment variables
func (a *Agent) newSensorConfigWithEnv(primarySensor, sysSensors, sensors string) *SensorConfig { // sensorsSet indicates if the SENSORS environment variable was explicitly set (even to empty string)
func (a *Agent) newSensorConfigWithEnv(primarySensor, sysSensors, sensorsEnvVal string, skipCollection bool) *SensorConfig {
config := &SensorConfig{ config := &SensorConfig{
context: context.Background(), context: context.Background(),
primarySensor: primarySensor, primarySensor: primarySensor,
skipCollection: skipCollection,
sensors: make(map[string]struct{}),
} }
// Set sensors context (allows overriding sys location for sensors) // Set sensors context (allows overriding sys location for sensors)
@@ -43,22 +48,18 @@ func (a *Agent) newSensorConfigWithEnv(primarySensor, sysSensors, sensors string
) )
} }
// Set sensors whitelist // handle blacklist
if sensors != "" { if strings.HasPrefix(sensorsEnvVal, "-") {
// handle blacklist config.isBlacklist = true
if strings.HasPrefix(sensors, "-") { sensorsEnvVal = sensorsEnvVal[1:]
config.isBlacklist = true }
sensors = sensors[1:]
}
config.sensors = make(map[string]struct{}) for sensor := range strings.SplitSeq(sensorsEnvVal, ",") {
for sensor := range strings.SplitSeq(sensors, ",") { sensor = strings.TrimSpace(sensor)
sensor = strings.TrimSpace(sensor) if sensor != "" {
if sensor != "" { config.sensors[sensor] = struct{}{}
config.sensors[sensor] = struct{}{} if strings.Contains(sensor, "*") {
if strings.Contains(sensor, "*") { config.hasWildcards = true
config.hasWildcards = true
}
} }
} }
} }
@@ -69,7 +70,7 @@ func (a *Agent) newSensorConfigWithEnv(primarySensor, sysSensors, sensors string
// updateTemperatures updates the agent with the latest sensor temperatures // updateTemperatures updates the agent with the latest sensor temperatures
func (a *Agent) updateTemperatures(systemStats *system.Stats) { func (a *Agent) updateTemperatures(systemStats *system.Stats) {
// skip if sensors whitelist is set to empty string // skip if sensors whitelist is set to empty string
if a.sensorConfig.sensors != nil && len(a.sensorConfig.sensors) == 0 { if a.sensorConfig.skipCollection {
slog.Debug("Skipping temperature collection") slog.Debug("Skipping temperature collection")
return return
} }
@@ -113,8 +114,8 @@ func (a *Agent) updateTemperatures(systemStats *system.Stats) {
// isValidSensor checks if a sensor is valid based on the sensor name and the sensor config // isValidSensor checks if a sensor is valid based on the sensor name and the sensor config
func isValidSensor(sensorName string, config *SensorConfig) bool { func isValidSensor(sensorName string, config *SensorConfig) bool {
// If no sensors configuration, everything is valid // if no sensors configured, everything is valid
if config.sensors == nil { if len(config.sensors) == 0 {
return true return true
} }
@@ -123,7 +124,7 @@ func isValidSensor(sensorName string, config *SensorConfig) bool {
return !config.isBlacklist return !config.isBlacklist
} }
// If no wildcards, return false if blacklist, true if whitelist // If no wildcards, return true if blacklist, false if whitelist
if !config.hasWildcards { if !config.hasWildcards {
return config.isBlacklist return config.isBlacklist
} }

View File

@@ -97,10 +97,13 @@ func TestIsValidSensor(t *testing.T) {
expectedValid: true, expectedValid: true,
}, },
{ {
name: "Nil sensor config", name: "No sensors configured",
sensorName: "any_temp", sensorName: "any_temp",
config: &SensorConfig{ config: &SensorConfig{
sensors: nil, sensors: map[string]struct{}{},
isBlacklist: false,
hasWildcards: false,
skipCollection: false,
}, },
expectedValid: true, expectedValid: true,
}, },
@@ -162,6 +165,7 @@ func TestNewSensorConfigWithEnv(t *testing.T) {
primarySensor string primarySensor string
sysSensors string sysSensors string
sensors string sensors string
skipCollection bool
expectedConfig *SensorConfig expectedConfig *SensorConfig
}{ }{
{ {
@@ -170,22 +174,38 @@ func TestNewSensorConfigWithEnv(t *testing.T) {
sysSensors: "", sysSensors: "",
sensors: "", sensors: "",
expectedConfig: &SensorConfig{ expectedConfig: &SensorConfig{
context: context.Background(), context: context.Background(),
primarySensor: "", primarySensor: "",
sensors: nil, sensors: map[string]struct{}{},
isBlacklist: false, isBlacklist: false,
hasWildcards: false, hasWildcards: false,
skipCollection: false,
}, },
}, },
{ {
name: "Primary sensor only", name: "Explicitly set to empty string",
primarySensor: "",
sysSensors: "",
sensors: "",
skipCollection: true,
expectedConfig: &SensorConfig{
context: context.Background(),
primarySensor: "",
sensors: map[string]struct{}{},
isBlacklist: false,
hasWildcards: false,
skipCollection: true,
},
},
{
name: "Primary sensor only - should create sensor map",
primarySensor: "cpu_temp", primarySensor: "cpu_temp",
sysSensors: "", sysSensors: "",
sensors: "", sensors: "",
expectedConfig: &SensorConfig{ expectedConfig: &SensorConfig{
context: context.Background(), context: context.Background(),
primarySensor: "cpu_temp", primarySensor: "cpu_temp",
sensors: nil, sensors: map[string]struct{}{},
isBlacklist: false, isBlacklist: false,
hasWildcards: false, hasWildcards: false,
}, },
@@ -238,6 +258,22 @@ func TestNewSensorConfigWithEnv(t *testing.T) {
hasWildcards: true, hasWildcards: true,
}, },
}, },
{
name: "Sensors with whitespace",
primarySensor: "cpu_temp",
sysSensors: "",
sensors: "cpu_*, gpu_temp",
expectedConfig: &SensorConfig{
context: context.Background(),
primarySensor: "cpu_temp",
sensors: map[string]struct{}{
"cpu_*": {},
"gpu_temp": {},
},
isBlacklist: false,
hasWildcards: true,
},
},
{ {
name: "With SYS_SENSORS path", name: "With SYS_SENSORS path",
primarySensor: "cpu_temp", primarySensor: "cpu_temp",
@@ -256,7 +292,7 @@ func TestNewSensorConfigWithEnv(t *testing.T) {
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
result := agent.newSensorConfigWithEnv(tt.primarySensor, tt.sysSensors, tt.sensors) result := agent.newSensorConfigWithEnv(tt.primarySensor, tt.sysSensors, tt.sensors, tt.skipCollection)
// Check primary sensor // Check primary sensor
assert.Equal(t, tt.expectedConfig.primarySensor, result.primarySensor) assert.Equal(t, tt.expectedConfig.primarySensor, result.primarySensor)

View File

@@ -199,16 +199,24 @@ func (a *Agent) getSystemStats() system.Stats {
if systemStats.Temperatures == nil { if systemStats.Temperatures == nil {
systemStats.Temperatures = make(map[string]float64, len(gpuData)) systemStats.Temperatures = make(map[string]float64, len(gpuData))
} }
highestTemp := 0.0
for _, gpu := range gpuData { for _, gpu := range gpuData {
if gpu.Temperature > 0 { if gpu.Temperature > 0 {
systemStats.Temperatures[gpu.Name] = gpu.Temperature systemStats.Temperatures[gpu.Name] = gpu.Temperature
if a.sensorConfig.primarySensor == gpu.Name { if a.sensorConfig.primarySensor == gpu.Name {
a.systemInfo.DashboardTemp = gpu.Temperature a.systemInfo.DashboardTemp = gpu.Temperature
} }
if gpu.Temperature > highestTemp {
highestTemp = gpu.Temperature
}
} }
// update high gpu percent for dashboard // update high gpu percent for dashboard
a.systemInfo.GpuPct = max(a.systemInfo.GpuPct, gpu.Usage) a.systemInfo.GpuPct = max(a.systemInfo.GpuPct, gpu.Usage)
} }
// use highest temp for dashboard temp if dashboard temp is unset
if a.systemInfo.DashboardTemp == 0 {
a.systemInfo.DashboardTemp = highestTemp
}
} }
} }

View File

@@ -67,7 +67,7 @@ fi
mkdir -p ~/.config/beszel ~/.cache/beszel mkdir -p ~/.config/beszel ~/.cache/beszel
echo "KEY=\"$KEY\"" >~/.config/beszel/beszel-agent.env echo "KEY=\"$KEY\"" >~/.config/beszel/beszel-agent.env
echo "PORT=$PORT" >>~/.config/beszel/beszel-agent.env echo "LISTEN=$PORT" >>~/.config/beszel/beszel-agent.env
brew tap henrygd/beszel brew tap henrygd/beszel
brew install beszel-agent brew install beszel-agent