feat: 启动时输出监控的文件夹和网卡名

monitoring/unit/disk.go

@@ -90,3 +90,27 @@ func isPhysicalDisk(part disk.PartitionStat) bool {
 
 	return true
 }
+
+func DiskList() ([]string, error) {
+	diskList := []string{}
+	if flags.IncludeMountpoints != "" {
+		includeMounts := strings.Split(flags.IncludeMountpoints, ";")
+		for _, mountpoint := range includeMounts {
+			mountpoint = strings.TrimSpace(mountpoint)
+			if mountpoint != "" {
+				diskList = append(diskList, mountpoint)
+			}
+		}
+	} else {
+		usage, err := disk.Partitions(false)
+		if err != nil {
+			return nil, err
+		}
+		for _, part := range usage {
+			if isPhysicalDisk(part) {
+				diskList = append(diskList, part.Mountpoint)
+			}
+		}
+	}
+	return diskList, nil
+}

monitoring/unit/net.go

@@ -321,3 +321,30 @@ func shouldInclude(nicName string, includeNics, excludeNics map[string]struct{})
 
 	return true
 }
+
+func InterfaceList() ([]string, error) {
+	includeNics := parseNics(flags.IncludeNics)
+	excludeNics := parseNics(flags.ExcludeNics)
+	interfaces := []string{}
+	if flags.MonthRotate != 0 {
+		vnstatData, err := getVnstatData()
+		if err == nil {
+			for interfaceName := range vnstatData {
+				if shouldInclude(interfaceName, includeNics, excludeNics) {
+					interfaces = append(interfaces, interfaceName)
+				}
+			}
+			return interfaces, nil
+		}
+	}
+	ioCounters, err := net.IOCounters(true)
+	if err != nil {
+		return nil, err
+	}
+	for _, interfaceStats := range ioCounters {
+		if shouldInclude(interfaceStats.Name, includeNics, excludeNics) {
+			interfaces = append(interfaces, interfaceStats.Name)
+		}
+	}
+	return interfaces, nil
+}