mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 00:10:25 +08:00
feat: message notify
This commit is contained in:
meilin.huang
257
server/pkg/httpx/httpx.go
Normal file
257
server/pkg/httpx/httpx.go
Normal file
@@ -0,0 +1,257 @@
|
||||
package httpx
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"mayfly-go/pkg/logx"
|
||||
"mayfly-go/pkg/utils/anyx"
|
||||
"mayfly-go/pkg/utils/collx"
|
||||
"mime/multipart"
|
||||
"net/http"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/may-fly/cast"
|
||||
)
|
||||
|
||||
// 默认超时
|
||||
const DefTimeout = 60
|
||||
|
||||
type Req struct {
|
||||
client http.Client
|
||||
url string
|
||||
method string
|
||||
timeout int
|
||||
body io.Reader
|
||||
header map[string]string
|
||||
}
|
||||
|
||||
type MultipartFile struct {
|
||||
FieldName string // 字段名
|
||||
FileName string // 文件名
|
||||
FilePath string // 文件路径，文件路径不为空，则优先读取文件路径的内容
|
||||
Bytes []byte // 文件内容
|
||||
}
|
||||
|
||||
// 创建一个请求
|
||||
func NewReq(url string) *Req {
|
||||
return &Req{url: url, client: http.Client{}}
|
||||
}
|
||||
|
||||
func (r *Req) Url(url string) *Req {
|
||||
r.url = url
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *Req) Header(name, value string) *Req {
|
||||
if r.header == nil {
|
||||
r.header = make(map[string]string)
|
||||
}
|
||||
r.header[name] = value
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *Req) Timeout(timeout int) *Req {
|
||||
r.timeout = timeout
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *Req) GetByQuery(queryMap collx.M) *Resp {
|
||||
var params string
|
||||
for k, v := range queryMap {
|
||||
if params != "" {
|
||||
params += "&"
|
||||
}
|
||||
params += k + "=" + anyx.ToString(v)
|
||||
}
|
||||
r.url += "?" + params
|
||||
return r.Get()
|
||||
}
|
||||
|
||||
func (r *Req) Get() *Resp {
|
||||
r.method = "GET"
|
||||
r.body = nil
|
||||
return sendRequest(r)
|
||||
}
|
||||
|
||||
func (r *Req) PostJson(body string) *Resp {
|
||||
buf := bytes.NewBufferString(body)
|
||||
r.method = "POST"
|
||||
r.body = buf
|
||||
if r.header == nil {
|
||||
r.header = make(map[string]string)
|
||||
}
|
||||
r.header["Content-type"] = "application/json"
|
||||
return sendRequest(r)
|
||||
}
|
||||
|
||||
func (r *Req) PostObj(body any) *Resp {
|
||||
marshal, err := json.Marshal(body)
|
||||
if err != nil {
|
||||
return &Resp{err: errors.New("解析json obj错误")}
|
||||
}
|
||||
return r.PostJson(string(marshal))
|
||||
}
|
||||
|
||||
func (r *Req) PostForm(params string) *Resp {
|
||||
buf := bytes.NewBufferString(params)
|
||||
r.method = "POST"
|
||||
r.body = buf
|
||||
if r.header == nil {
|
||||
r.header = make(map[string]string)
|
||||
}
|
||||
r.header["Content-type"] = "application/x-www-form-urlencoded"
|
||||
return sendRequest(r)
|
||||
}
|
||||
|
||||
func (r *Req) PostMulipart(files []MultipartFile, reqParams collx.M) *Resp {
|
||||
buf := &bytes.Buffer{}
|
||||
// 文件写入 buf
|
||||
writer := multipart.NewWriter(buf)
|
||||
for _, uploadFile := range files {
|
||||
var reader io.Reader
|
||||
// 如果文件路径不为空，则读取该路径文件，否则使用bytes
|
||||
if uploadFile.FilePath != "" {
|
||||
file, err := os.Open(uploadFile.FilePath)
|
||||
if err != nil {
|
||||
return &Resp{err: err}
|
||||
}
|
||||
defer file.Close()
|
||||
reader = file
|
||||
} else {
|
||||
reader = bytes.NewBuffer(uploadFile.Bytes)
|
||||
}
|
||||
|
||||
part, err := writer.CreateFormFile(uploadFile.FieldName, uploadFile.FileName)
|
||||
if err != nil {
|
||||
return &Resp{err: err}
|
||||
}
|
||||
io.Copy(part, reader)
|
||||
}
|
||||
// 如果有其他参数，则写入body
|
||||
for k, v := range reqParams {
|
||||
if err := writer.WriteField(k, cast.ToString(v)); err != nil {
|
||||
return &Resp{err: err}
|
||||
}
|
||||
}
|
||||
if err := writer.Close(); err != nil {
|
||||
return &Resp{err: err}
|
||||
}
|
||||
|
||||
r.method = "POST"
|
||||
r.body = buf
|
||||
if r.header == nil {
|
||||
r.header = make(map[string]string)
|
||||
}
|
||||
r.header["Content-type"] = writer.FormDataContentType()
|
||||
return sendRequest(r)
|
||||
}
|
||||
|
||||
func sendRequest(rw *Req) *Resp {
|
||||
respWrapper := &Resp{}
|
||||
timeout := rw.timeout
|
||||
if timeout > 0 {
|
||||
rw.client.Timeout = time.Duration(timeout) * time.Second
|
||||
} else {
|
||||
timeout = DefTimeout
|
||||
}
|
||||
|
||||
req, err := http.NewRequest(rw.method, rw.url, rw.body)
|
||||
if err != nil {
|
||||
respWrapper.err = fmt.Errorf("创建请求错误-%s", err.Error())
|
||||
return respWrapper
|
||||
}
|
||||
setRequestHeader(req, rw.header)
|
||||
resp, err := rw.client.Do(req)
|
||||
return &Resp{resp: resp, err: err}
|
||||
}
|
||||
|
||||
func setRequestHeader(req *http.Request, header map[string]string) {
|
||||
req.Header.Set("User-Agent", "golang/mayfly-go")
|
||||
for k, v := range header {
|
||||
req.Header.Set(k, v)
|
||||
}
|
||||
}
|
||||
|
||||
type Resp struct {
|
||||
resp *http.Response
|
||||
err error
|
||||
}
|
||||
|
||||
// BodyTo 将响应体通过json解析转为指定结构体
|
||||
func (r *Resp) BodyTo(ptr any) error {
|
||||
bodyBytes, err := r.BodyBytes()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = json.Unmarshal(bodyBytes, &ptr)
|
||||
if err != nil {
|
||||
return fmt.Errorf("解析响应体-json解析失败-%s", err.Error())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// BodyToString 将响应体转为strings
|
||||
func (r *Resp) BodyToString() (string, error) {
|
||||
bodyBytes, err := r.BodyBytes()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return string(bodyBytes), nil
|
||||
}
|
||||
|
||||
// BodyToMap 将响应体通过json解析转为map
|
||||
func (r *Resp) BodyToMap() (map[string]any, error) {
|
||||
var res map[string]any
|
||||
return res, r.BodyTo(&res)
|
||||
}
|
||||
|
||||
// BodyBytes 获取响应体的字节数组
|
||||
func (r *Resp) BodyBytes() ([]byte, error) {
|
||||
bodyReader, err := r.BodyReader()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer bodyReader.Close()
|
||||
body, err := io.ReadAll(bodyReader)
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("读取响应体数据失败-%s", err.Error())
|
||||
}
|
||||
return body, err
|
||||
}
|
||||
|
||||
// BodyReader 获取响应体的reader
|
||||
func (r *Resp) BodyReader() (io.ReadCloser, error) {
|
||||
resp, err := r.GetHttpResp()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return resp.Body, nil
|
||||
}
|
||||
|
||||
// GetHttpResp 获取http响应结果结构体
|
||||
func (r *Resp) GetHttpResp() (*http.Response, error) {
|
||||
if r.err != nil {
|
||||
return nil, fmt.Errorf("请求失败-%s", r.err.Error())
|
||||
}
|
||||
if r.resp == nil {
|
||||
return nil, errors.New("请求失败-响应结构体为空,请检查请求url等信息")
|
||||
}
|
||||
|
||||
statusCode := r.resp.StatusCode
|
||||
if isFailureStatusCode(statusCode) {
|
||||
logx.Warnf("请求响应状态码为为失败状态: %v", statusCode)
|
||||
}
|
||||
|
||||
return r.resp, nil
|
||||
}
|
||||
|
||||
func isFailureStatusCode(statusCode int) bool {
|
||||
return statusCode < http.StatusOK || statusCode >= http.StatusBadRequest
|
||||
}
|
||||
35
server/pkg/httpx/httpx_test.go
Normal file
35
server/pkg/httpx/httpx_test.go
Normal file
@@ -0,0 +1,35 @@
|
||||
package httpx
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"mayfly-go/pkg/utils/collx"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type TestStruct struct {
|
||||
Id uint64
|
||||
Username string
|
||||
}
|
||||
|
||||
func TestGet(t *testing.T) {
|
||||
res, err := NewReq("www.baidu.com").Get().BodyToString()
|
||||
fmt.Println(err)
|
||||
fmt.Println(res)
|
||||
}
|
||||
|
||||
func TestGetBodyToMap(t *testing.T) {
|
||||
res, err := NewReq("http://go.mayfly.run/api/syslogs?pageNum=1&pageSize=10").Get().BodyToMap()
|
||||
fmt.Println(err)
|
||||
fmt.Println(res["msg"])
|
||||
fmt.Println(res["code"])
|
||||
}
|
||||
|
||||
func TestGetQueryBodyToMap(t *testing.T) {
|
||||
res, err := NewReq("http://go.mayfly.run/api/syslogs").
|
||||
Header("Authorization", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2OTUzOTQ5NTIsImlkIjoxLCJ1c2VybmFtZSI6ImFkbWluIn0.pGrczVZqk5nlId-FZPkjW_O5Sw3-2yjgzACp_j4JEXY").
|
||||
GetByQuery(collx.M{"pageNum": 1, "pageSize": 10}).
|
||||
BodyToMap()
|
||||
fmt.Println(err)
|
||||
fmt.Println(res["msg"])
|
||||
fmt.Println(res["code"])
|
||||
}
|
||||
Reference in New Issue
Block a user