Added cursors/push-ttl endpoint.
This commit is contained in:
@@ -76,3 +76,18 @@ async def close_cursor(cursor_id: str) -> None:
|
||||
|
||||
await cached_cursor.close()
|
||||
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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user