Skip to content

Commit f731023

Browse files
tjoubertjoowani
authored andcommitted
Added QueueTimeApiExecutor to exports
1 parent 20f9a0e commit f731023

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

arango/executor.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
"AsyncApiExecutor",
55
"BatchApiExecutor",
66
"TransactionApiExecutor",
7+
"QueueTimeApiExecutor",
78
]
89

910
from collections import OrderedDict
@@ -32,6 +33,7 @@
3233
"AsyncApiExecutor",
3334
"BatchApiExecutor",
3435
"TransactionApiExecutor",
36+
"QueueTimeApiExecutor",
3537
]
3638

3739
T = TypeVar("T")
@@ -430,7 +432,6 @@ def abort(self) -> bool:
430432
raise TransactionAbortError(resp, request)
431433

432434

433-
434435
class QueueTimeApiExecutor:
435436
"""API executor that handles queue time.
436437
@@ -446,17 +447,22 @@ def __init__(self, connection: Connection) -> None:
446447
def context(self) -> str:
447448
return "queue-time"
448449

449-
def execute(self, request: Request, response_handler: Callable[[Response], T], max_queue_time_seconds: int) -> T:
450+
def execute(
451+
self,
452+
request: Request,
453+
response_handler: Callable[[Response], T],
454+
max_queue_time_seconds: int,
455+
) -> T:
450456
"""Execute an API request and return the result.
451457
452458
:param request: HTTP request.
453459
:type request: arango.request.Request
454460
:param response_handler: HTTP response handler.
455461
:type response_handler: callable
456462
:return: API execution result.
457-
"""
463+
"""
458464
request.headers["x-arango-max-queue-time-seconds"] = max_queue_time_seconds
459465
resp = self._conn.send_request(request)
460-
466+
461467
queue_time_seconds = resp.headers["x-arango-queue-time-seconds"]
462468
return response_handler(resp, queue_time_seconds)

0 commit comments

Comments
 (0)