时间:2021-04-03 07:48:01 | 栏目:Golang | 点击:次
在GOPATH 中创建 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() //设置默认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中发送http请求的几种常见情况
整理一下golang中各种http的发送方式
最初的一些项目中,很多的地方都使用到了golang发送http请求,之后再对请求收到的结果进行一些处理,所用到的模式也比较固定,这里整理一下集中http的发送方式,先记录这么多,再一点一点添加。
最基本的场景
方式一 使用http.Newrequest
先生成http.client -> 再生成 http.request -> 之后提交请求:client.Do(request) -> 处理返回结果,每一步的过程都可以设置一些具体的参数,下面是一个最朴素最基本的例子:
//question ???将stdout重定向为response信息??? package main import ( "fmt" "io" "net/http" "os" ) func main() { //生成client 参数为默认 client := &http.Client{} //生成要访问的url url := "http://www.baidu.com" //提交请求 reqest, err := http.NewRequest("GET", url, nil) if err != nil { panic(err) } //处理返回结果 response, _ := client.Do(reqest) //将结果定位到标准输出 也可以直接打印出来 或者定位到其他地方进行相应的处理 stdout := os.Stdout _, err = io.Copy(stdout, response.Body) //返回的状态码 status := response.StatusCode fmt.Println(status) }
方式二 先生成client,之后用client.get/post..
client结构自己也有一些发送api的方法,比如client.get,client.post,client.postform..等等。基本上涵盖了主要的http请求的类型,通常不进行什么特殊的配置的话,这样就可以了,其实client的get或者post方法,也是对http.Newerequest方法的封装,里面还额外添加了req.Header.Set("Content-Type", bodyType)一般用的话,也是ok的
方式三 http. Get/Post..
具体实现的时候,还是采用的先前提到的模式,先生成一个默认的client,之后调用http.Newrequest方法。
对每个步骤进行细节性的说明
生成client时候的参数配置
最常见的一个参数是使用https的方式发送信息时候client端的设置。如果生成client的时候,什么信息都不添加,就会使用默认的值。具体的信息包括:
Transport RoundTripper CheckRedirect func(req *Request, via []*Request) error Jar CookieJar Timeout time.Duration
第一个参数是一个RoundTripper接口,里面包含了一个RoundTrip函数,指定了一些http请求的基本机制。
http.Transport中涉及到的参数较多,要是不指定的话,就会使用默认的DefaultTransport参数,里面包含一些默认的请求时间以及proxy机制之类的。
具体的细节参数涉及到好多,有的都没有使用到过比如那些我握手时间之类的,目前使用到的最多的就是https的相关参数:TLSClientConfig,这是一个*tls.Config类型,其中涉及到的参数还是有很多,一个基本的是用案例如下,仅仅是在配置中制定了rooca以及客户度端使用的证书。
通常发送https请求的时候,前面的参数可以使用如下方式进行处理:
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时候的参数配置
生成request的时候,主要的是几个基本的参数。NewRequest函数有三个基本的参数,NewRequest(method, urlStr string, body io.Reader)第一个是请求的类型,GET, POST, PUT, etc.要设成大写的形式。第二个参数是请求要访问的url,第三个参数是请求的body中的内容,需要是一个io.Reader的类型。
注意io.Reader的接口中是一个Read方法,实现了Read方法的类型应该都可以作为io.Reader来返回,Read(p []byte) (n int, err error)函数具体的功能就是读入len(p)长度的内容到p中,返回读入的长度以及错误信息。
通常是采用strings.NewReader函数,将一个string类型转化为io.Reader类型,或者bytes.NewBuffer函数,将[]byte类型转化为io.Reader类型。
此外还可以给request的header中添加一些额外的信息,比如下面例子中添加了请求的body的类型以及token的信息。
reqest.Header.Set("Content-Type", "application/x-www-form-urlencoded")
reqest.Header.Set("Authorization", "qwertyuiopasdfghjklzxcvbnm1234567890")
还有比如模拟表单提交,可以把提交的类型设置为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))
最常见的一种情况是发送一个json文件过去,可以把Header的类型设置成为:
"Content-Type", "application/json; charset=utf-8"
其余的部分按照先前同样的方式进行设置发送提交就好。
request的类型的属性还是比较多的,慢慢整理。
生成的response结果的处理
一般在client构建好之后,要采用client.Do(request)方法提交client请求,之后会返回一个*Response类型。response中的参数一般也比较多,我们需要的最多的通常是Body参数,一般通过body, _ := ioutil.ReadAll(resp.Body)会把body转化为[]byte类型返回过来, 之后再进行其他的处理。