Merge pull request #86 from CrabappleProject/fit_windows

FIX: remove `os.stat` in order to fit in windows
This commit is contained in:
Ben Kochie 2022-10-20 11:40:28 +01:00 committed by GitHub
commit cee06eca3c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 34 additions and 21 deletions

35
main.go
View File

@ -94,19 +94,36 @@ func main() {
level.Info(logger).Log("msg", "Starting smartctl_exporter", "version", version.Info())
level.Info(logger).Log("msg", "Build context", "build_context", version.BuildContext())
devices := *smartctlDevices
// Scan the host devices
json := readSMARTctlDevices(logger)
scanDevices := json.Get("devices").Array()
scanDevicesSet := make(map[string]bool)
var scanDeviceNames []string
for _, d := range scanDevices {
deviceName := d.Get("name").String()
level.Debug(logger).Log("msg", "Found device", "name", deviceName)
scanDevicesSet[deviceName] = true
scanDeviceNames = append(scanDeviceNames, deviceName)
}
if len(devices) == 0 {
level.Info(logger).Log("msg", "No devices specified, trying to load them automatically")
json := readSMARTctlDevices(logger)
scannedDevices := json.Get("devices").Array()
for _, d := range scannedDevices {
device := d.Get("name").String()
level.Info(logger).Log("msg", "Found device", "device", device)
devices = append(devices, device)
// Read the configuration and verify that it is available
devices := *smartctlDevices
var readDeviceNames []string
for _, device := range devices {
if _, ok := scanDevicesSet[device]; ok {
readDeviceNames = append(readDeviceNames, device)
} else {
level.Warn(logger).Log("msg", "Device unavailable", "name", device)
}
}
if len(readDeviceNames) > 0 {
devices = readDeviceNames
} else {
level.Info(logger).Log("msg", "No devices specified, trying to load them automatically")
devices = scanDeviceNames
}
if len(devices) == 0 {
level.Error(logger).Log("msg", "No devices found")
os.Exit(1)

View File

@ -16,7 +16,6 @@ package main
import (
"fmt"
"io/ioutil"
"os"
"os/exec"
"strings"
"time"
@ -94,19 +93,16 @@ func readData(logger log.Logger, device string) (gjson.Result, error) {
return readFakeSMARTctl(logger, device), nil
}
if _, err := os.Stat(device); err == nil {
cacheValue, cacheOk := jsonCache[device]
if !cacheOk || time.Now().After(cacheValue.LastCollect.Add(*smartctlInterval)) {
json, ok := readSMARTctl(logger, device)
if ok {
jsonCache[device] = JSONCache{JSON: json, LastCollect: time.Now()}
return jsonCache[device].JSON, nil
}
return gjson.Parse("{}"), fmt.Errorf("smartctl returned bad data for device %s", device)
cacheValue, cacheOk := jsonCache[device]
if !cacheOk || time.Now().After(cacheValue.LastCollect.Add(*smartctlInterval)) {
json, ok := readSMARTctl(logger, device)
if ok {
jsonCache[device] = JSONCache{JSON: json, LastCollect: time.Now()}
return jsonCache[device].JSON, nil
}
return cacheValue.JSON, nil
return gjson.Parse("{}"), fmt.Errorf("smartctl returned bad data for device %s", device)
}
return gjson.Parse("{}"), fmt.Errorf("Device %s unavialable", device)
return cacheValue.JSON, nil
}
// Parse smartctl return code