Merge pull request #240 from crimson-gao/addTimeout

Add set retry timeout interface

client.go

@@ -165,6 +165,11 @@ func (c *Client) SetHTTPClient(client *http.Client) {
 	c.HTTPClient = client
 }
 
+// SetRetryTimeout set retry timeout
+func (c *Client) SetRetryTimeout(timeout time.Duration) {
+	c.RetryTimeOut = timeout
+}
+
 // SetAuthVersion set signature version that the client used
 func (c *Client) SetAuthVersion(version AuthVersionType) {
 	c.accessKeyLock.Lock()

client_interface.go

@@ -79,6 +79,8 @@ type ClientInterface interface {
 	SetUserAgent(userAgent string)
 	// SetHTTPClient set a custom http client, all request will send to sls by this client
 	SetHTTPClient(client *http.Client)
+	// SetRetryTimeout set retry timeout, client will retry util retry timeout
+	SetRetryTimeout(timeout time.Duration)
 	// #################### Client Operations #####################
 	// ResetAccessKeyToken reset client's access key token
 	ResetAccessKeyToken(accessKeyID, accessKeySecret, securityToken string)

token_auto_update_client.go

@@ -146,6 +146,11 @@ func (c *TokenAutoUpdateClient) SetHTTPClient(client *http.Client) {
 	c.logClient.SetHTTPClient(client)
 }
 
+// SetRetryTimeout set retry timeout
+func (c *TokenAutoUpdateClient) SetRetryTimeout(timeout time.Duration) {
+	c.logClient.SetRetryTimeout(timeout)
+}
+
 // SetAuthVersion set auth version that the client used
 func (c *TokenAutoUpdateClient) SetAuthVersion(version AuthVersionType) {
 	c.logClient.SetAuthVersion(version)