Golang 自己封装下http get和post

封装

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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
package helper
 
import (
    "encoding/json"
    "io"
    "io/ioutil"
    "net/http"
    "net/url"
    "strings"
)
 
type JsonHttpClientHelper struct {
    client *http.Client
}
 
func NewHttpClientHelper() *JsonHttpClientHelper {
    return &JsonHttpClientHelper{
        client: &http.Client{},
    }
}
 
func NewHttpClientHelperFromClient(client *http.Client) *JsonHttpClientHelper {
    return &JsonHttpClientHelper{
        client: client,
    }
}
 
// Get
//  @Description: 发起Get请求
//  @receiver clientHelper
//  @param urlPath url请求绝对路径 https://www.***.com/users
//  @param headers http请求头
//  @param params  url中附带的参数信息,最后会拼接到绝对路径上 如: https://www.***.com/users?name=u&age=18
//  @return string 请求结果的字符串,推荐为 json格式
//  @return error  请求发生的错误,需要捕获
//
func (clientHelper *JsonHttpClientHelper) Get(urlPath string, headers, params map[string]string) (string, error) {
    //init url params
    requestUrl, err := makeRequestUrl(urlPath, params)
    if err != nil {
        return "", err
    }
 
    req, err := makeRequest(requestUrl, headers, "GET", nil)
    if err != nil {
        return "", err
    }
 
    response, err := clientHelper.client.Do(req)
    if err != nil {
        return "", err
    }
    defer response.Body.Close()
 
    //read from response
    b, err := ioutil.ReadAll(response.Body)
    if err != nil {
        return "", err
    }
 
    return string(b), nil
}
 
func makeRequest(requestUrl *url.URL, headers map[string]string, method string, body io.Reader) (*http.Request, error) {
    req, err := http.NewRequest(method, requestUrl.String(), body)
    if err != nil {
        return nil, err
    }
 
    //set header
    for key, value := range headers {
        req.Header.Add(key, value)
    }
 
    return req, nil
}
 
func makeRequestUrl(urlPath string, params map[string]string) (*url.URL, error) {
    urlParams := url.Values{}
    for key, value := range params {
        urlParams.Set(key, value)
    }
    requestURI, err := url.ParseRequestURI(urlPath)
    if err != nil {
        return nil, err
    }
    requestURI.RawQuery = urlParams.Encode()
 
    return requestURI, nil
}
 
// Post
//  @Description: 发起POST请求
//  @receiver clientHelper
//  @param urlPath url请求绝对路径 https://www.***.com/users
//  @param data 请求时需要传输的body
//  @param headers http请求头 Content-Type只能为 application/x-www-form-urlencoded 或 application/json
//  @param urlParams  url中附带的参数信息,最后会拼接到绝对路径上 如: https://www.***.com/users?name=u&age=18
//  @param bodyParams  Content-Type为application/x-www-form-urlencoded 特殊片段
//  @return string 请求结果的字符串,推荐为 json格式
//  @return error  请求发生的错误,需要捕获
//
func (clientHelper *JsonHttpClientHelper) Post(urlPath string, data interface{}, headers, urlParams, bodyParams map[string]string) (string, error) {
    //init url urlParams
    requestUrl, err := makeRequestUrl(urlPath, urlParams)
    if err != nil {
        return "", err
    }
 
    var req *http.Request
    if isJsonHttp(headers) {
        bodyJsonStr, err := convertDataToJson(data)
        if err != nil {
            return "", err
        }
        req, err = makeRequest(requestUrl, headers, "POST", strings.NewReader(bodyJsonStr))
    } else {
        bodyUrlValues := makeRequestBodyForFormUpload(bodyParams)
        req, err = makeRequest(requestUrl, headers, "POST", strings.NewReader(bodyUrlValues.Encode()))
        if err != nil {
            return "", err
        }
    }
 
    response, err := clientHelper.client.Do(req)
    if err != nil {
        return "", err
    }
    defer response.Body.Close()
 
    //read from response
    b, err := ioutil.ReadAll(response.Body)
    if err != nil {
        return "", err
    }
 
    return string(b), nil
}
 
func makeRequestBodyForFormUpload(maps map[string]string) *url.Values {
    values := url.Values{}
    if maps != nil {
        for key, value := range maps {
            values.Set(key, value)
        }
    }
 
    return &values
}
 
func isJsonHttp(headers map[string]string) bool {
    v, ok := headers["Content-Type"]
    if ok && v == "application/json" {
        return true
    }
 
    if !ok {
        return true
    }
 
    return false
}
 
func convertDataToJson(data interface{}) (string, error) {
    bodyJsonByte, err := json.Marshal(data)
    if err != nil {
        return "", err
    }
 
    bodyJsonStr := string(bodyJsonByte)
    if bodyJsonStr == "" || bodyJsonStr == "null" || bodyJsonStr == "NULL" || bodyJsonStr == "nil" {
        bodyJsonStr = "{}"
    }
    return bodyJsonStr, nil
}

  调用demo

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
package test
 
import (
   "fmt"
   "hello/src/helper"
   "testing"
)
 
func TestHttpClient(t *testing.T) {
   params := make(map[string]string, 10)
   params["number"] = "99999"
   resultJson, err := helper.NewHttpClientHelper().Post("http://127.0.0.1:6789/ping/call", nil, nil, params,nil)
   if err != nil {
      //
      fmt.Println(err.Error())
   } else {
      fmt.Println(resultJson)
   }
}

  

  这样不需要自己每次都填充 httpHeader httpBody之类的

posted @   FCmmmmmm  阅读(449)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 【.NET】调用本地 Deepseek 模型
· CSnakes vs Python.NET:高效嵌入与灵活互通的跨语言方案对比
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· Plotly.NET 一个为 .NET 打造的强大开源交互式图表库
点击右上角即可分享
微信分享提示