Skip to content

Commit 65597e2

Browse files
committed
fix: typing, middleware methods
1 parent 448624f commit 65597e2

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

taskiq/middlewares/taskiq_admin_middleware.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ async def shutdown(self) -> None:
7878
if self._client is not None:
7979
await self._client.close()
8080

81-
async def _spawn_request(
81+
def _spawn_request(
8282
self,
8383
endpoint: str,
8484
payload: dict[str, Any],
@@ -105,7 +105,7 @@ async def _send() -> None:
105105
self._pending.add(task)
106106
task.add_done_callback(self._pending.discard)
107107

108-
async def post_send(
108+
def post_send(
109109
self,
110110
message: TaskiqMessage,
111111
) -> Union[None, Coroutine[Any, Any, None], "CoroutineType[Any, Any, None]"]:
@@ -117,7 +117,7 @@ async def post_send(
117117
118118
:param message: kicked message.
119119
"""
120-
await self._spawn_request(
120+
self._spawn_request(
121121
f"/api/tasks/{message.task_id}/queued",
122122
{
123123
"args": message.args,
@@ -129,7 +129,7 @@ async def post_send(
129129
)
130130
return super().post_send(message)
131131

132-
async def pre_execute(
132+
def pre_execute(
133133
self,
134134
message: TaskiqMessage,
135135
) -> Union[
@@ -146,7 +146,7 @@ async def pre_execute(
146146
:param message: incoming parsed taskiq message.
147147
:return: modified message.
148148
"""
149-
await self._spawn_request(
149+
self._spawn_request(
150150
f"/api/tasks/{message.task_id}/started",
151151
{
152152
"args": message.args,
@@ -158,7 +158,7 @@ async def pre_execute(
158158
)
159159
return super().pre_execute(message)
160160

161-
async def post_execute(
161+
def post_execute(
162162
self,
163163
message: TaskiqMessage,
164164
result: TaskiqResult[Any],
@@ -172,7 +172,7 @@ async def post_execute(
172172
:param message: incoming message.
173173
:param result: result of execution for current task.
174174
"""
175-
await self._spawn_request(
175+
self._spawn_request(
176176
f"/api/tasks/{message.task_id}/executed",
177177
{
178178
"finishedAt": self._now_iso(),

0 commit comments

Comments
 (0)