使用缓冲区作为输出时压缩文件夹已损坏

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《使用缓冲区作为输出时压缩文件夹已损坏》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

我正在尝试将 zip 文件写入缓冲区而不是文件，以最终将其传递给 http 响应。下面是模拟它的代码。

package main

import (
    "archive/zip"
    "bytes"
    "io"
    "io/ioutil"
    "os"
    "path/filepath"
    "strings"
)

func main() {
    data, err := zipit("myfolder")
    if err != nil {
        panic(err)
    }
    ioutil.writefile("output.zip", data, os.modeperm)
}

func zipit(source string) ([]byte, error) {
    buf := new(bytes.buffer)
    archive := zip.newwriter(buf)
    defer archive.close()
    info, err := os.stat(source)
    if err != nil {
        return nil, nil
    }

    var basedir string
    if info.isdir() {
        basedir = filepath.base(source)
    }
    filepath.walk(source, func(path string, info os.fileinfo, err error) error {
        if err != nil {
            return err
        }

        header, err := zip.fileinfoheader(info)
        if err != nil {
            return err
        }

        if basedir != "" {
            header.name = filepath.join(basedir, strings.trimprefix(path, source))
        }

        if info.isdir() {
            header.name += "/"
        } else {
            header.method = zip.deflate
        }

        writer, err := archive.createheader(header)
        if err != nil {
            return err
        }

        if info.isdir() {
            return nil
        }

        file, err := os.open(path)
        if err != nil {
            return err
        }
        defer file.close()
        _, err = io.copy(writer, file)
        return err
    })
    return buf.bytes(), err
}

但是此 zip 过程的输出已损坏。如果我使用文件而不是缓冲区，它会起作用。

zipfile, err := os.Create(target)
if err != nil {
    return err
}
defer zipfile.Close()

archive := zip.NewWriter(zipfile)
defer archive.Close()

os.file 和 bytes.buffer 都实现了 io.writer 接口，并且有资格作为写入器传递给 zip.newwrite() 方法。

任何解决此问题的指示将不胜感激。

解决方案

您在关闭 zip.writer 之前读取了缓冲区，因此需要刷新到缓冲区的任何最终数据都已丢失。您应该删除 defer archive.close() 并在从缓冲区中取出字节之前关闭存档。例如

err = archive.Close()
    return buf.Bytes(), err

它适用于您的文件箱，因为您推迟关闭文件。延迟将后进先出应用，因此存档会在文件关闭之前关闭。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。