From a32ab2887203354866e5e5cf4f4b09908928d50c Mon Sep 17 00:00:00 2001
From: zhanghan <zhanghan@kylinos.cn>
Date: Thu, 9 Dec 2021 17:08:08 +0800
Subject: [PATCH] add get disk info

---
 pkg/utils/os/disk.go | 84 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)

diff --git a/pkg/utils/os/disk.go b/pkg/utils/os/disk.go
index cf1f67e0..2753e97f 100644
--- a/pkg/utils/os/disk.go
+++ b/pkg/utils/os/disk.go
@@ -1 +1,85 @@
 package os
+
+import (
+	"fmt"
+
+	"github.com/shirou/gopsutil/disk"
+)
+
+type DiskIOInfo struct {
+	PartitionName string
+	Label         string
+	ReadCount     uint64
+	WriteCount    uint64
+	ReadBytes     uint64
+	WriteBytes    uint64
+	IOTime        uint64
+}
+
+type DiskUsageINfo struct {
+	Device      string  `json:"device"`
+	Path        string  `json:"path"`
+	Fstype      string  `json:"fstype"`
+	Total       uint64  `json:"total"`
+	Free        uint64  `json:"free"`
+	Used        uint64  `json:"used"`
+	UsedPercent float64 `json:"usedPercent"`
+}
+
+// 获取磁盘的使用情况
+func GetDiskUsageInfo() []DiskUsageINfo {
+	diskusage := make([]DiskUsageINfo, 0)
+	parts, err := disk.Partitions(true)
+	if err != nil {
+		fmt.Printf("get Partitions failed, err:%v\n", err)
+		return nil
+	}
+	for _, part := range parts {
+		diskInfo, _ := disk.Usage(part.Mountpoint)
+		device := part.Device
+		path := diskInfo.Path
+		fstype := diskInfo.Fstype
+		total := diskInfo.Total
+		free := diskInfo.Free
+		used := diskInfo.Used
+		usedPercent := diskInfo.UsedPercent
+		tmp := DiskUsageINfo{
+			Device:      device,
+			Path:        path,
+			Fstype:      fstype,
+			Total:       total,
+			Free:        free,
+			Used:        used,
+			UsedPercent: usedPercent,
+		}
+		diskusage = append(diskusage, tmp)
+	}
+	return diskusage
+}
+
+// 获取磁盘的IO信息
+func GetDiskInfo() []DiskIOInfo {
+
+	diskinfo := make([]DiskIOInfo, 0)
+	ioStat, _ := disk.IOCounters()
+	for k, v := range ioStat {
+		label := v.Label
+		readCount := v.ReadCount
+		writeCount := v.WriteCount
+		readBytes := v.ReadBytes
+		writeBytes := v.WriteBytes
+		ioTime := v.IoTime
+
+		tmp := DiskIOInfo{
+			PartitionName: k,
+			Label:         label,
+			ReadCount:     readCount,
+			WriteCount:    writeCount,
+			ReadBytes:     readBytes,
+			WriteBytes:    writeBytes,
+			IOTime:        ioTime,
+		}
+		diskinfo = append(diskinfo, tmp)
+	}
+	return diskinfo
+}
-- 
Gitee