gzip 压缩到 http responseWriter

Question

我是 Go 新手。但是我正在玩 REST Api。我无法从 json.Marshal 获得与 json.Encoder 在我拥有的两个函数中相同的行为

我想使用此功能对我的回复进行 gzip 压缩：

func gzipFast(a *[]byte) []byte {
    var b bytes.Buffer
    gz := gzip.NewWriter(&b)
    defer gz.Close()
    if _, err := gz.Write(*a); err != nil {
        panic(err)
    }
    return b.Bytes()
}

但是这个函数returns这个：

curl http://localhost:8081/compressedget --compressed --verbose
*   Trying 127.0.0.1...
* Connected to localhost (127.0.0.1) port 8081 (#0)
> GET /compressedget HTTP/1.1
> Host: localhost:8081
> User-Agent: curl/7.47.0
> Accept: */*
> Accept-Encoding: deflate, gzip
> 
< HTTP/1.1 200 OK
< Content-Encoding: gzip
< Content-Type: application/json
< Date: Wed, 24 Aug 2016 00:59:38 GMT
< Content-Length: 30
< 
* Connection #0 to host localhost left intact

这是 go 函数：

func CompressedGet(w http.ResponseWriter, r *http.Request, ps  httprouter.Params) {

    box := Box{Width: 10, Height: 20, Color: "gree", Open: false}
    box.ars = make([]int, 100)
    for i := range box.ars {
        box.ars[i] = i
    }
    //fmt.Println(r.Header.Get("Content-Encoding"))

    w.Header().Set("Content-Type", "application/json")
    w.Header().Set("Content-Encoding", "gzip")
    b, _ := json.Marshal(box)
    //fmt.Println(len(b))
    //fmt.Println(len(gzipFast(&b)))

    fmt.Fprint(w, gzipFast(&b))
    //fmt.Println(len(gzipSlow(b)))
    //gz := gzip.NewWriter(w)
    //defer gz.Close()
    //json.NewEncoder(gz).Encode(box)
    r.Body.Close()

}

但是当我取消注释时：

//gz := gzip.NewWriter(w)
//defer gz.Close()
//json.NewEncoder(gz).Encode(box)

它工作正常。

Answer 1

我认为问题出在fmt.Fprint(w, gzipFast(&b))的使用上。

如果你看看 gzipFast 的定义它 returns 一个 []byte。您正在将此切片放入打印函数中，即 "printing" 将所有内容放入 w。

如果你看一下io.Writer的定义：

type Writer interface {
        Write(p []byte) (n int, err error) }

您看到编写器可以处理 []byte 作为输入。

您应该使用 w.Write(gzipFast(&b)) 而不是 fmt.Fprint(w, gzipFast(&b))。那么你不需要取消注释：

//gz := gzip.NewWriter(w)
//defer gz.Close()
//json.NewEncoder(gz).Encode(box)

一切都是一个小例子，它显示了您的代码中发生的事情：

https://play.golang.org/p/6rzqLWTGiI

Answer 2

我会避免手动 gzip-ing []byte。您可以轻松地使用标准库中已有的编写器。另外，看看我认为应该使用的 compress/flate 而不是 gzip。

package main

import (
    "net/http"
    "encoding/json"
    "compress/gzip"
    "log"
)

type Box struct {
    Width int `json:"width"`
}

func writeJsonResponseCompressed(w http.ResponseWriter, r *http.Request) {

    box := &Box{Width: 10}

    w.Header().Set("Content-Type", "application/json")
    w.Header().Set("Content-Encoding", "gzip")

    body, err := json.Marshal(box)
    if err != nil {
        // Your error handling
        return
    }

    writer, err := gzip.NewWriterLevel(w, gzip.BestCompression)
    if err != nil {
        // Your error handling
        return
    }

    defer writer.Close()

    writer.Write(body)
}

func main() {
    http.HandleFunc("/compressedget", writeJsonResponseCompressed)
    log.Fatal(http.ListenAndServe(":8081", nil))
}

Answer 3

在访问底层字节之前，您需要刷新或关闭 gzip writer，例如

func gzipFast(a *[]byte) []byte {
    var b bytes.Buffer
    gz := gzip.NewWriter(&b)
    if _, err := gz.Write(*a); err != nil {
        gz.Close()
        panic(err)
    }
    gz.Close()
    return b.Bytes()
}

否则，在 gzip 编写器中缓冲但尚未写入最终流的内容不会被收集起来。

gzip 压缩到 http responseWriter

gzip compression to http responseWriter

rest

json

gzip

http

go