优化 ai_false diskstate

FloatTech · Mar 11, 2023 · cc26eb1 · cc26eb1
1 parent 366c7ac
commit cc26eb1
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/plugin/ai_false/ai_false.go b/plugin/ai_false/ai_false.go
@@ -567,23 +567,21 @@ func diskstate() (stateinfo []*status, err error) {
 	if err != nil {
 		return
 	}
-	stateinfo = make([]*status, len(parts))
-	for i, v := range parts {
+	stateinfo = make([]*status, 0, len(parts))
+	for _, v := range parts {
 		mp := v.Mountpoint
+		if strings.HasPrefix(mp, "/snap/") || strings.HasPrefix(mp, "/apex/") {
+			continue
+		}
 		diskusage, err := disk.Usage(mp)
-		usage := ""
-		precent := 0.0
 		if err != nil {
-			usage = err.Error()
-		} else {
-			usage = storagefmt(float64(diskusage.Used)) + " / " + storagefmt(float64(diskusage.Total))
-			precent = math.Round(diskusage.UsedPercent)
+			continue
 		}
-		stateinfo[i] = &status{
-			precent: precent,
+		stateinfo = append(stateinfo, &status{
+			precent: math.Round(diskusage.UsedPercent),
 			name:    mp,
-			text:    []string{usage},
-		}
+			text:    []string{storagefmt(float64(diskusage.Used)) + " / " + storagefmt(float64(diskusage.Total))},
+		})
 	}
 	return stateinfo, nil
 }