Skip to content

Commit

Permalink
support configure HostClient (#1214)
Browse files Browse the repository at this point in the history
Co-authored-by: linlonghui <linlonghui@chandashi.com>
  • Loading branch information
farss and linlonghui committed Feb 15, 2022
1 parent 632e222 commit c94581c
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions client.go
Expand Up @@ -297,6 +297,9 @@ type Client struct {
// By default will use isIdempotent function
RetryIf RetryIfFunc

// ConfigureClient configures the fasthttp.HostClient.
ConfigureClient func(hc *HostClient) error

mLock sync.Mutex
m map[string]*HostClient
ms map[string]*HostClient
Expand Down Expand Up @@ -510,6 +513,13 @@ func (c *Client) Do(req *Request, resp *Response) error {
clientReaderPool: &c.readerPool,
clientWriterPool: &c.writerPool,
}

if c.ConfigureClient != nil {
if err := c.ConfigureClient(hc); err != nil {
return err
}
}

m[string(host)] = hc
if len(m) == 1 {
startCleaner = true
Expand Down

0 comments on commit c94581c

Please sign in to comment.