我想撰寫一個 api，它將通過 POST 發送 gzipped json 資料。雖然下面可以處理正文中的簡單 json，但如果 json 被 gzip 壓縮，則不會處理。
我們需要在使用前顯式處理解壓c.ShouldBindJSON嗎？
如何繁殖
package main
import (
"github.com/gin-gonic/gin"
"log"
"net/http"
)
func main() {
r := gin.Default()
r.POST("/postgzip", func(c *gin.Context) {
type PostData struct {
Data string `binding:"required" json:"data"`
}
var postdata PostData
if err := c.ShouldBindJSON(&postdata); err != nil {
log.Println("Error parsing request body", "error", err)
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
log.Printf("%s", postdata)
if !c.IsAborted() {
c.String(200, postdata.Data)
}
})
r.Run()
}
? echo '{"data" : "hello"}' | curl -X POST -H "Content-Type: application/json" -d @- localhost:8080/postgzip
hello
期望
$ echo '{"data" : "hello"}' | gzip | curl -v -i -X POST -H "Content-Type: application/json" -H "Content-Encoding: gzip" --data-binary @- localhost:8080/postgzip
hello
實際結果
$ echo '{"data" : "hello"}' | gzip | curl -v -i -X POST -H "Content-Type: application/json" -H "Content-Encoding: gzip" --data-binary @- localhost:8080/postgzip
{"error":"invalid character '\\x1f' looking for beginning of value"}
環境
- 去版本：
go version go1.17.2 darwin/amd64 - gin 版本（或提交參考）：
v1.7.4 - 作業系統：MacOS Monterey
uj5u.com熱心網友回復：
我們是否需要在使用 c.ShouldBindJSON 之前顯式處理解壓？
當然。Gin 不ShouldBindJSON知道您的有效載荷可能會或可能不會被編碼。正如方法名稱所暗示的那樣，它需要 JSON input。
如果您希望撰寫可重用的代碼，則可以實作該Binding介面。
一個非常小的例子：
type GzipJSONBinding struct {
}
func (b *GzipJSONBinding) Name() string {
return "gzipjson"
}
func (b *GzipJSONBinding) Bind(req *http.Request, dst interface{}) error {
r, err := gzip.NewReader(req.Body)
if err != nil {
return err
}
raw, err := io.ReadAll(r)
if err != nil {
return err
}
return json.Unmarshal(raw, dst)
}
然后可用 with c.ShouldBindWith，它允許使用任意系結引擎：
err := c.ShouldBindWith(&postData, &GzipJSONBinding{})
uj5u.com熱心網友回復：
打開的完整作業示例 Content-Encoding
curl 嘗試使用它
$ echo '{"data" : "hello"}' | gzip | curl -X POST -H "Content-Type: application/json" -H "Content-Encoding: gzip" --data-binary @- localhost:8080/json
hello
$ curl -X POST -H "Content-Type: application/json" --data-raw '{"data" : "hello"}' localhost:8080/json
hello
package main
import (
"bytes"
"compress/gzip"
"encoding/json"
"fmt"
"github.com/gin-gonic/gin"
"github.com/gin-gonic/gin/binding"
"io"
"log"
"net/http"
)
type PostData struct {
Data string `binding:"required" json:"data"`
}
func main() {
r := gin.Default()
r.POST("/json", func(c *gin.Context) {
var postdata PostData
contentEncodingHeader := c.GetHeader("Content-Encoding")
switch contentEncodingHeader {
case "gzip":
if err := c.ShouldBindBodyWith(&postdata, gzipJSONBinding{}); err != nil {
log.Println("Error parsing GZIP JSON request body", "error", err)
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
case "":
if err := c.ShouldBindJSON(&postdata); err != nil {
log.Println("Error parsing JSON request body", "error", err)
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
default:
log.Println("unsupported Content-Encoding")
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "unsupported Content-Encoding"})
return
}
log.Printf("%s", postdata)
if !c.IsAborted() {
c.String(200, postdata.Data)
}
})
r.Run()
}
type gzipJSONBinding struct{}
func (gzipJSONBinding) Name() string {
return "gzipjson"
}
func (gzipJSONBinding) Bind(req *http.Request, obj interface{}) error {
if req == nil || req.Body == nil {
return fmt.Errorf("invalid request")
}
r, err := gzip.NewReader(req.Body)
if err != nil {
return err
}
raw, err := io.ReadAll(r)
if err != nil {
return err
}
return json.Unmarshal(raw, obj)
}
func (gzipJSONBinding) BindBody(body []byte, obj interface{}) error {
r, err := gzip.NewReader(bytes.NewReader(body))
if err != nil {
return err
}
return decodeJSON(r, obj)
}
func decodeJSON(r io.Reader, obj interface{}) error {
decoder := json.NewDecoder(r)
if err := decoder.Decode(obj); err != nil {
return err
}
return validate(obj)
}
func validate(obj interface{}) error {
if binding.Validator == nil {
return nil
}
return binding.Validator.ValidateStruct(obj)
}
轉載請註明出處，本文鏈接：https://www.uj5u.com/net/351608.html