mirror of
https://gitee.com/durcframework/SOP.git
synced 2025-08-11 21:57:56 +08:00
128 lines
3.2 KiB
Go
128 lines
3.2 KiB
Go
package common
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"io"
|
||
"io/ioutil"
|
||
"mime/multipart"
|
||
"net/http"
|
||
"net/url"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
)
|
||
|
||
type UploadFile struct {
|
||
// 表单名称
|
||
Name string
|
||
// 文件全路径
|
||
Filepath string
|
||
}
|
||
|
||
// 请求客户端
|
||
var httpClient = &http.Client{}
|
||
|
||
func Get(reqUrl string, reqParams map[string]string, headers map[string]string) string {
|
||
urlParams := url.Values{}
|
||
Url, _ := url.Parse(reqUrl)
|
||
for key, val := range reqParams {
|
||
urlParams.Set(key, val)
|
||
}
|
||
|
||
//如果参数中有中文参数,这个方法会进行URLEncode
|
||
Url.RawQuery = urlParams.Encode()
|
||
// 得到完整的url，http://xx?query
|
||
urlPath := Url.String()
|
||
|
||
httpRequest,_ := http.NewRequest("GET", urlPath, nil)
|
||
// 添加请求头
|
||
if headers != nil {
|
||
for k, v := range headers {
|
||
httpRequest.Header.Add(k,v)
|
||
}
|
||
}
|
||
// 发送请求
|
||
resp, err := httpClient.Do(httpRequest)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
defer resp.Body.Close()
|
||
response, _ := ioutil.ReadAll(resp.Body)
|
||
return string(response)
|
||
}
|
||
|
||
func PostForm(reqUrl string, reqParams map[string]string, headers map[string]string) string {
|
||
return post(reqUrl, reqParams, "application/x-www-form-urlencoded", nil, headers)
|
||
}
|
||
|
||
func PostJson(reqUrl string, reqParams map[string]string, headers map[string]string) string {
|
||
return post(reqUrl, reqParams, "application/json", nil, headers)
|
||
}
|
||
|
||
func PostFile(reqUrl string, reqParams map[string]string, files []UploadFile, headers map[string]string) string {
|
||
return post(reqUrl, reqParams, "multipart/form-data", files, headers)
|
||
}
|
||
|
||
|
||
func post(reqUrl string, reqParams map[string]string, contentType string, files []UploadFile, headers map[string]string) string {
|
||
requestBody, realContentType := getReader(reqParams, contentType, files)
|
||
httpRequest,_ := http.NewRequest("POST", reqUrl, requestBody)
|
||
// 添加请求头
|
||
httpRequest.Header.Add("Content-Type", realContentType)
|
||
if headers != nil {
|
||
for k, v := range headers {
|
||
httpRequest.Header.Add(k,v)
|
||
}
|
||
}
|
||
// 发送请求
|
||
resp, err := httpClient.Do(httpRequest)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
defer resp.Body.Close()
|
||
response, _ := ioutil.ReadAll(resp.Body)
|
||
return string(response)
|
||
}
|
||
|
||
func getReader(reqParams map[string]string, contentType string, files []UploadFile) (io.Reader, string) {
|
||
if strings.Index(contentType, "json") > -1 {
|
||
bytesData, _ := json.Marshal(reqParams)
|
||
return bytes.NewReader(bytesData), contentType
|
||
} else if files != nil {
|
||
body := &bytes.Buffer{}
|
||
// 文件写入 body
|
||
writer := multipart.NewWriter(body)
|
||
for _, uploadFile := range files {
|
||
file, err := os.Open(uploadFile.Filepath)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
part, err := writer.CreateFormFile(uploadFile.Name, filepath.Base(uploadFile.Filepath))
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
_, err = io.Copy(part, file)
|
||
file.Close()
|
||
}
|
||
// 其他参数列表写入 body
|
||
for k, v := range reqParams {
|
||
if err := writer.WriteField(k, v); err != nil {
|
||
panic(err)
|
||
}
|
||
}
|
||
if err := writer.Close(); err != nil {
|
||
panic(err)
|
||
}
|
||
// 上传文件需要自己专用的contentType
|
||
return body, writer.FormDataContentType()
|
||
} else {
|
||
urlValues := url.Values{}
|
||
for key, val := range reqParams {
|
||
urlValues.Set(key, val)
|
||
}
|
||
reqBody:= urlValues.Encode()
|
||
return strings.NewReader(reqBody), contentType
|
||
}
|
||
}
|