Golang发送HTTP请求
import (
"bytes"
"encoding/json"
"errors"
"io"
"io/ioutil"
"net/http"
"github.com/golang/glog"
)
// AuthResp 授权码返回结构
type AuthResp struct {
Token string `json:"token"`
}
// GetAuthHeader 获取auth header
func GetAuthHeader() (header map[string]string, err error) {
// 1、获取授权码
auth, err := getAuthorization()
if err != nil {
glog.Errorf("getAuthorization failed, err=%v", err)
return
}
// 2、生成header
header = map[string]string{
"Authorization": auth,
}
return
}
// getAuthorization 获取授权码
func getAuthorization() (auth string, err error) {
// 1、构建需要的参数
data := map[string]string{
"username": "username",
"password": "password",
}
jsonData, err := json.Marshal(data)
if err != nil {
return
}
// 2、请求获取授权码
resp, err := sendRequest("https://ip:port/auth", bytes.NewReader(jsonData), nil, "POST")
if err != nil {
glog.Errorf("sendRequest failed, err=%v", err)
return
}
// 3、从结果中获取token
authResp := AuthResp{}
err = json.Unmarshal(resp, &authResp)
if err != nil {
return
}
auth = authResp.Token
return
}
// sendRequest 发送request
func sendRequest(url string, body io.Reader, addHeaders map[string]string, method string) (resp []byte, err error) {
// 1、创建req
req, err := http.NewRequest(method, url, body)
if err != nil {
return
}
req.Header.Add("Content-Type", "application/json")
// 2、设置headers
if len(addHeaders) > 0 {
for k, v := range addHeaders {
req.Header.Add(k, v)
}
}
// 3、发送http请求
client := &http.Client{}
response, err := client.Do(req)
if err != nil {
return
}
defer response.Body.Close()
if response.StatusCode != 200 {
err = errors.New("http status err")
glog.Errorf("sendRequest failed, url=%v, response status code=%d", url, response.StatusCode)
return
}
// 4、结果读取
resp, err = ioutil.ReadAll(response.Body)
return
}
Postman生成代码
也可用postman生成代码后自己改造下
