U
vi, @ s d Z ddlZddlmZmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
dd lmZ ed
ZG d d d e
ZdS )
z7Implementation for Retry strategy during HTTP requests. N)datetime timedelta) takewhilerandom)Callable)Retry)
MaxRetryError
ResponseError)
InfluxDBErrorz"influxdb_client.client.write.retryc sz e Zd ZdZdeegef d fd d
Z fd d ZdddZ dd Z
fddZ d fdd Z dd Z
dd Z ZS ) WritesRetrya
Writes retry configuration.
The next delay is computed as random value between range
`retry_interval * exponential_base^(attempts-1)` and `retry_interval * exponential_base^(attempts)
Example:
for retry_interval=5, exponential_base=2, max_retry_delay=125, total=5
retry delays are random distributed values within the ranges of
[5-10, 10-20, 20-40, 40-80, 80-125]
r } N)retry_callbackc sP t jf | || _|| _|| _|| _|| _|| _t t
|d | _ || _ dS )a
Initialize defaults.
:param int jitter_interval: random milliseconds when retrying writes
:param num max_retry_delay: maximum delay when retrying write in seconds
:param int max_retry_time: maximum total retry timeout in seconds,
attempt after this timout throws MaxRetryError
:param int total: maximum number of retries
:param num retry_interval: initial first retry delay range in seconds
:param int exponential_base: base for the exponential retry delay,
:param Callable[[Exception], int] retry_callback: the callable ``callback`` to run after retryable
error occurred.
The callable must accept one argument:
- `Exception`: an retryable error
)secondsN)
super__init__jitter_intervaltotalretry_intervalmax_retry_delaymax_retry_timeexponential_baser nowr
retry_timeoutr ) selfr r r r r r r kw __class__ S/opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/write/retry.pyr s zWritesRetry.__init__c s d|kr| j |d< d|kr$| j|d< d|kr6| j|d< d|krH| j|d< d|krZ| j|d< d|krl| j|d< t jf |}| j|_|S )zInitialize defaults.r r r r r r ) r r r r r r r newr )r r r# r r! r" r# 9 s
zWritesRetry.newFc C s | |sdS | jo|dkS )z]is_retry doesn't require retry_after header. If there is not Retry-After we will use backoff.Fi )_is_method_retryabler )r method status_codehas_retry_afterr! r! r" is_retryL s
zWritesRetry.is_retryc C s t ttdd t| j}|d8 }|dk r0dS | j}| j| j }d}||krr|d7 }|}|| j }|| jkrFqrqF|| jkr| j}||| | S )zTVariant of exponential backoff with initial and max delay and a random jitter delay.c S s
| j d kS N)redirect_location)xr! r! r" X z.WritesRetry.get_backoff_time.. r ) lenlistr reversedhistoryr r r _random)r consecutive_errors_lenZ range_startZ
range_stopir! r! r" get_backoff_timeS s(
z WritesRetry.get_backoff_timec s t |}|r || 7 }|S )zCGet the value of Retry-After header and append random jitter delay.)r get_retry_after
_jitter_delay)r response retry_afterr r! r" r7 p s zWritesRetry.get_retry_afterc
s | j t k r"t|||p tdt ||||||}|dk rLt|d}n|dk rZ|}n
d| }d| d} t|tr| d|j d7 } | j
r|
| t | |S ) z:Return a new Retry object with incremented retry counters.zmax_retry_time exceededN)r9 zFailed request to: z6The retriable error occurred during request. Reason: 'z'.z
Retry in zs.)
r r r r r
r incrementr
isinstancer: r loggerwarning)
r r% urlr9 error_pool _stacktrace new_retryZ parsed_errormessager r! r" r; w s
zWritesRetry.incrementc C s | j t S r) )r r r r! r! r" r8 s zWritesRetry._jitter_delayc C s t S r) r rE r! r! r" r3 s zWritesRetry._random)r r
r r r r N)F)NNNNNN)__name__
__module__ __qualname____doc__r Exceptionintr r# r( r6 r7 r; r8 r3
__classcell__r! r! r r" r s
r )rI loggingr r itertoolsr r typingr urllib3r Zurllib3.exceptionsr r
Z!influxdb_client.client.exceptionsr getLoggerr= r r! r! r! r" s