droos02800
2017-02-12 20:06 浏览 175
已采纳

有效地从文件中读取前两个字节-Golang

I'm trying to find a good way of reading the first two bytes from a file using Go.

I have some .zip files in my current directory, mixed in with other files.

I would like to loop through all the files in the directory and check if the first two bytes contain the right .zip identifier, namely 50 4B.

What would be a good way to accomplish this using the standard library without having to read the entire file?

Going through the available functions in the io package I managed to find:

func LimitReader(r Reader, n int64) Reader

Which seems to fit my description, it reads from Reader (How do I get a Reader?) but stops after n bytes. Since I'm rather new to Go, I'm not sure how to go about it.

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享
  • 邀请回答

1条回答 默认 最新

  • 已采纳
    dongqiang5932 dongqiang5932 2017-02-12 20:19

    You get the initial reader by opening the file. For 2 bytes, I wouldn't use the LimitReader though. Just reading 2 bytes with io.ReadFull is easier.

    r, err := os.Open(file)
    if err != nil {
        return err
    }
    
    defer r.Close()
    
    var header [2]byte
    n, err := io.ReadFull(r, header[:])
    if err != nil {
        return err
    }
    
    点赞 评论 复制链接分享

相关推荐