使用golang語言怎么對http請求進行封裝

這篇文章給大家介紹使用golang語言怎么對http請求進行封裝，內(nèi)容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

在GOPATH 中創(chuàng)建 utils 文件夾 放置這兩個文件

http.go

package utils 
import (
 "crypto/tls"
 "encoding/json"
 "errors"
 "fmt"
 "io/ioutil"
 "net/http"
 "net/url"
 "strings"
 "sync"
) 
var (
 GET_METHOD = "GET"
 POST_METHOD = "POST"
 SENDTYPE_FROM = "from"
 SENDTYPE_JSON = "json"
) 
type HttpSend struct {
 Link  string
 SendType string
 Header map[string]string
 Body  map[string]string
 sync.RWMutex
} 
func NewHttpSend(link string) *HttpSend {
 return &HttpSend{
  Link:  link,
  SendType: SENDTYPE_FROM,
 }
} 
func (h *HttpSend) SetBody(body map[string]string) {
 h.Lock()
 defer h.Unlock()
 h.Body = body
} 
func (h *HttpSend) SetHeader(header map[string]string) {
 h.Lock()
 defer h.Unlock()
 h.Header = header
} 
func (h *HttpSend) SetSendType(send_type string) {
 h.Lock()
 defer h.Unlock()
 h.SendType = send_type
} 
func (h *HttpSend) Get() ([]byte, error) {
 return h.send(GET_METHOD)
} 
func (h *HttpSend) Post() ([]byte, error) {
 return h.send(POST_METHOD)
} 
func GetUrlBuild(link string, data map[string]string) string {
 u, _ := url.Parse(link)
 q := u.Query()
 for k, v := range data {
  q.Set(k, v)
 }
 u.RawQuery = q.Encode()
 return u.String()
} 
func (h *HttpSend) send(method string) ([]byte, error) {
 var (
  req  *http.Request
  resp  *http.Response
  client http.Client
  send_data string
  err  error
 ) 
 if len(h.Body) > 0 {
  if strings.ToLower(h.SendType) == SENDTYPE_JSON {
   send_body, json_err := json.Marshal(h.Body)
   if json_err != nil {
    return nil, json_err
   }
   send_data = string(send_body)
  } else {
   send_body := http.Request{}
   send_body.ParseForm()
   for k, v := range h.Body {
    send_body.Form.Add(k, v)
   }
   send_data = send_body.Form.Encode()
  }
 } 
 //忽略https的證書
 client.Transport = &http.Transport{
  TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
 } 
 req, err = http.NewRequest(method, h.Link, strings.NewReader(send_data))
 if err != nil {
  return nil, err
 }
 defer req.Body.Close()
 
 //設(shè)置默認header
 if len(h.Header) == 0 {
  //json
  if strings.ToLower(h.SendType) == SENDTYPE_JSON {
   h.Header = map[string]string{
    "Content-Type": "application/json; charset=utf-8",
   }
  } else { //form
   h.Header = map[string]string{
    "Content-Type": "application/x-www-form-urlencoded",
   }
  }
 } 
 for k, v := range h.Header {
  if strings.ToLower(k) == "host" {
   req.Host = v
  } else {
   req.Header.Add(k, v)
  }
 } 
 resp, err = client.Do(req)
 if err != nil {
  return nil, err
 }
 defer resp.Body.Close()
 
 if resp.StatusCode != http.StatusOK {
  return nil, errors.New(fmt.Sprintf("error http code :%d", resp.StatusCode))
 } 
 return ioutil.ReadAll(resp.Body)
}

http_test.go

