golang - 如何从golang服务器下载浏览器中的文件？

Question

我的代码从远程URL获取文件并在浏览器中下载文件:

func Index(w http.ResponseWriter, r *http.Request) {
    url := "http://upload.wikimedia.org/wikipedia/en/b/bc/Wiki.png"

    ...

    resp, err := client.Get(url)
    if err != nil {
        fmt.Println(err)
    }
    defer resp.Body.Close()

    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        fmt.Println(err)
    }

    fmt.Println(len(body))
    //download the file in browser

}

func main() {
    http.HandleFunc("/", Index)
    err := http.ListenAndServe(":8000", nil)

    if err != nil {
        fmt.Println(err)
    }
}

代码:http://play.golang.org/p/x-EyR2zFjv

获取文件是可以的,但如何在浏览器中下载？

谢谢.

Answer 1

要使浏览器打开下载对话框,请在响应中添加一个Content-Disposition和Content-Type标题:

w.Header().Set("Content-Disposition", "attachment; filename=WHATEVER_YOU_WANT")
w.Header().Set("Content-Type", r.Header.Get("Content-Type"))

在将内容发送到客户端之前执行此操作.您可能还希望将Content-Length响应的标头复制到客户端,以显示正确的进度.

要将响应主体流式传输到客户端而不将其完全加载到内存中(对于大文件这很重要) - 只需将主体读取器复制到响应编写器:

io.Copy(w, resp.Body)

io.Copy是一个很好的小函数,它采用读取器接口和编写器接口,从一个读取数据并将其写入另一个.对这种东西非常有用!

我修改了你的代码来做到这一点:http://play.golang.org/p/v9IAu2Xu3_

Answer 2

如果您已经在磁盘上有该文件，只需使用http.ServeFile()。它会自动处理Content-Length以便浏览器可以显示下载进度。

w.Header().Set("Content-Disposition", "attachment; filename="+strconv.Quote(filename))
w.Header().Set("Content-Type", "application/octet-stream")
http.ServeFile(w, r, filePath)