快捷導(dǎo)航

Golang語(yǔ)言如何讀取http.Request中body的內(nèi)容

更新時(shí)間：2024年03月28日 10:14:37 作者：許野平

這篇文章主要介紹了Golang語(yǔ)言如何讀取http.Request中body的內(nèi)容問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

Golang讀取http.Request中body內(nèi)容

不羅嗦了，直接貼代碼，不曉得為什么搞這么復(fù)雜，是不是因?yàn)閾?dān)心 body 內(nèi)容一次接受不全，所以搞了個(gè)接口來(lái)讀取其中的內(nèi)容？

import (
	...
	"io/ioutil"
	...
)

...

func myPost(w http.ResponseWriter, r *http.Request) {
	s, _ := ioutil.ReadAll(r.Body) //把	body 內(nèi)容讀入字符串 s
	fmt.Fprintf(w, "%s", s)        //在返回頁(yè)面中顯示內(nèi)容。
}

...

Golang http.Request復(fù)用

針對(duì)除了Get以外的請(qǐng)求

package main

import (
    "net/http"
    "strings"
    )

func main(){
        reader := strings.NewReader("hello")
        req,_ := http.NewRequest("POST","http://www.abc.com",reader)
        client := http.Client{}
        client.Do(req) // 第一次會(huì)請(qǐng)求成功
        client.Do(req) // 請(qǐng)求失敗
}

第二次請(qǐng)求會(huì)出錯(cuò)

http: ContentLength=5 with Body length 0

原因是第一次請(qǐng)求后req.Body已經(jīng)讀取到結(jié)束位置，所以第二次請(qǐng)求時(shí)無(wú)法再讀取body，

解決方法：

重新定義一個(gè)ReadCloser的實(shí)現(xiàn)類替換req.Body

package reader

import (
    "io"
    "net/http"
    "strings"
    "sync/atomic"
)

type Repeat struct{
    reader io.ReaderAt
    offset int64
}

// Read 重寫讀方法，使每次讀取request.Body時(shí)能從指定位置讀取
func (p *Repeat) Read(val []byte) (n int, err error) {
    n, err = p.reader.ReadAt(val, p.offset)
    atomic.AddInt64(&p.offset, int64(n))
    return
}

// Reset 重置偏移量
func (p *Repeat) Reset(){
        atomic.StoreInt64(&p.offset,0)
}

func (p *Repeat) Close() error {
    // 因?yàn)閞eq.Body實(shí)現(xiàn)了readcloser接口，所以要實(shí)現(xiàn)close方法
    // 但是repeat中的值有可能是只讀的，所以這里只是嘗試關(guān)閉一下。
    if p.reader != nil {
            if rc, ok := p.reader.(io.Closer); ok {
                return rc.Close()
            }
        }
    return nil
}

func doPost()  {
    client := http.Client{}
    reader := strings.NewReader("hello")
    req , _ := http.NewRequest("POST","http://www.abc.com",reader)
    req.Body = &Repeat{reader:reader,offset:0}
    client.Do(req)
    // 將偏移量重置為0
    req.Body.(*Repeat).Reset()
    client.Do(req)
}

這樣就不會(huì)報(bào)錯(cuò)了，因?yàn)橐仓貙懥薈lose()方法，所以同時(shí)也解決了request重用時(shí)，req.Body自動(dòng)關(guān)閉的問(wèn)題。