💩 add util my http

Author: xingmegshuo

request/request.go

@@ -0,0 +1,40 @@
+/***************************
+@File        : request.go
+@Time        : 2022/07/06 17:47:49
+@AUTHOR      : small_ant
+@Email       : [email protected]
+@Desc        : req send
+****************************/
+
+package request
+
+import (
+    "bytes"
+    "io/ioutil"
+    "net/http"
+)
+
+// Do
+// A simple http client.
+func (c *Cli) Do(r *Req) (body []byte, err error) {
+    req, err := http.NewRequest(r.Method, r.Url, bytes.NewBuffer(r.Data))
+    if err != nil {
+        return nil, err
+    }
+    for k, v := range c.Heder {
+        req.Header.Add(k, v)
+    }
+    resp, err := c.Client.Do(req)
+    if err != nil {
+        return nil, err
+    }
+    defer resp.Body.Close()
+    if resp.StatusCode != 200 {
+        return nil, err
+    }
+    body, err = ioutil.ReadAll(resp.Body)
+    if err != nil {
+        return nil, err
+    }
+    return body, nil
+}

request/request_test.go

@@ -0,0 +1,22 @@
+/***************************
+@File        : request_test.go
+@Time        : 2022/07/06 18:06:56
+@AUTHOR      : small_ant
+@Email       : [email protected]
+@Desc        : test request send
+****************************/
+
+package request
+
+import "testing"
+
+func TestSend(t *testing.T) {
+    cli := NewCli()
+    // cli.SetTimeout(5)
+    req := NewReq("GET", "http://www.baidu.com", nil)
+    body, err := cli.Do(req)
+    if err != nil {
+        t.Error(err)
+    }
+    t.Log(string(body))
+}

request/types.go

@@ -0,0 +1,48 @@
+/***************************
+@File        : types.go
+@Time        : 2022/07/06 17:35:50
+@AUTHOR      : small_ant
+@Email       : [email protected]
+@Desc        : Request send
+****************************/
+
+package request
+
+import (
+    "net/http"
+    "time"
+)
+
+type Cli struct {
+    Client  *http.Client
+    Timeout int64
+    Heder   map[string]string
+}
+
+type Req struct {
+    Method string
+    Url    string
+    Data   []byte
+}
+
+// NewCli returns a new Cli with a default http.Client.
+func NewCli() *Cli {
+    return &Cli{
+        Client: &http.Client{},
+    }
+}
+
+// Setting the timeout for the http client.
+func (c *Cli) SetTimeout(timeout int64) {
+    c.Timeout = timeout
+    c.Client = &http.Client{Timeout: time.Duration(timeout)}
+}
+
+// NewReq returns a pointer to a Req with the given method, url, and data.
+func NewReq(method, url string, data []byte) *Req {
+    return &Req{
+        Method: method,
+        Url:    url,
+        Data:   data,
+    }
+}