Skip to content

Commit 4ab4c26

Browse files
exiaohuliyuxuan-bd
authored andcommitted
fix: add debug log for batch retrying
1 parent a3966e0 commit 4ab4c26

File tree

1 file changed

+26
-0
lines changed
  • volcenginesdkarkruntime/resources/batch

1 file changed

+26
-0
lines changed

volcenginesdkarkruntime/resources/batch/_utils.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,11 +87,24 @@ def with_batch_retry(
8787
return func(*args, **kwargs)
8888
except ArkAPIConnectionError:
8989
waitTime = _calculate_retry_timeout(retry_times)
90+
log.debug(
91+
"Retry due to connection error, wait time: %is, retry times: %i",
92+
waitTime,
93+
retry_times,
94+
)
95+
9096
if datetime.now() + timedelta(seconds=waitTime) > deadline:
9197
raise ArkAPITimeoutError(None, None)
9298
time.sleep(waitTime)
9399
except ArkAPIStatusError as err:
94100
retry_after = _get_retry_after(err.response)
101+
log.debug(
102+
"Got status error, retry after: %is, retry times: %i, error: %s",
103+
retry_after,
104+
retry_times,
105+
err,
106+
)
107+
95108
if retry_after is not None and retry_after > 0:
96109
breaker.reset(retry_after)
97110
if not _should_retry(err.response):
@@ -116,11 +129,24 @@ async def async_with_batch_retry(
116129
return await func(*args, **kwargs)
117130
except ArkAPIConnectionError:
118131
waitTime = _calculate_retry_timeout(retry_times)
132+
log.debug(
133+
"Retry due to connection error, wait time: %is, retry times: %i",
134+
waitTime,
135+
retry_times,
136+
)
137+
119138
if datetime.now() + timedelta(seconds=waitTime) > deadline:
120139
raise ArkAPITimeoutError(None, None)
121140
await asyncio.sleep(waitTime)
122141
except ArkAPIStatusError as err:
123142
retry_after = _get_retry_after(err.response)
143+
log.debug(
144+
"Got status error, retry after: %is, retry times: %i, error: %s",
145+
retry_after,
146+
retry_times,
147+
err,
148+
)
149+
124150
if retry_after is not None and retry_after > 0:
125151
breaker.reset(retry_after)
126152
if not _should_retry(err.response):

0 commit comments

Comments
 (0)