package utils 
import (
 "testing"
) 
func Test_Get(t *testing.T) {
 h := NewHttpSend(GetUrlBuild("http://127.0.0.1/test.php", map[string]string{"name": "xiaochuan"}))
 _, err := h.Get()
 if err != nil {
 t.Error("請求錯誤:", err)
 } else {
 t.Log("正常返回")
 }
} 
func Test_Post(t *testing.T) {
 h := NewHttpSend("http://127.0.0.1/test.php")
 h.SetBody(map[string]string{"name": "xiaochuan"})
 _, err := h.Post()
 if err != nil {
 t.Error("請求錯誤:", err)
 } else {
 t.Log("正常返回")
 }
} 
func Test_Json(t *testing.T) {
 h := NewHttpSend("http://127.0.0.1/test.php")
 h.SetSendType("JSON")
 h.SetBody(map[string]string{"name": "xiaochuan"})
 _, err := h.Post()
 if err != nil {
 t.Error("請求錯誤:", err)
 } else {
 t.Log("正常返回")
 }
} 
func Benchmark_GET(b *testing.B) {
 for i := 0; i < b.N; i++ {
 h := NewHttpSend(GetUrlBuild("http://127.0.0.1/test.php", map[string]string{"name": "xiaochuan"}))
 _, err := h.Get()
 if err != nil {
 b.Error("請求錯誤:", err)
 } else {
 b.Log("正常返回")
 }
 }
}

補充：golang中發(fā)送http請求的幾種常見情況

整理一下golang中各種http的發(fā)送方式

最初的一些項目中，很多的地方都使用到了golang發(fā)送http請求，之后再對請求收到的結(jié)果進行一些處理，所用到的模式也比較固定，這里整理一下集中http的發(fā)送方式，先記錄這么多，再一點一點添加。

最基本的場景

方式一 使用http.Newrequest

先生成http.client -> 再生成 http.request -> 之后提交請求：client.Do(request) -> 處理返回結(jié)果，每一步的過程都可以設(shè)置一些具體的參數(shù)，下面是一個最樸素最基本的例子：

//question ???將stdout重定向為response信息？？？
package main 
import (
 "fmt"
 "io"
 "net/http"
 "os"
) 
func main() {
 //生成client 參數(shù)為默認
 client := &http.Client{} 
 //生成要訪問的url
 url := "http://www.baidu.com"  
 //提交請求
 reqest, err := http.NewRequest("GET", url, nil) 
 if err != nil {
 panic(err)
 } 
 //處理返回結(jié)果
 response, _ := client.Do(reqest) 
 //將結(jié)果定位到標(biāo)準(zhǔn)輸出 也可以直接打印出來 或者定位到其他地方進行相應(yīng)的處理
 stdout := os.Stdout
 _, err = io.Copy(stdout, response.Body) 
 //返回的狀態(tài)碼
 status := response.StatusCode 
 fmt.Println(status) 
}

方式二 先生成client，之后用client.get/post..

client結(jié)構(gòu)自己也有一些發(fā)送api的方法，比如client.get,client.post,client.postform..等等?；旧虾w了主要的http請求的類型，通常不進行什么特殊的配置的話，這樣就可以了，其實client的get或者post方法，也是對http.Newerequest方法的封裝，里面還額外添加了req.Header.Set("Content-Type", bodyType)一般用的話，也是ok的

方式三 http. Get/Post..

具體實現(xiàn)的時候，還是采用的先前提到的模式，先生成一個默認的client，之后調(diào)用http.Newrequest方法。

對每個步驟進行細節(jié)性的說明

生成client時候的參數(shù)配置

最常見的一個參數(shù)是使用https的方式發(fā)送信息時候client端的設(shè)置。如果生成client的時候，什么信息都不添加，就會使用默認的值。具體的信息包括：

 Transport RoundTripper
 CheckRedirect func(req *Request, via []*Request) error
 Jar CookieJar
 Timeout time.Duration

第一個參數(shù)是一個RoundTripper接口，里面包含了一個RoundTrip函數(shù)，指定了一些http請求的基本機制。

http.Transport中涉及到的參數(shù)較多，要是不指定的話，就會使用默認的DefaultTransport參數(shù)，里面包含一些默認的請求時間以及proxy機制之類的。

