Added cursors/push-ttl endpoint.

This commit is contained in:
2025-02-26 04:36:48 +03:00
parent 4c1d706938
commit 218077725d

View File

@@ -76,3 +76,18 @@ async def close_cursor(cursor_id: str) -> None:
await cached_cursor.close() await cached_cursor.close()
del mysql.cached_cursors[cursor_id] del mysql.cached_cursors[cursor_id]
@router.post(
"push-ttl/{cursor_id}",
dependencies=[Depends(get_current_user)],
status_code=status.HTTP_200_OK,
)
async def cursor_push_ttl(cursor_id: str, new_ttl: int|None=None) -> CachedCursorOut:
cached_cursor = mysql.cached_cursors.get(cursor_id, None)
if cached_cursor is None:
raise CursorNotFound
cached_cursor.ttl = new_ttl if new_ttl else cached_cursor.ttl
cached_cursor.close_at = cached_cursor.upgrade_close_at()
return cached_cursor