如何连接到 sqlite3 db 文件并在 fastapi 中获取内容？

Question

我有一个 sqlite.db 文件，有 5 列和 1000 万行。我已经使用 fastapi 创建了一个 api，现在在其中一个 api 方法中我想连接到 sqlite.db 文件并根据某些条件（基于存在的列）获取内容。我主要会使用 SELECT 和 WHERE。

我怎样才能通过利用异步请求来做到这一点。我遇到过 Tortoise ORM，但我不知道如何正确使用它来获取结果。

from fastapi import FastAPI, UploadFile, File, Form
from fastapi.middleware.cors import CORSMiddleware

DATABASE_URL = "sqlite:///test.db"


@app.post("/test")
async def fetch_data(id: int):
    query = "SELECT * FROM tablename WHERE ID={}".format(str(id))

    # how can I fetch such query faster from 10 million records while taking advantage of async func
    return  results

Answer 1

你在这里遗漏了一点，定义一个函数async是不够的。您需要使用异步数据库驱动程序才能利用协程的优势。

Encode的数据库库非常适合此目的。

pip install databases

您还可以使用以下命令安装所需的数据库驱动程序：

pip install databases[sqlite]

就您而言，这应该会起到很好的作用。

from fastapi import FastAPI, UploadFile, File, Form
from fastapi.middleware.cors import CORSMiddleware
from databases import Database

database = Database("sqlite:///test.db")


@app.on_event("startup")
async def database_connect():
    await database.connect()


@app.on_event("shutdown")
async def database_disconnect():
    await database.disconnect()


@app.post("/test")
async def fetch_data(id: int):
    query = "SELECT * FROM tablename WHERE ID={}".format(str(id))
    results = await database.fetch_all(query=query)

    return  results