具體的細節(jié)參數(shù)涉及到好多，有的都沒有使用到過比如那些我握手時間之類的，目前使用到的最多的就是https的相關(guān)參數(shù)：TLSClientConfig，這是一個*tls.Config類型，其中涉及到的參數(shù)還是有很多，一個基本的是用案例如下，僅僅是在配置中制定了rooca以及客戶度端使用的證書。

通常發(fā)送https請求的時候，前面的參數(shù)可以使用如下方式進行處理：

pool := x509.NewCertPool()
 caCertPath := "certs/cert_server/ca.crt" 
 caCrt, err := ioutil.ReadFile(caCertPath)
 if err != nil {
 fmt.Println("ReadFile err:", err)
 return
 }
 pool.AppendCertsFromPEM(caCrt) 
 cliCrt, err := tls.LoadX509KeyPair("certs/cert_server/client.crt", "certs/cert_server/client.key")
 if err != nil {
 fmt.Println("Loadx509keypair err:", err)
 return
 } 
 tr := &http.Transport{
 TLSClientConfig: &tls.Config{
 RootCAs:  pool,
 Certificates: []tls.Certificate{cliCrt},
 },
 }
 client := &http.Client{Transport: tr}

生成request時候的參數(shù)配置

生成request的時候，主要的是幾個基本的參數(shù)。NewRequest函數(shù)有三個基本的參數(shù)，NewRequest(method, urlStr string, body io.Reader)第一個是請求的類型，GET, POST, PUT, etc.要設(shè)成大寫的形式。第二個參數(shù)是請求要訪問的url，第三個參數(shù)是請求的body中的內(nèi)容，需要是一個io.Reader的類型。

注意io.Reader的接口中是一個Read方法，實現(xiàn)了Read方法的類型應(yīng)該都可以作為io.Reader來返回,Read(p []byte) (n int, err error)函數(shù)具體的功能就是讀入len(p)長度的內(nèi)容到p中，返回讀入的長度以及錯誤信息。

通常是采用strings.NewReader函數(shù)，將一個string類型轉(zhuǎn)化為io.Reader類型，或者bytes.NewBuffer函數(shù)，將[]byte類型轉(zhuǎn)化為io.Reader類型。

此外還可以給request的header中添加一些額外的信息，比如下面例子中添加了請求的body的類型以及token的信息。

reqest.Header.Set("Content-Type", "application/x-www-form-urlencoded")

reqest.Header.Set("Authorization", "qwertyuiopasdfghjklzxcvbnm1234567890")

還有比如模擬表單提交，可以把提交的類型設(shè)置為url.Values類型再進行Encode：

// use map as struct
 var clusterinfo = url.Values{}
 //var clusterinfo = map[string]string{}
 clusterinfo.Add("userName", user)
 clusterinfo.Add("password", pw)
 clusterinfo.Add("cloudName", clustername)
 clusterinfo.Add("masterIp", masterip)
 clusterinfo.Add("cacrt", string(caCrt)) 
 data := clusterinfo.Encode() 
 url := "https://10.10.105.124:8443/user/checkAndUpdate"
 reqest, err := http.NewRequest("POST", url, strings.NewReader(data))

最常見的一種情況是發(fā)送一個json文件過去，可以把Header的類型設(shè)置成為：

"Content-Type", "application/json; charset=utf-8"

其余的部分按照先前同樣的方式進行設(shè)置發(fā)送提交就好。

request的類型的屬性還是比較多的，慢慢整理。

生成的response結(jié)果的處理

一般在client構(gòu)建好之后，要采用client.Do(request)方法提交client請求，之后會返回一個*Response類型。response中的參數(shù)一般也比較多，我們需要的最多的通常是Body參數(shù)，一般通過body, _ := ioutil.ReadAll(resp.Body)會把body轉(zhuǎn)化為[]byte類型返回過來, 之后再進行其他的處理。

關(guān)于使用golang語言怎么對http請求進行封裝就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學(xué)到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。