如何获取高效文件系统I/O的「首选」块大小——跨操作系统通用方案

Rust标准库本身没有提供跨平台统一的API来获取文件系统的首选I/O块大小，因为不同操作系统的底层实现差异较大。不过你有两种可靠的解决方案：手动适配各平台的系统API，或者借助成熟的第三方crates。

方案一：手动适配平台API #

通过条件编译分别处理Unix类系统（Linux、BSD、macOS等）和Windows系统：

Unix类系统 #

Unix系系统可以直接使用标准库的std::os::unix::fs::MetadataExt trait，它提供了st_blksize方法，直接返回文件系统的首选块大小：

use std::fs::metadata;
use std::os::unix::fs::MetadataExt;
use std::io;

fn get_block_size_unix(path: &str) -> io::Result<u64> {
    let meta = metadata(path)?;
    Ok(meta.st_blksize() as u64)
}

Windows系统 #

Windows需要调用系统API获取磁盘簇大小（簇大小等价于文件系统的首选I/O块大小），可以通过winapi crate实现：

use std::fs::metadata;
use std::io;
use std::ffi::OsStr;
use std::os::windows::ffi::OsStrExt;
use winapi::um::fileapi::GetDiskFreeSpaceW;

fn get_block_size_windows(path: &str) -> io::Result<u64> {
    // 将路径转换为Windows宽字符格式
    let path_wide: Vec<u16> = OsStr::new(path)
        .encode_wide()
        .chain(std::iter::once(0))
        .collect();
    
    let mut sectors_per_cluster = 0;
    let mut bytes_per_sector = 0;
    let mut _free_clusters = 0;
    let mut _total_clusters = 0;

    unsafe {
        if GetDiskFreeSpaceW(
            path_wide.as_ptr(),
            &mut sectors_per_cluster,
            &mut bytes_per_sector,
            &mut _free_clusters,
            &mut _total_clusters,
        ) == 0
        {
            return Err(io::Error::last_os_error());
        }
    }

    Ok((sectors_per_cluster * bytes_per_sector) as u64)
}

把两者结合成跨平台函数：

use std::fs::metadata;
use std::io;

fn get_preferred_block_size(path: &str) -> io::Result<u64> {
    let _ = metadata(path)?; // 确保路径有效

    #[cfg(unix)]
    {
        use std::os::unix::fs::MetadataExt;
        let meta = metadata(path)?;
        Ok(meta.st_blksize() as u64)
    }

    #[cfg(windows)]
    {
        use std::ffi::OsStr;
        use std::os::windows::ffi::OsStrExt;
        use winapi::um::fileapi::GetDiskFreeSpaceW;

        let path_wide: Vec<u16> = OsStr::new(path)
            .encode_wide()
            .chain(std::iter::once(0))
            .collect();
        
        let mut sectors_per_cluster = 0;
        let mut bytes_per_sector = 0;
        let mut _free_clusters = 0;
        let mut _total_clusters = 0;

        unsafe {
            if GetDiskFreeSpaceW(
                path_wide.as_ptr(),
                &mut sectors_per_cluster,
                &mut bytes_per_sector,
                &mut _free_clusters,
                &mut _total_clusters,
            ) == 0
            {
                return Err(io::Error::last_os_error());
            }
        }

        Ok((sectors_per_cluster * bytes_per_sector) as u64)
    }

    #[cfg(not(any(unix, windows)))]
    {
        Err(io::Error::new(io::ErrorKind::Unsupported, "当前平台不支持获取块大小"))
    }
}

方案二：使用第三方crates #

如果不想手动处理跨平台逻辑，推荐使用fs3 crate，它已经封装了跨平台的文件系统信息获取逻辑：

1. 在Cargo.toml中添加依赖：

fs3 = "0.6"

2. 调用示例：

use fs3::FileSystem;
use std::fs::File;
use std::io;

fn get_preferred_block_size(path: &str) -> io::Result<u64> {
    let file = File::open(path)?;
    let fs_info = file.file_system_info()?;
    Ok(fs_info.block_size())
}

注意事项 #

• 块大小是和文件所在的文件系统绑定的，所以必须传入一个存在的文件或目录路径来获取对应文件系统的信息
• 对于少数无法获取块大小的平台，可以 fallback 到4KB作为默认值（这是现代文件系统最常用的块大小）

内容的提问来源于stack exchange，提问作者Harry