Add queue status command

This commit is contained in:
2026-02-07 22:57:36 +03:00
parent 77801e9123
commit 8d5eda3244
4 changed files with 29 additions and 7 deletions

View File

@@ -54,7 +54,7 @@ async def cmd_artifacts_upload(msg: Message):
finally: finally:
release_lock("artifacts") release_lock("artifacts")
pos = await enqueue(job) pos = await enqueue("artifact-upload", job)
await msg.answer(f"🕓 Upload queued (#{pos})", reply_markup=artifacts_kb) await msg.answer(f"🕓 Upload queued (#{pos})", reply_markup=artifacts_kb)

View File

@@ -7,7 +7,7 @@ from app import dp
from auth import is_admin_msg from auth import is_admin_msg
from keyboards import backup_kb from keyboards import backup_kb
from lock_utils import acquire_lock, release_lock from lock_utils import acquire_lock, release_lock
from services.queue import enqueue from services.queue import enqueue, format_status
from services.backup import backup_badge, restore_help from services.backup import backup_badge, restore_help
from services.runner import run_cmd from services.runner import run_cmd
@@ -122,7 +122,7 @@ async def cmd_backup_now(msg: Message):
finally: finally:
release_lock("backup") release_lock("backup")
pos = await enqueue(job) pos = await enqueue("backup", job)
await msg.answer(f"🕓 Backup queued (#{pos})", reply_markup=backup_kb) await msg.answer(f"🕓 Backup queued (#{pos})", reply_markup=backup_kb)
@@ -190,6 +190,12 @@ async def ls(msg: Message):
await cmd_last_snapshot(msg) await cmd_last_snapshot(msg)
@dp.message(F.text == "🧾 Queue")
async def qb(msg: Message):
if is_admin_msg(msg):
await msg.answer(format_status(), reply_markup=backup_kb)
@dp.message(F.text == "▶️ Run backup") @dp.message(F.text == "▶️ Run backup")
async def br(msg: Message): async def br(msg: Message):
if is_admin_msg(msg): if is_admin_msg(msg):

View File

@@ -29,7 +29,8 @@ backup_kb = ReplyKeyboardMarkup(
keyboard=[ keyboard=[
[KeyboardButton(text="📦 Status"), KeyboardButton(text="📦 Last snapshot")], [KeyboardButton(text="📦 Status"), KeyboardButton(text="📦 Last snapshot")],
[KeyboardButton(text="📊 Repo stats"), KeyboardButton(text="🧯 Restore help")], [KeyboardButton(text="📊 Repo stats"), KeyboardButton(text="🧯 Restore help")],
[KeyboardButton(text="▶️ Run backup"), KeyboardButton(text="⬅️ Назад")], [KeyboardButton(text="▶️ Run backup"), KeyboardButton(text="🧾 Queue")],
[KeyboardButton(text="⬅️ Назад")],
], ],
resize_keyboard=True, resize_keyboard=True,
) )

View File

@@ -3,17 +3,32 @@ from typing import Awaitable, Callable
_queue: asyncio.Queue = asyncio.Queue() _queue: asyncio.Queue = asyncio.Queue()
_current_label: str | None = None
async def enqueue(job: Callable[[], Awaitable[None]]) -> int: async def enqueue(label: str, job: Callable[[], Awaitable[None]]) -> int:
await _queue.put(job) await _queue.put((label, job))
return _queue.qsize() return _queue.qsize()
async def worker(): async def worker():
global _current_label
while True: while True:
job = await _queue.get() label, job = await _queue.get()
_current_label = label
try: try:
await job() await job()
finally: finally:
_current_label = None
_queue.task_done() _queue.task_done()
def format_status() -> str:
pending = [label for label, _ in list(_queue._queue)]
lines = ["🧾 Queue"]
lines.append(f"🔄 Running: {_current_label or 'idle'}")
lines.append(f"⏳ Pending: {len(pending)}")
if pending:
preview = ", ".join(pending[:5])
lines.append(f"➡️ Next: {preview}")
return "\n".join(lines)