feat(httpreq): 添加设置最大空闲连接数方法

新增 SetMaxIdleConns 方法用于设置 HTTP 客户端的最大空闲连接数和每主机最大空闲连接数

Author: inhere

netutil/httpreq/client.go

@@ -68,6 +68,15 @@ func (h *Client) SetTimeout(ms int) *Client {
 	return h
 }
 
+// SetMaxIdleConns Set the maximum number of idle connections.
+func (h *Client) SetMaxIdleConns(maxIdleConns, maxIdleConnsPerHost int) {
+	if hc, ok := h.client.(*http.Client); ok {
+		transport := hc.Transport.(*http.Transport)
+		transport.MaxIdleConns = maxIdleConns
+		transport.MaxIdleConnsPerHost = maxIdleConnsPerHost
+	}
+}
+
 // BaseURL set request base URL
 func (h *Client) BaseURL(baseURL string) *Client {
 	h.baseURL = baseURL

netutil/httpreq/client_test.go

@@ -5,6 +5,7 @@ import (
 	"net/http"
 	"strings"
 	"testing"
+	"time"
 
 	"github.com/gookit/goutil/dump"
 	"github.com/gookit/goutil/jsonutil"
@@ -127,3 +128,186 @@ func TestHttpReq_MustSend(t *testing.T) {
 	assert.True(t, httpreq.IsOK(sc))
 	assert.True(t, httpreq.IsSuccessful(sc))
 }
+
+func TestClient_NewWithTimeout(t *testing.T) {
+	cli := httpreq.NewWithTimeout(5000)
+	assert.NotNil(t, cli)
+	assert.NotNil(t, cli.Doer())
+}
+
+func TestClient_NewWithDoer(t *testing.T) {
+	customClient := &http.Client{Timeout: 3000 * time.Millisecond}
+	cli := httpreq.NewWithDoer(customClient)
+	assert.NotNil(t, cli)
+	assert.Eq(t, customClient, cli.Doer())
+}
+
+func TestClient_SetClient(t *testing.T) {
+	cli := httpreq.New()
+	originalDoer := cli.Doer()
+
+	newClient := &http.Client{Timeout: 2000 * time.Millisecond}
+	ret := cli.SetClient(newClient)
+
+	assert.Eq(t, newClient, cli.Doer())
+	assert.Eq(t, cli, ret)
+	assert.NotEq(t, originalDoer, cli.Doer())
+}
+
+func TestClient_SetTimeout(t *testing.T) {
+	cli := httpreq.New()
+	ret := cli.SetTimeout(5000)
+
+	assert.NotNil(t, ret)
+	assert.Eq(t, cli, ret)
+
+	hc, ok := cli.Doer().(*http.Client)
+	assert.True(t, ok)
+	assert.Eq(t, 5000*time.Millisecond, hc.Timeout)
+}
+
+func TestClient_PostJSON(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+	data := map[string]string{"name": "inhere", "city": "chengdu"}
+
+	resp, err := cli.PostJSON("/post", data)
+	assert.NoErr(t, err)
+	assert.True(t, httpreq.IsOK(resp.StatusCode))
+
+	rr := testutil.ParseRespToReply(resp)
+	assert.Eq(t, "POST", rr.Method)
+	assert.StrContains(t, rr.Body, `"name":"inhere"`)
+}
+
+func TestClient_WithData(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	opt := cli.WithData("name=inhere&age=18")
+	assert.NotNil(t, opt)
+	assert.NotNil(t, opt.Data)
+}
+
+func TestClient_WithBody(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+	body := strings.NewReader("test body")
+
+	opt := cli.WithBody(body)
+	assert.NotNil(t, opt)
+	assert.NotNil(t, opt.Body)
+}
+
+func TestClient_BytesBody(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	opt := cli.BytesBody([]byte("bytes body"))
+	assert.NotNil(t, opt)
+	assert.NotNil(t, opt.Body)
+}
+
+func TestClient_StringBody(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	opt := cli.StringBody("string body")
+	assert.NotNil(t, opt)
+	assert.NotNil(t, opt.Body)
+}
+
+func TestClient_FormBody(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	opt := cli.FormBody(map[string]string{"name": "inhere"})
+	assert.NotNil(t, opt)
+	assert.NotNil(t, opt.Body)
+}
+
+func TestClient_JSONBody(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	opt := cli.JSONBody(map[string]string{"name": "inhere"})
+	assert.NotNil(t, opt)
+	assert.NotNil(t, opt.Body)
+	assert.Eq(t, httpctype.JSON, opt.ContentType)
+}
+
+func TestClient_JSONBytesBody(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	opt := cli.JSONBytesBody([]byte(`{"name":"inhere"}`))
+	assert.NotNil(t, opt)
+	assert.NotNil(t, opt.Body)
+	assert.Eq(t, httpctype.JSON, opt.ContentType)
+}
+
+func TestClient_AnyBody(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	t.Run("string body", func(t *testing.T) {
+		opt := cli.AnyBody("string data")
+		assert.NotNil(t, opt.Body)
+	})
+
+	t.Run("bytes body", func(t *testing.T) {
+		opt := cli.AnyBody([]byte("bytes data"))
+		assert.NotNil(t, opt.Body)
+	})
+
+	t.Run("map body", func(t *testing.T) {
+		opt := cli.AnyBody(map[string]string{"key": "value"})
+		assert.NotNil(t, opt.Body)
+	})
+}
+
+func TestClient_WithOption(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+
+	opt := cli.WithOption(httpreq.WithJSONType)
+	assert.NotNil(t, opt)
+	assert.Eq(t, httpctype.JSON, opt.ContentType)
+}
+
+func TestClient_SendWithOpt(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+	opt := httpreq.NewOpt().WithMethod(http.MethodGet)
+
+	resp, err := cli.SendWithOpt("/get", opt)
+	assert.NoErr(t, err)
+	assert.True(t, httpreq.IsOK(resp.StatusCode))
+
+	rr := testutil.ParseRespToReply(resp)
+	assert.Eq(t, "GET", rr.Method)
+}
+
+func TestClient_SendRequest(t *testing.T) {
+	cli := httpreq.New(testSrvAddr)
+	req, err := http.NewRequest(http.MethodGet, testSrvAddr+"/get", nil)
+	assert.NoErr(t, err)
+
+	opt := httpreq.NewOpt()
+	resp, err := cli.SendRequest(req, opt)
+	assert.NoErr(t, err)
+	assert.True(t, httpreq.IsOK(resp.StatusCode))
+}
+
+func TestClient_BaseURL(t *testing.T) {
+	cli := httpreq.New()
+	ret := cli.BaseURL("http://localhost:8080")
+
+	assert.Eq(t, cli, ret)
+}
+
+func TestClient_DefaultMethod(t *testing.T) {
+	cli := httpreq.New()
+
+	ret := cli.DefaultMethod(http.MethodPost)
+	assert.Eq(t, cli, ret)
+
+	ret = cli.DefaultMethod("")
+	assert.Eq(t, cli, ret)
+}
+
+func TestClient_ContentType(t *testing.T) {
+	cli := httpreq.New()
+
+	ret := cli.ContentType(httpctype.JSON)
+	assert.Eq(t, cli, ret)
+}