快捷導(dǎo)航

go-zero讀取請求體出現(xiàn)EOF錯(cuò)誤的解決方法

更新時(shí)間：2025年02月05日 08:32:54 作者：PandaMiner

這篇文章主要為大家詳細(xì)介紹了go-zero讀取請求體出現(xiàn)EOF錯(cuò)誤時(shí)如何解決,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

前言

最近自己在搭 go-zero 的腳手架，遇到一個(gè)問題，原先的一個(gè) post 請求是執(zhí)行成功，當(dāng)我添加了一個(gè)過濾器之后執(zhí)行該請求就會報(bào)錯(cuò)請求體讀取 EOF 錯(cuò)誤，我感到有點(diǎn)奇怪，為什么過濾器中的邏輯會影響到后續(xù)的請求呢？在此記錄下這次的坑。

問題分析

先上結(jié)論，這是由于 go 中 http.Request.Body 請求體是一個(gè)流，它只能被讀取一次，假如你在過濾器已經(jīng)進(jìn)行過流的讀取，那么在后續(xù)的請求中就讀取不到了。

示例的請求邏輯是這樣，當(dāng)請求到達(dá)后臺時(shí)，先經(jīng)過過濾器檢查請求頭中是否攜帶 token，否則則請求失敗，成功則繼續(xù) post 請求邏輯，本次的 post 請求和過濾器只是做示例使用，邏輯很簡單，過濾器中的邏輯就是去獲取請求 header 中 token 字段，如果這個(gè)字段為空，則請求失敗，代碼示例如下：

func tokenFilter(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
       type Request struct {
          Token string `header:"token"`
       }
       var req Request
       err := httpx.Parse(r, &req)
       if err != nil || req.Token == "" {
          resp, _ := utils.ToJsonString(models.ErrorResponseWithCode(http.StatusUnauthorized, "token required"))
          routers.HttpError(w, http.StatusUnauthorized, resp)
          return
       }
       next(w, r)
    }
}

按照官方文檔推薦，在過濾器我使用了 httpx.Parse(r, &req) 獲取請求體數(shù)據(jù)，導(dǎo)致后續(xù) post 請求邏輯讀取請求體失敗，也就是說，這個(gè)方法中會消耗掉請求體流的數(shù)據(jù)，查看源碼可以發(fā)現(xiàn)，這個(gè)方法是會讀取請求中的 path，form，header，body 數(shù)據(jù)，賦值到我們的 struct 中：

func Parse(r *http.Request, v any) error {
    kind := mapping.Deref(reflect.TypeOf(v)).Kind()
    if kind != reflect.Array && kind != reflect.Slice {
       if err := ParsePath(r, v); err != nil {
          return err
       }

       if err := ParseForm(r, v); err != nil {
          return err
       }

       if err := ParseHeaders(r, v); err != nil {
          return err
       }
    }

    if err := ParseJsonBody(r, v); err != nil {
       return err
    }

    if valid, ok := v.(validation.Validator); ok {
       return valid.Validate()
    } else if val := validator.Load(); val != nil {
       return val.(Validator).Validate(r, v)
    }

    return nil
}

繼續(xù)往下查看源碼可以發(fā)現(xiàn)，其中的 ParseForm 方法和 ParseJsonBody 方法就會消耗掉請求體流的數(shù)據(jù)，ParsePath 方法 ParseHeaders 方法則不會，而在這個(gè)過濾器中，我們只需要獲取 header 的數(shù)據(jù)，body 的數(shù)據(jù)需要由后續(xù)邏輯獲取，所以在這里需要修改過濾只獲取 header 數(shù)據(jù)的方法，如果需要的是 path 數(shù)據(jù)也是同樣的道理，修改后的過濾器：

func tokenFilter(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
       type Request struct {
          Token string `header:"token"`
       }
       var req Request
       err := httpx.ParseHeaders(r, &req)
       if err != nil || req.Token == "" {
          resp, _ := utils.ToJsonString(models.ErrorResponseWithCode(http.StatusUnauthorized, "token required"))
          routers.HttpError(w, http.StatusUnauthorized, resp)
          return
       }
       next(w, r)
    }
}

修改后后續(xù)節(jié)點(diǎn)即可正常獲取請求體數(shù)據(jù)。

完整示例代碼請參考：igolang

到此這篇關(guān)于go-zero讀取請求體出現(xiàn)EOF錯(cuò)誤的解決方法的文章就介紹到這了,更多相關(guān)解決go-zero讀取請求體出現(xiàn)EOF錯(cuò)誤內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: