kurbezz/discord-bot
1
0
Fork 0
mirror of https://github.com/kurbezz/discord-bot.git synced 2026-03-03 11:30:46 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add on_stream_state_change retries

Browse Source
This commit is contained in:
Bulat Kurbanov
2024-11-19 14:14:50 +01:00
parent d7463ed316
commit 2f1e2c0ffa
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/core/broker.py
View File

@@ -1,4 +1,5 @@
from taskiq import TaskiqScheduler from taskiq import TaskiqScheduler
from taskiq.middlewares.retry_middleware import SimpleRetryMiddleware
from taskiq.schedule_sources import LabelScheduleSource from taskiq.schedule_sources import LabelScheduleSource
from taskiq_redis import ListQueueBroker, RedisAsyncResultBackend from taskiq_redis import ListQueueBroker, RedisAsyncResultBackend
@@ -6,6 +7,9 @@ from core.config import config
broker = ListQueueBroker(url=config.REDIS_URI) \ broker = ListQueueBroker(url=config.REDIS_URI) \
.with_middlewares(
SimpleRetryMiddleware(default_retry_count=5)
) \
.with_result_backend(RedisAsyncResultBackend( .with_result_backend(RedisAsyncResultBackend(
redis_url=config.REDIS_URI, redis_url=config.REDIS_URI,
result_ex_time=60 * 60 * 24 * 7, result_ex_time=60 * 60 * 24 * 7,

5
src/modules/stream_notifications/tasks.py
View File

@@ -8,7 +8,10 @@ from .watcher import StateWatcher
from .twitch.authorize import authorize from .twitch.authorize import authorize
@broker.task("stream_notifications.twitch.on_stream_state_change") @broker.task(
"stream_notifications.twitch.on_stream_state_change",
retry_on_error=True
)
async def on_stream_state_change( async def on_stream_state_change(
streamer_id: int, new_state: State | None = None streamer_id: int, new_state: State | None = None
): ):