当前位置 博文首页 > golang http请求封装代码

    golang http请求封装代码

    作者:一名路过的小码农 时间:2021-02-09 09:22

    在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类型返回过来, 之后再进行其他的处理。

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

    js
    下一篇:没有了