Merge pull request #26402 from qudongfang/ensure_client_transport_be_closed

ensure transport.Client be closed
This commit is contained in:
Tibor Vass 2016-10-26 11:51:51 -07:00 committed by GitHub
commit 335a3c6149
2 changed files with 18 additions and 0 deletions

View File

@ -168,6 +168,19 @@ func NewClient(host string, version string, client *http.Client, httpHeaders map
}, nil
}
// Close ensures that transport.Client is closed
// especially needed while using NewClient with *http.Client = nil
// for example
// client.NewClient("unix:///var/run/docker.sock", nil, "v1.18", map[string]string{"User-Agent": "engine-api-cli-1.0"})
func (cli *Client) Close() error {
if t, ok := cli.client.Transport.(*http.Transport); ok {
t.CloseIdleConnections()
}
return nil
}
// getAPIPath returns the versioned request path to call the api.
// It appends the query parameters to the path if they are not empty.
func (cli *Client) getAPIPath(p string, query url.Values) string {

View File

@ -162,6 +162,11 @@ func TestGetAPIPath(t *testing.T) {
if g != cs.e {
t.Fatalf("Expected %s, got %s", cs.e, g)
}
err = c.Close()
if nil != err {
t.Fatalf("close client failed, error message: %s", err)
}
}
}