Raised the limit for one query execution

This commit is contained in:
2025-12-25 06:54:45 +03:00
parent 3ca8b328dd
commit 1dc1b8154b

View File

@@ -29,7 +29,7 @@ database_changes_active_websocket: WebSocket | None = None
async def execute_select(
query_id: int,
connection_id: int,
page_size: Annotated[int, Field(ge=1, le=100)] = 50,
page_size: Annotated[int, Field(ge=1, le=10000)] = 50,
db=Depends(get_db),
) -> SelectResult:
query = await read_select_query(db=db, query_id=query_id)
@@ -95,6 +95,7 @@ async def get_database_tables(connection_id:int):
print(r)
return r
@router.get(
"/sse-stream-cursor", dependencies=[Depends(get_current_user)], status_code=200
)
@@ -152,8 +153,6 @@ async def websocket_stream_cursor(
await websocket.close(reason="Done")
@router.websocket("/databases_changes")
async def websocket_endpoint(
websocket: WebSocket,
@@ -183,7 +182,8 @@ async def websocket_endpoint(
try:
await feed_databases_changes_ws(websocket=websocket)
except WebSocketDisconnect:
print('Closed websocket.')
print("Closed websocket.")
def serialize_list(l: list):
serialized = []
@@ -199,19 +199,19 @@ def serialize_list(l:list):
return serialized
async def feed_databases_changes_ws(websocket: WebSocket):
last_update = 0
while True:
try:
change = changes_queue.get_nowait()
if change.action == 'UPDATE':
if change.action == "UPDATE":
change.after_values = serialize_list(change.after_values)
change.before_values = serialize_list(change.before_values)
else:
change.values = serialize_list(change.values)
await websocket.send_json({"message": "change", 'change': change.model_dump()})
await websocket.send_json(
{"message": "change", "change": change.model_dump()}
)
except queue.Empty:
if last_update + 10 < time.time():
await websocket.send_json({"message": "status", "status": "Alive."})