golang http请求封装代码

时间:2022-09-11 19:25:38

在GOPATH 中创建 utils 文件夹 放置这两个文件

http.go

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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) -> 处理返回结果,每一步的过程都可以设置一些具体的参数,下面是一个最朴素最基本的例子:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
//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的时候,什么信息都不添加,就会使用默认的值。具体的信息包括:

?
1
2
3
4
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请求的时候,前面的参数可以使用如下方式进行处理:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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:

?
1
2
3
4
5
6
7
8
9
10
11
// 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类型返回过来, 之后再进行其他的处理。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持服务器之家。如有错误或未考虑完全的地方,望不吝赐教。

原文链接:https://xiaochuan.blog.csdn.net/article/details/81539820