|
1 | | -package chief |
2 | | - |
3 | | -import ( |
4 | | - "encoding/json" |
5 | | - "fmt" |
6 | | - "strconv" |
7 | | - "syscall" |
8 | | - "unsafe" |
9 | | - "xiaowumin-SFM/Struct" |
10 | | -) |
11 | | - |
12 | | -type DiskUsage struct { |
13 | | - Name string |
14 | | - TotalSpace float64 |
15 | | - UsedSpace float64 |
16 | | - FreeSpace float64 |
17 | | -} |
18 | | - |
19 | | -const ( |
20 | | - DRIVE_FIXED = 3 |
21 | | - DRIVE_UNKNOWN = 0 |
22 | | -) |
23 | | - |
24 | | -var kernel32 = syscall.NewLazyDLL("kernel32.dll") |
25 | | -var getDiskFreeSpaceEx = kernel32.NewProc("GetDiskFreeSpaceExW") |
26 | | -var getDriveType = kernel32.NewProc("GetDriveTypeW") |
27 | | - |
28 | | -func getDiskUsage() ([]DiskUsage, error) { |
29 | | - var diskUsages []DiskUsage |
30 | | - |
31 | | - kernel32 := syscall.NewLazyDLL("kernel32.dll") |
32 | | - GetLogicalDrives := kernel32.NewProc("GetLogicalDrives") |
33 | | - bitmask, _, _ := GetLogicalDrives.Call() |
34 | | - |
35 | | - for i := 0; i < 26; i++ { |
36 | | - if (bitmask>>i)&1 == 0 { |
37 | | - continue // Continue if the drive doesn't exist |
38 | | - } |
39 | | - |
40 | | - drive := string(rune('A'+i)) + ":\\" |
41 | | - driveType, _, _ := getDriveType.Call(uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(drive)))) |
42 | | - if driveType != DRIVE_FIXED { |
43 | | - continue // Continue if the drive is not a fixed drive |
44 | | - } |
45 | | - |
46 | | - lpFreeBytesAvailable := int64(0) |
47 | | - lpTotalNumberOfBytes := int64(0) |
48 | | - lpTotalNumberOfFreeBytes := int64(0) |
49 | | - |
50 | | - _, _, _ = getDiskFreeSpaceEx.Call( |
51 | | - uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(drive))), |
52 | | - uintptr(unsafe.Pointer(&lpFreeBytesAvailable)), |
53 | | - uintptr(unsafe.Pointer(&lpTotalNumberOfBytes)), |
54 | | - uintptr(unsafe.Pointer(&lpTotalNumberOfFreeBytes)), |
55 | | - ) |
56 | | - |
57 | | - diskUsages = append(diskUsages, DiskUsage{ |
58 | | - Name: drive, |
59 | | - TotalSpace: float64(lpTotalNumberOfBytes) / (1024 * 1024 * 1024), |
60 | | - UsedSpace: (float64(lpTotalNumberOfBytes) - float64(lpTotalNumberOfFreeBytes)) / (1024 * 1024 * 1024), |
61 | | - FreeSpace: float64(lpTotalNumberOfFreeBytes) / (1024 * 1024 * 1024), |
62 | | - }) |
63 | | - } |
64 | | - |
65 | | - return diskUsages, nil |
66 | | -} |
67 | | - |
68 | | -func Config() []byte { |
69 | | - diskUsages, err := getDiskUsage() |
70 | | - if err != nil { |
71 | | - fmt.Println("Failed to retrieve disk usage:", err) |
72 | | - |
73 | | - } |
74 | | - |
75 | | - var DiskAll []string |
76 | | - var DiskName []string |
77 | | - var DiskFreeSpace []string |
78 | | - var DiskUse []string |
79 | | - for i := 0; i < len(diskUsages); i++ { |
80 | | - DiskAll = append(DiskAll, strconv.FormatFloat(diskUsages[i].TotalSpace, 'f', 2, 64)+" GB") |
81 | | - DiskName = append(DiskName, diskUsages[i].Name) |
82 | | - DiskFreeSpace = append(DiskFreeSpace, strconv.FormatFloat(diskUsages[i].FreeSpace, 'f', 2, 64)+" GB") |
83 | | - DiskUse = append(DiskUse, strconv.FormatFloat(diskUsages[i].UsedSpace, 'f', 2, 64)+" GB") |
84 | | - } |
85 | | - |
86 | | - fwqc := Struct.Config{ |
87 | | - Disks: struct { |
88 | | - All []string `json:"All"` |
89 | | - Name []string `json:"Name"` |
90 | | - Numbers int `json:"Numbers"` |
91 | | - Residue []string `json:"Residue"` |
92 | | - Use []string `json:"Use"` |
93 | | - }{ |
94 | | - All: DiskAll, |
95 | | - Name: DiskName, |
96 | | - Numbers: len(diskUsages), |
97 | | - Residue: DiskFreeSpace, |
98 | | - Use: DiskUse, |
99 | | - }, |
100 | | - } |
101 | | - // 打印JSON字符串 |
102 | | - // 将结构体转换为JSON字符串 |
103 | | - jsonData, err := json.Marshal(fwqc) |
104 | | - if err != nil { |
105 | | - fmt.Println("转换为JSON时出错:", err) |
106 | | - |
107 | | - } |
108 | | - |
109 | | - // 返回JSON |
110 | | - return jsonData |
111 | | -} |
| 1 | +package chief |
| 2 | + |
| 3 | +import ( |
| 4 | + "encoding/json" |
| 5 | + "fmt" |
| 6 | + "strconv" |
| 7 | + "syscall" |
| 8 | + "unsafe" |
| 9 | + "xiaowumin-SFM/Struct" |
| 10 | +) |
| 11 | + |
| 12 | +type DiskUsage struct { |
| 13 | + Name string |
| 14 | + TotalSpace float64 |
| 15 | + UsedSpace float64 |
| 16 | + FreeSpace float64 |
| 17 | +} |
| 18 | + |
| 19 | +const ( |
| 20 | + DRIVE_FIXED = 3 |
| 21 | + DRIVE_UNKNOWN = 0 |
| 22 | +) |
| 23 | + |
| 24 | +var kernel32 = syscall.NewLazyDLL("kernel32.dll") |
| 25 | +var getDiskFreeSpaceEx = kernel32.NewProc("GetDiskFreeSpaceExW") |
| 26 | +var getDriveType = kernel32.NewProc("GetDriveTypeW") |
| 27 | + |
| 28 | +func getDiskUsage() ([]DiskUsage, error) { |
| 29 | + var diskUsages []DiskUsage |
| 30 | + |
| 31 | + kernel32 := syscall.NewLazyDLL("kernel32.dll") |
| 32 | + GetLogicalDrives := kernel32.NewProc("GetLogicalDrives") |
| 33 | + bitmask, _, _ := GetLogicalDrives.Call() |
| 34 | + |
| 35 | + for i := 0; i < 26; i++ { |
| 36 | + if (bitmask>>i)&1 == 0 { |
| 37 | + continue // 如果驱动器不存在，继续 |
| 38 | + } |
| 39 | + |
| 40 | + drive := string(rune('A'+i)) + ":\\" |
| 41 | + driveType, _, _ := getDriveType.Call(uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(drive)))) |
| 42 | + if driveType != DRIVE_FIXED { |
| 43 | + continue // 如果驱动器不是固定驱动器，请继续 |
| 44 | + } |
| 45 | + |
| 46 | + lpFreeBytesAvailable := int64(0) |
| 47 | + lpTotalNumberOfBytes := int64(0) |
| 48 | + lpTotalNumberOfFreeBytes := int64(0) |
| 49 | + |
| 50 | + _, _, _ = getDiskFreeSpaceEx.Call( |
| 51 | + uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(drive))), |
| 52 | + uintptr(unsafe.Pointer(&lpFreeBytesAvailable)), |
| 53 | + uintptr(unsafe.Pointer(&lpTotalNumberOfBytes)), |
| 54 | + uintptr(unsafe.Pointer(&lpTotalNumberOfFreeBytes)), |
| 55 | + ) |
| 56 | + |
| 57 | + diskUsages = append(diskUsages, DiskUsage{ |
| 58 | + Name: drive, |
| 59 | + TotalSpace: float64(lpTotalNumberOfBytes) / (1024 * 1024 * 1024), |
| 60 | + UsedSpace: (float64(lpTotalNumberOfBytes) - float64(lpTotalNumberOfFreeBytes)) / (1024 * 1024 * 1024), |
| 61 | + FreeSpace: float64(lpTotalNumberOfFreeBytes) / (1024 * 1024 * 1024), |
| 62 | + }) |
| 63 | + } |
| 64 | + |
| 65 | + return diskUsages, nil |
| 66 | +} |
| 67 | + |
| 68 | +func Config() []byte { |
| 69 | + diskUsages, err := getDiskUsage() |
| 70 | + if err != nil { |
| 71 | + fmt.Println("检索磁盘使用率失败:", err) |
| 72 | + } |
| 73 | + |
| 74 | + var DiskAll []string |
| 75 | + var DiskName []string |
| 76 | + var DiskFreeSpace []string |
| 77 | + var DiskUse []string |
| 78 | + for i := 0; i < len(diskUsages); i++ { |
| 79 | + DiskAll = append(DiskAll, strconv.FormatFloat(diskUsages[i].TotalSpace, 'f', 2, 64)+" GB") |
| 80 | + DiskName = append(DiskName, diskUsages[i].Name) |
| 81 | + DiskFreeSpace = append(DiskFreeSpace, strconv.FormatFloat(diskUsages[i].FreeSpace, 'f', 2, 64)+" GB") |
| 82 | + DiskUse = append(DiskUse, strconv.FormatFloat(diskUsages[i].UsedSpace, 'f', 2, 64)+" GB") |
| 83 | + } |
| 84 | + |
| 85 | + fwqc := Struct.Config{ |
| 86 | + Disks: struct { |
| 87 | + All []string `json:"All"` |
| 88 | + Name []string `json:"Name"` |
| 89 | + Numbers int `json:"Numbers"` |
| 90 | + Residue []string `json:"Residue"` |
| 91 | + Use []string `json:"Use"` |
| 92 | + }{ |
| 93 | + All: DiskAll, |
| 94 | + Name: DiskName, |
| 95 | + Numbers: len(diskUsages), |
| 96 | + Residue: DiskFreeSpace, |
| 97 | + Use: DiskUse, |
| 98 | + }, |
| 99 | + } |
| 100 | + // 打印JSON字符串 |
| 101 | + // 将结构体转换为JSON字符串 |
| 102 | + jsonData, err := json.Marshal(fwqc) |
| 103 | + if err != nil { |
| 104 | + fmt.Println("转换为JSON时出错:", err) |
| 105 | + |
| 106 | + } |
| 107 | + |
| 108 | + // 返回JSON |
| 109 | + return jsonData |
| 110 | +} |
0 commit comments