Go语言中将二进制File转换为[]byte的最优实现方法及代码修复

现有代码故障原因 #

你提供的代码无法稳定运行的核心问题是：

• os.File实现的io.Reader接口的Read()方法不保证一次性填满传入的整个缓冲区，仅会返回本次实际读取到的字节长度。如果文件较大或操作系统IO调度存在限制，单次Read()调用可能只读取了部分文件内容，导致返回的字节数组后半部分为0值的无效内容。
• 冗余的错误打印不符合工具函数的设计规范，一般不需要在函数内部自行打印错误，直接将错误返回给上层调用方处理即可。

现有代码修复版本 #

使用标准库io.ReadFull代替原生Read方法，即可保证缓冲区被完全填满：

import (
    "os"
    "io"
)

func ReadFileAndReturnByteArray(extractedFilePath string) ([]byte, error) {
    file, err := os.Open(extractedFilePath)
    if err != nil {
        return nil, err
    }
    defer file.Close()

    fileinfo, err := file.Stat()
    if err != nil {
        return nil, err
    }

    filesize := fileinfo.Size()
    buffer := make([]byte, filesize)

    _, err = io.ReadFull(file, buffer)
    if err != nil {
        return nil, err
    }
    return buffer, nil
}

最优实现方式 #

Go 1.16及以上版本的标准库已经提供了官方封装的全文件读取方法os.ReadFile，底层已经做了IO性能优化，不需要自行实现读取逻辑，是目前的最优方案：

import "os"

func ReadFileAndReturnByteArray(extractedFilePath string) ([]byte, error) {
    return os.ReadFile(extractedFilePath)
}

• 该方法内部已经完整处理了文件打开、资源关闭、多次读取保证内容完整的逻辑，代码简洁度和性能都优于自行实现的版本。
• 如果你需要兼容Go 1.16之前的版本，可以使用ioutil.ReadFile，功能和os.ReadFile完全一致，1.16版本后ioutil的相关方法已经被迁移到os包下。

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