Получение списка дисков в Windows с помощью Golang

Поиск более эффективного способа поиска по всем дискам в системе Windows для конкретного типа файла, программисты Go могут задаться вопросом, можно ли автоматически получить список доступных дисков без указания пользователем input.

Решение с использованием 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 вызывается для получения битовой маски. Затем битовая маска обрабатывается с использованием методов битовой манипуляции для извлечения доступных букв дисков и сохранения их в срезе дисков. Просматривая этот фрагмент, вы можете легко получить доступ и обработать все доступные диски в системе.