init

monitoring/disk.go

@@ -0,0 +1,35 @@
+package monitoring
+
+import (
+	"github.com/shirou/gopsutil/disk"
+)
+
+type DiskInfo struct {
+	Total uint64 `json:"total"`
+	Used  uint64 `json:"used"`
+}
+
+func Disk() DiskInfo {
+	diskinfo := DiskInfo{}
+	usage, err := disk.Partitions(true)
+	if err != nil {
+		diskinfo.Total = 0
+		diskinfo.Used = 0
+	} else {
+		for _, part := range usage {
+			if part.Mountpoint != "/tmp" && part.Mountpoint != "/var/tmp" && part.Mountpoint != "/dev/shm" {
+				// Skip /tmp, /var/tmp, and /dev/shm
+				// 获取磁盘使用情况
+				u, err := disk.Usage(part.Mountpoint)
+				if err != nil {
+					diskinfo.Total = 0
+					diskinfo.Used = 0
+				} else {
+					diskinfo.Total += u.Total
+					diskinfo.Used += u.Used
+				}
+			}
+		}
+	}
+	return diskinfo
+}