如何使用 Gin 在 HTTP 服务器中即时生成 zip / 7z 存档？

Question

我使用Gin创建一个 HTTP 服务器，我想向用户提供一个动态生成的 zip 存档。

理论上，我可以首先在文件系统上生成一个 zip 文件，然后提供它。但这确实是一个糟糕的方法（在开始下载之前等待 5 分钟）。我想立即开始将其提供给用户并在生成内容时推送内容。

我找到了 DataFromReader ，但在存档完成之前不知道 ContentLength。

func DownloadEndpoint(c *gin.Context) {

    ...

    c.DataFromReader(

        http.StatusOK,

        ContentLength,

        ContentType,

        Body,

        map[string]string{

            "Content-Disposition": "attachment; filename=\"archive.zip\""),

        },

    )

}

我怎样才能做到这一点？

Answer 1

使用流方法和archive/zip，您可以动态创建 zip 并将它们流式传输到服务器。package mainimport (    "os"    "archive/zip"    "github.com/gin-gonic/gin")func main() {    r := gin.Default()    r.GET("/", func(c *gin.Context) {        c.Writer.Header().Set("Content-type", "application/octet-stream")        c.Stream(func(w io.Writer) bool {            // Create a zip archive.            ar := zip.NewWriter(w)            file1, _ := os.Open("filename1")            file2, _ := os.Open("filename2")            c.Writer.Header().Set("Content-Disposition", "attachment; filename='filename.zip'")            f1, _ := ar.Create("filename1")            io.Copy(f1, file1)            f2, _ := ar.Create("filename2")            io.Copy(f2, file2)            ar.Close()            return false        })    })    r.Run()}直接使用 ResponseWriterpackage mainimport (    "io"    "os"    "archive/zip"    "github.com/gin-gonic/gin")func main() {    r := gin.Default()    r.GET("/", func(c *gin.Context) {        c.Writer.Header().Set("Content-type", "application/octet-stream")        c.Writer.Header().Set("Content-Disposition", "attachment; filename='filename.zip'")        ar :=  zip.NewWriter(c.Writer)        file1, _ := os.Open("filename1")        file2, _ := os.Open("filename2")        f1, _ := ar.Create("filename1")        io.Copy(f1, file1)        f2, _ := ar.Create("filename1")        io.Copy(f1, file2)        ar.Close()    })    r.Run()}