用golang進行http請求類型多了,總結備忘一下。golang
1.普通的post\get請求web
var r http.Request r.ParseForm() r.Form.Add("uuid", orderUUID) bodystr := strings.TrimSpace(r.Form.Encode()) request, err := http.NewRequest("GET", url, strings.NewReader(bodystr)) if err != nil { //TODO: } request.Header.Set("Content-Type", "application/x-www-form-urlencoded") request.Header.Set("Connection", "Keep-Alive") var resp *http.Response resp, err = http.DefaultClient.Do(request) if err != nil { //TODO: }
2. body所有二進制數據流進行postapp
//body提交二進制數據 func DoBytesPost(url string, data []byte) ([]byte, error) { body := bytes.NewReader(data) request, err := http.NewRequest(POST_METHOD, url, body) if err != nil { log.Println("http.NewRequest,[err=%s][url=%s]", err, url) return []byte(""), err } request.Header.Set("Connection", "Keep-Alive") var resp *http.Response resp, err = http.DefaultClient.Do(request) if err != nil { log.Println("http.Do failed,[err=%s][url=%s]", err, url) return []byte(""), err } defer resp.Body.Close() b, err := ioutil.ReadAll(resp.Body) if err != nil { log.Println("http.Do failed,[err=%s][url=%s]", err, url) } return b, err }
3. 模擬web表單文件上傳進行postsocket
func newfileUploadRequest(uri string, params map[string]string, paramName, path string) (*http.Request, error) { file, err := os.Open(path) if err != nil { return nil, err } defer file.Close() body := &bytes.Buffer{} writer := multipart.NewWriter(body) part, err := writer.CreateFormFile(paramName, path) if err != nil { return nil, err }
# 這裏的io.Copy實現,會把file文件都讀取到內存裏面,而後當作一個buffer傳給NewRequest. 對於大文件來講會佔用不少內存 _, err = io.Copy(part, file) for key, val := range params { _ = writer.WriteField(key, val) } err = writer.Close() if err != nil { return nil, err } request, err := http.NewRequest("POST", uri, body) request.Header.Set("Content-Type", writer.FormDataContentType()) return request, err }
下面這種方式,就是傳入文件句柄,而後由request copy到socket的緩衝區,而不用所有讀取到內存中:post
bodyBuf := bytes.NewBufferString("") bodyWriter := multipart.NewWriter(bodyBuf) //創建文件的http第一部分數據,文件信息 _, err := bodyWriter.CreateFormFile(paramName, path) if err != nil { return nil, err } //讀取文件,當作http第二部分數據 fh, err := os.Open(path) if err != nil { return nil, err } //mulitipart/form-data時,須要獲取本身關閉的boundary boundary := bodyWriter.Boundary() closeBuf := bytes.NewBufferString(fmt.Sprintf("\r\n--%s--\r\n", boundary)) //創建寫入socket的reader對象 requestReader := io.MultiReader(bodyBuf, fh, closeBuf) fi, err := fh.Stat() if err != nil { return nil, err } req, err := http.NewRequest("POST", uri, requestReader) if err != nil { return nil, err } //設置http頭 req.Header.Add("Content-Type", "multipart/form-data; boundary="+boundary) req.ContentLength = fi.Size() + int64(bodyBuf.Len()) + int64(closeBuf.Len())
go package 的實現源碼:ui
http://golang.org/src/io/io.go?s=12247:12307#L340google
https://golang.org/src/io/multi.go?h=MultiReader#L31url
https://golang.org/src/net/http/transfer.go?h=WriteBody#L199spa
https://golang.org/src/net/http/request.go?h=write#L366code
參考鏈接:
https://groups.google.com/forum/#!topic/golang-nuts/Zjg5l4nKcQ0