golang http请求封装代码
作者：一名路过的小码农
在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类型返回过来, 之后再进行其他的处理。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。