使用 Golang 获取 Windows 上的驱动器列表

寻求一种更有效的方法来在 Windows 系统上的所有驱动器中搜索特定的驱动器文件类型，Go 程序员可能想知道是否可以在没有用户指定输入的情况下自动获取可用驱动器列表。

解决方案使用GetLogicalDrives 和位操作：

要列出 Windows 系统上的驱动器，可以利用 GetLogicalDrives 功能。此函数返回一个位掩码，每个位代表从“A”到“Z”的驱动器盘符的可用性。

下面是演示该过程的 Golang 代码片段：

package main

import (
    "fmt"
    "syscall"
)

func main() {

    kernel32, _ := syscall.LoadLibrary("kernel32.dll")
    getLogicalDrivesHandle, _ := syscall.GetProcAddress(kernel32, "GetLogicalDrives")

    var drives []string

    if ret, _, callErr := syscall.Syscall(uintptr(getLogicalDrivesHandle), 0, 0, 0, 0); callErr != 0 {
        // handle error
    } else {
        drives = bitsToDrives(uint32(ret))
    }

    fmt.Printf("%v", drives)

}

func bitsToDrives(bitMap uint32) (drives []string) {
    availableDrives := []string{"A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"}

    for i := range availableDrives {
        if bitMap&1 == 1 {
            drives = append(drives, availableDrives[i])
        }
        bitMap >>= 1
    }

    return
}

在此代码中，调用GetLogicalDrives函数来获取位掩码。然后使用位操作技术处理位掩码，以提取可用的驱动器号并将其存储在驱动器片中。通过迭代此切片，您可以轻松访问和处理系统上的所有可用驱动器。