[agent-wdd] 基本完成Info部分的整理

agent-wdd/config/Disk.go

@@ -0,0 +1,133 @@
+package config
+
+import (
+	"agent-wdd/log"
+	"bufio"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"syscall"
+)
+
+var CommonDiskPath = []string{
+	"/",
+	"/var",
+	"/var/lib/docker",
+	"/home",
+	"/user",
+	"var/log",
+}
+
+func CheckCommonMounts() (map[string]bool, error) {
+	// 读取/proc/mounts获取所有挂载点
+	file, err := os.Open("/proc/mounts")
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+
+	mountPoints := make(map[string]bool)
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		line := scanner.Text()
+		fields := strings.Fields(line)
+		if len(fields) < 2 {
+			continue
+		}
+		mountPoint := filepath.Clean(fields[1])
+		mountPoints[mountPoint] = true
+	}
+
+	if err := scanner.Err(); err != nil {
+		return nil, err
+	}
+
+	return mountPoints, err
+}
+
+func DiskListGather() {
+
+	log.Info("Gathering INFO => DISK !")
+
+	var diskList []Disk
+
+	// 拿到所有挂载点
+	mountPoints, _ := CheckCommonMounts()
+
+	// 常用目录
+	for _, diskPath := range CommonDiskPath {
+		// 转换为绝对路径并标准化
+		absPath, err := filepath.Abs(diskPath)
+		if err != nil {
+			// 处理错误，例如路径无效，这里选择跳过
+			continue
+		}
+		cleanPath := filepath.Clean(absPath)
+		if mountPoints[cleanPath] {
+			// 挂载点存在，计算使用情况
+
+			disk := Disk{Path: cleanPath}
+			disk.calculateDiskUsage()
+			diskList = append(diskList, disk)
+		}
+
+	}
+
+	// 赋值回去
+	ConfigCache.Agent.Disks = diskList
+
+	//utils.BeautifulPrint(diskList)
+}
+
+func (disk *Disk) calculateDiskUsage() {
+	var stat syscall.Statfs_t
+	err := syscall.Statfs(disk.Path, &stat)
+	if err != nil {
+		log.Error("disk syscall error of %v", err)
+
+		disk.Size = "0B"
+		disk.Usage = "0B"
+		disk.Percent = "0.00%"
+		return
+	}
+
+	// 计算存储空间大小
+	totalBytes := stat.Blocks * uint64(stat.Bsize)
+	availBytes := stat.Bavail * uint64(stat.Bsize)
+	usedBytes := totalBytes - availBytes
+
+	// 格式化输出
+	disk.Size = formatDiskSize(totalBytes)
+	disk.Usage = formatDiskSize(usedBytes)
+
+	if totalBytes == 0 {
+		disk.Percent = "0.00%"
+	} else {
+		percent := float64(usedBytes) / float64(totalBytes) * 100
+		disk.Percent = fmt.Sprintf("%.2f%%", percent)
+	}
+}
+
+func formatDiskSize(bytes uint64) string {
+	units := []string{"B", "KB", "MB", "GB", "TB", "PB"}
+	var unitIndex int
+	size := float64(bytes)
+
+	for size >= 1000 && unitIndex < len(units)-1 {
+		size /= 1000
+		unitIndex++
+	}
+
+	if unitIndex == 0 {
+		return strconv.FormatUint(bytes, 10) + units[unitIndex]
+	}
+	return strconv.FormatFloat(size, 'f', 1, 64) + units[unitIndex]
+}
+
+func DiskListSaveConfig() {
+	log.Info("Saving INFO => DISK !")
+
+	SaveConfig()
+}