refactor findMaxReadsDevice to use disk.IOCounters

beszel/internal/agent/agent.go

@@ -658,36 +658,17 @@ func (a *Agent) closeIdleConnections(err error) (isTimeout bool) {
 // Returns the device with the most reads in /proc/diskstats
 // (fallback in case the root device is not supplied or detected)
 func findMaxReadsDevice() string {
-	content, err := os.ReadFile("/proc/diskstats")
+	var maxReadBytes uint64
-	if err != nil {
+	maxReadDevice := "/"
-		return "/"
+	counters, err := disk.IOCounters()
-	}
-
-	lines := strings.Split(string(content), "\n")
-	var maxReadsSectors int64
-	var maxReadsDevice string
-
-	for _, line := range lines {
-		fields := strings.Fields(line)
-		if len(fields) < 7 {
-			continue
-		}
-
-		deviceName := fields[2]
-		readsSectors, err := strconv.ParseInt(fields[5], 10, 64)
 		if err != nil {
-			continue
+		return maxReadDevice
-		}
+	}
-
+	for _, d := range counters {
-		if readsSectors > maxReadsSectors {
+		if d.ReadBytes > maxReadBytes {
-			maxReadsSectors = readsSectors
+			maxReadBytes = d.ReadBytes
-			maxReadsDevice = deviceName
+			maxReadDevice = d.Name
 		}
 	}
-
+	return maxReadDevice
-	if maxReadsDevice == "" {
-		return "/"
-	}
-
-	return maxReadsDevice
 }