Go archive/zip.Reader怎样逐项读取压缩包内文件怎样完成一个可复用的最小示例

用 Go 逐项读取 ZIP 文件时，核心路径很固定：先用 zip.OpenReader 打开磁盘文件，再遍历 ReadCloser.File，对目录跳过，对普通条目调用 File.Open 读取内容。已有内存映射或随机读取对象时，也可以用 zip.NewReader 创建 zip.Reader。

选择 OpenReader 或 NewReader

archive/zip 同时支持文件路径和已有的随机读取对象。最小场景是直接打开路径；如果数据来自内存映射、对象存储缓存或其他实现了 io.ReaderAt 的对象，则传给 NewReader，并提供准确的字节长度。

package main

import (
    "archive/zip"
    "log"
)

func main() {
    // OpenReader 负责打开磁盘 ZIP；遍历完成后关闭文件描述符。
    reader, err := zip.OpenReader("assets.zip")
    if err != nil {
        log.Fatal(err)
    }
    defer reader.Close()

    // reader.File 保存中央目录中的条目元数据。
    for _, file := range reader.File {
        log.Println(file.Name)
    }
}

OpenReader 返回 *zip.ReadCloser，它既包含可遍历的 Reader，也持有底层文件资源。把 Close 放在成功打开之后，可以避免打开失败时调用空对象。

逐项遍历并区分目录

ZIP 条目不一定都是文件，也可能是以斜杠结尾的目录。使用 file.Mode().IsDir() 判断目录后再决定是否读取，避免把目录当作内容文件处理。名称是归档内路径，不能直接等同于本地文件系统路径。

func listFiles(reader *zip.Reader) []string {
    var names []string

    for _, file := range reader.File {
        // 目录只有结构意义，跳过它可以让结果只保留可读取文件。
        if file.Mode().IsDir() {
            continue
        }
        names = append(names, file.Name)
    }
    return names
}

如果只需要清单，遍历到这里就足够；如果要读取正文，再为单个条目调用 Open。这种分层可以减少不必要的解压工作，也方便先过滤扩展名、路径前缀或条目大小。

打开条目并限制读取范围

File.Open 返回一个 io.ReadCloser，读取时会按 ZIP 的压缩方法解压，并在读取器关闭时释放相关资源。小文本可以使用 io.ReadAll；面对外部 ZIP 或不确定大小的条目，更适合用限制读取器。

package main

import (
    "archive/zip"
    "fmt"
    "io"
)

func readEntry(file *zip.File, maxBytes int64) ([]byte, error) {
    // 目录没有正文；调用方应把它当作输入边界错误处理。
    if file.Mode().IsDir() {
        return nil, fmt.Errorf("目录不能作为文件读取: %s", file.Name)
    }

    body, err := file.Open()
    if err != nil {
        return nil, fmt.Errorf("打开条目 %q: %w", file.Name, err)
    }
    defer body.Close()

    // 限制最大读取量，避免超大条目一次性占满内存。
    limited := io.LimitReader(body, maxBytes+1)
    data, err := io.ReadAll(limited)
    if err != nil {
        return nil, fmt.Errorf("读取条目 %q: %w", file.Name, err)
    }
    if int64(len(data)) > maxBytes {
        return nil, fmt.Errorf("条目 %q 超过读取上限", file.Name)
    }
    return data, nil
}

这里多读一个字节是为了区分“刚好达到上限”和“已经超过上限”。读取错误、关闭错误和业务上限错误要分别保留语义，调用方才可以决定跳过条目、记录告警还是终止导入。

封装成可复用的条目读取函数

实际项目通常不会在业务循环里散落 Open、关闭和错误包装。可以把输出统一成名称、字节数和内容，调用方只接收明确的结果；若条目很多，也可以把 Content 改成写入回调，进一步降低内存占用。

type Entry struct {
    Name string
    Data []byte
}

func readZip(path string, maxBytes int64) ([]Entry, error) {
    // 统一管理归档生命周期，避免调用方忘记关闭底层文件。
    reader, err := zip.OpenReader(path)
    if err != nil {
        return nil, err
    }
    defer reader.Close()

    entries := make([]Entry, 0, len(reader.File))
    for _, file := range reader.File {
        // 目录不进入内容结果；普通文件才调用受限读取函数。
        if file.Mode().IsDir() {
            continue
        }
        data, err := readEntry(file, maxBytes)
        if err != nil {
            return nil, err
        }
        entries = append(entries, Entry{Name: file.Name, Data: data})
    }
    return entries, nil
}

常见边界和处理建议

相关问题

OpenReader 和 NewReader 有什么区别？

OpenReader 面向磁盘路径并负责底层文件关闭；NewReader 面向已有的 io.ReaderAt 和总大小，由调用方管理底层资源。

为什么 file.Open 还要 Close？

它返回的是 io.ReadCloser。即使 ZIP 文件本身已经打开，条目读取器也可能持有解压或校验相关资源，读取完成后应及时关闭。

可以直接把 file.Name 写到磁盘吗？

不能只凭名称就写入。用户上传的 ZIP 还需要检查绝对路径、父目录跳转、符号链接策略和展开总量；本文示例只负责读取内容。