快捷導(dǎo)航

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

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

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

前言

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

問(wèn)題分析

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

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

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)
    }
}

按照官方文檔推薦，在過(guò)濾器我使用了 httpx.Parse(r, &req) 獲取請(qǐng)求體數(shù)據(jù)，導(dǎo)致后續(xù) post 請(qǐng)求邏輯讀取請(qǐng)求體失敗，也就是說(shuō)，這個(gè)方法中會(huì)消耗掉請(qǐng)求體流的數(shù)據(jù)，查看源碼可以發(fā)現(xiàn)，這個(gè)方法是會(huì)讀取請(qǐng)求中的 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 方法就會(huì)消耗掉請(qǐng)求體流的數(shù)據(jù)，ParsePath 方法 ParseHeaders 方法則不會(huì)，而在這個(gè)過(guò)濾器中，我們只需要獲取 header 的數(shù)據(jù)，body 的數(shù)據(jù)需要由后續(xù)邏輯獲取，所以在這里需要修改過(guò)濾只獲取 header 數(shù)據(jù)的方法，如果需要的是 path 數(shù)據(jù)也是同樣的道理，修改后的過(guò)濾器：

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)即可正常獲取請(qǐng)求體數(shù)據(jù)。

完整示例代碼請(qǐng)參考：igolang

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

您可能感興趣的文章: