golang http請求封裝代碼
在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é)果定位到標準輸出 也可以直接打印出來 或者定位到其他地方進行相應的處理
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方法的類型應該都可以作為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類型返回過來, 之后再進行其他的處理。
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
最近在做性能優(yōu)化,有個函數(shù)里面的耗時特別長,看里面的操作大多是一些字符串拼接的操作,而字符串拼接在 golang 里面其實有很多種實現(xiàn),下面這篇文章主要給大家介紹了關(guān)于Golang語言如何高效拼接字符串的相關(guān)資料,需要的朋友可以參考下2023-02-02
在golang中，nil只能賦值給指針、channel、func、interface、map或slice類型的變量。如果未遵循這個規(guī)則，則會引發(fā)panic。2017-08-08
這篇文章主要介紹了Golang Map實現(xiàn)賦值和擴容的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-04-04
這篇文章主要介紹了Golang設(shè)計模式之外觀模式的實現(xiàn)，外觀模式是一種常用的設(shè)計模式之一，是一種結(jié)構(gòu)型設(shè)計模式，它提供了一個簡單的接口來訪問復雜系統(tǒng)的各種功能，從而降低了系統(tǒng)的復雜度，需要詳細了解可以參考下文2023-05-05
這篇文章主要介紹了goland中導包報紅和go mod問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-03-03
這篇文章主要介紹了golang操作mongodb的方法,涉及Go語言操作mongodb的連接、讀取及顯示的方法,具有一定參考借鑒價值,需要的朋友可以參考下2015-02-02
Go語言并發(fā)編程之控制并發(fā)數(shù)量實現(xiàn)實例
這篇文章主要為大家介紹了Go語言并發(fā)編程之控制并發(fā)數(shù)量實例探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01
Mutex是Golang常見的并發(fā)原語，在開發(fā)過程中經(jīng)常使用到，本文主要介紹了Golang Mutex實現(xiàn)互斥的具體方法，具有一定的參考價值，感興趣的可以了解一下2023-04-04
這篇文章主要為大家介紹了Go語言建議多使用切片少使用數(shù)組原理探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01
這篇文章主要介紹了golang rate令牌桶源碼分析實現(xiàn)方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-07-07