pkg/curlx
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化一些方法

Browse Source
This commit is contained in:
Yun
2024-01-24 23:47:42 +08:00
parent 5a7b3b6245
commit 9e89fa432c
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
curlx.go
+5 -5
View File
@@ -118,10 +118,10 @@ func (c *Curlx) WithAddress(ctx context.Context, addr string) {
/** /**
* 简单请求 * 简单请求
*/ */
func (c *Curlx) Send(ctx context.Context, p ...Param) (res string, httpcode int, err error) { func (c *Curlx) Send(ctx context.Context, p ...Param) (res []byte, httpcode int, err error) {
_, response, err := c.SendExec(ctx, p...) _, response, err := c.SendExec(ctx, p...)
if err != nil { if err != nil {
return "", -1, err return nil, -1, err
} }
defer response.Body.Close() // 处理完关闭 defer response.Body.Close() // 处理完关闭
@@ -135,7 +135,7 @@ func (c *Curlx) Send(ctx context.Context, p ...Param) (res string, httpcode int,
case "gzip": case "gzip":
reader, err := gzip.NewReader(response.Body) reader, err := gzip.NewReader(response.Body)
if err != nil { if err != nil {
return "", status, err return nil, status, err
} }
for { for {
buf := make([]byte, 1024) buf := make([]byte, 1024)
@@ -152,7 +152,7 @@ func (c *Curlx) Send(ctx context.Context, p ...Param) (res string, httpcode int,
body, _ = io.ReadAll(response.Body) body, _ = io.ReadAll(response.Body)
} }
c.opts.Logger.Infof(ctx, "curlx.Send body:%s", string(body)) c.opts.Logger.Infof(ctx, "curlx.Send body:%s", string(body))
return string(body), status, nil return body, status, nil
} }
/** /**
@@ -226,7 +226,7 @@ func (c *Curlx) SendExec(ctx context.Context, ps ...Param) (req *http.Request, r
/** /**
* 流式请求 * 流式请求
*/ */
func (c *Curlx) SendChan(ctx context.Context, ps ...Param) (<-chan string, error) { func (c *Curlx) SendStream(ctx context.Context, ps ...Param) (<-chan string, error) {
data := make(chan string, 1000) data := make(chan string, 1000)