martsforever
24 lines
599 B
Python
24 lines
599 B
Python
from fastapi import FastAPI
|
|
from sqlalchemy import text
|
|
|
|
from app.utils.mysql_utils import async_session
|
|
|
|
|
|
async def next_id(num: int = 1):
|
|
async with async_session() as session:
|
|
sql_string = "select " + ",".join([f"uuid() as _{index}" for index in range(num)])
|
|
print(sql_string)
|
|
result = await session.execute(text(sql_string))
|
|
val = result.first()
|
|
print(val)
|
|
arr = list(val or [])
|
|
return arr[0] if num == 1 else arr
|
|
|
|
|
|
def add_next_id_route(app: FastAPI):
|
|
@app.get("/next_id")
|
|
async def _next_id(num: int = 1):
|
|
return {
|
|
"data": await next_id(num),
|
|
}
|