Merge ed29771 into 8c037de

runtime/tchannel_client.go

@@ -74,6 +74,9 @@ type TChannelClientOption struct {
 
 	// MaxAttempts is the maximum retry count for a client
 	MaxAttempts int
+
+	// RetryOn is the types of errors to retry on.
+	RetryOn tchannel.RetryOn
 }
 
 // TChannelClient implements TChannelCaller and makes outgoing Thrift calls.
@@ -98,6 +101,7 @@ type TChannelClient struct {
 	headerPatterns       []string
 	altChannelMap        map[string]*tchannel.SubChannel
 	maxAttempts          int
+	retryOn              tchannel.RetryOn
 }
 
 // NewTChannelClient is deprecated, use NewTChannelClientContext instead
@@ -143,6 +147,7 @@ func NewTChannelClientContext(
 		headerPatterns:       opt.HeaderPatterns,
 		altChannelMap:        opt.AltChannelMap,
 		maxAttempts:          opt.MaxAttempts,
+		retryOn:              opt.RetryOn,
 	}
 	return client
 }
@@ -205,6 +210,7 @@ func (c *TChannelClient) call(
 	retryOpts := tchannel.RetryOptions{
 		TimeoutPerAttempt: c.timeoutPerAttempt,
 		MaxAttempts:       c.maxAttempts,
+		RetryOn:           c.retryOn,
 	}
 
 	//override timeout and retry config with endpoint level’s config