FastAPI + SQLite:从基础CRUD到安全并发的实战指南
核心摘要本文将带你超越FastAPI SQLite的基础CRUD搭建聚焦于安全防护认证、授权、输入验证与并发处理数据库连接池、异步优化两大实战痛点。你会获得一套可直接复用的项目骨架并理解其背后的设计逻辑确保你的应用在业务增长时依然稳健。 主要内容脉络 1. 快速搭建5分钟创建一个带CRUD的FastAPI应用⚡ 模型定义、数据库连接、基础API 2. 安全第一给你的API加上“门禁”和“监控”⚡ OAuth2密码流、JWT令牌、依赖注入保护路由 3. 应对高并发别让数据库连接成为瓶颈⚡ 连接池配置、异步会话管理、后台任务 4. 完整代码与避坑指南⚡ 项目结构、关键配置、常见陷阱 第一部分为什么是FastAPI SQLiteFastAPI就像一个高效、现代的餐厅点餐系统你客户端递上菜单JSON请求厨房后端逻辑立刻开动快速出餐JSON响应。而SQLite则是这家餐厅初期最合适的“本地仓库”——无需复杂配置一个文件搞定所有库存对于中小型应用或原型开发来说轻量且完全够用。但问题来了当顾客并发请求暴增仓库管理员数据库连接手忙脚乱或者有人冒充服务员未授权访问进入后厨。我们今天就要解决这些问题。️ 第二部分从基础CRUD到安全堡垒1. 搭建基础框架首先安装必备工具包pip install fastapi uvicorn sqlalchemy databases[aiosqlite] python-jose[cryptography] passlib[bcrypt]接下来我们定义数据模型并建立连接。SQLAlchemy的ORM让我们能用Python类操作数据库# models.py from sqlalchemy import Column, Integer, String, create_engine from sqlalchemy.ext.declarative import declarative_base DATABASE_URL sqlite:///./test.db # 注意check_same_threadFalse 仅用于SQLite简化示例生产环境需用更安全的方式 engine create_engine(DATABASE_URL, connect_args{check_same_thread: False}) Base declarative_base() class Item(Base): __tablename__ items id Column(Integer, primary_keyTrue, indexTrue) name Column(String, indexTrue) description Column(String, nullableTrue) # 创建表 Base.metadata.create_all(bindengine)然后编写FastAPI的核心CRUD接口# main.py (基础版) from fastapi import FastAPI, Depends, HTTPException from sqlalchemy.orm import Session from . import models from .database import SessionLocal, engine app FastAPI() # 依赖项获取数据库会话 def get_db(): db SessionLocal() try: yield db finally: db.close() app.post(/items/) def create_item(name: str, description: str None, db: Session Depends(get_db)): db_item models.Item(namename, descriptiondescription) db.add(db_item) db.commit() db.refresh(db_item) return db_item app.get(/items/{item_id}) def read_item(item_id: int, db: Session Depends(get_db)): item db.query(models.Item).filter(models.Item.id item_id).first() if item is None: raise HTTPException(status_code404, detailItem not found) return item # ... 更新和删除接口类似2. 构筑安全防线 警告以下安全设置是生产应用的基石切勿跳过。把API直接暴露在网上就像把家钥匙放在门垫下。我们需要认证你是谁和授权你能干什么。我们采用OAuth2密码流业界标准与JWTJSON Web Tokens组合。第一步处理密码。永远不要明文存储密码# security.py from passlib.context import CryptContext pwd_context CryptContext(schemes[bcrypt], deprecatedauto) def verify_password(plain_password, hashed_password): return pwd_context.verify(plain_password, hashed_password) def get_password_hash(password): return pwd_context.hash(password)第二步创建和验证JWT令牌。# auth.py from jose import JWTError, jwt from datetime import datetime, timedelta SECRET_KEY your-secret-key-change-this-in-production # 生产环境必须用强密钥且从环境变量读取 ALGORITHM HS256 ACCESS_TOKEN_EXPIRE_MINUTES 30 def create_access_token(data: dict): to_encode data.copy() expire datetime.utcnow() timedelta(minutesACCESS_TOKEN_EXPIRE_MINUTES) to_encode.update({exp: expire}) encoded_jwt jwt.encode(to_encode, SECRET_KEY, algorithmALGORITHM) return encoded_jwt async def get_current_user(token: str Depends(oauth2_scheme), db: Session Depends(get_db)): credentials_exception HTTPException(status_code401, detail无效凭证) try: payload jwt.decode(token, SECRET_KEY, algorithms[ALGORITHM]) username: str payload.get(sub) if username is None: raise credentials_exception except JWTError: raise credentials_exception user db.query(models.User).filter(models.User.username username).first() if user is None: raise credentials_exception return user第三步用依赖项保护路由。现在只有携带有效令牌的请求才能创建项目app.post(/secure-items/) def create_secure_item( item_data: schemas.ItemCreate, db: Session Depends(get_db), current_user: models.User Depends(get_current_user) # 关键的保护层 ): # 现在可以安全地创建因为用户已通过认证 ...⚡ 第三部分驯服并发——连接池与异步当100个顾客同时点餐只有1个服务员数据库连接会怎样拥堵SQLite在默认设置下对并发的支持较弱但通过正确配置连接池可以极大改善。核心使用databases库和异步SQLAlchemy。# database_async.py from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession from sqlalchemy.orm import sessionmaker # 使用aiosqlite驱动 ASYNC_DATABASE_URL sqliteaiosqlite:///./test_async.db async_engine create_async_engine(ASYNC_DATABASE_URL, pool_pre_pingTrue, pool_size10, max_overflow20) AsyncSessionLocal sessionmaker( bindasync_engine, class_AsyncSession, expire_on_commitFalse ) async def get_async_db(): async with AsyncSessionLocal() as session: yield session # 在FastAPI路由中使用 app.post(/async-items/, response_modelschemas.Item) async def create_async_item( item: schemas.ItemCreate, db: AsyncSession Depends(get_async_db) ): db_item models.Item(**item.dict()) db.add(db_item) await db.commit() await db.refresh(db_item) return db_item关键参数解释-pool_size10保持10个常开连接随时待命。-max_overflow20允许在繁忙时临时创建最多20个额外连接。-pool_pre_pingTrue自动检查连接是否存活避免使用已断开的连接。重要对于发送邮件、处理大文件等耗时但非数据库密集型操作务必使用BackgroundTasks避免阻塞主线程影响API响应速度。 第四部分整合与避坑指南项目结构建议your_project/ ├── main.py # FastAPI应用创建和路由聚合 ├── models.py # SQLAlchemy数据模型 ├── schemas.py # Pydantic请求/响应模型用于数据验证 ├── crud.py # 核心的数据库操作函数 ├── database.py # 数据库引擎和会话工厂同步/异步 ├── auth.py # 认证、令牌相关函数 ├── security.py # 密码哈希函数 └── dependencies.py # 可重用的FastAPI依赖项一个强化版的主文件示例# main.py 精简示例 from fastapi import FastAPI, Depends, HTTPException, BackgroundTasks from fastapi.middleware.cors import CORSMiddleware from . import models, schemas, crud, auth from .database import AsyncSessionLocal, get_async_db app FastAPI(titleMy Secure API) # 添加CORS中间件根据需求配置 app.add_middleware(CORSMiddleware, allow_origins[*]) # 生产环境应指定具体域名 app.post(/token) async def login_for_access_token(form_data: OAuth2PasswordRequestForm Depends(), db: AsyncSession Depends(get_async_db)): user await authenticate_user(form_data.username, form_data.password, db) if not user: raise HTTPException(status_code400, detail用户名或密码错误) access_token auth.create_access_token(data{sub: user.username}) return {access_token: access_token, token_type: bearer} app.get(/users/me/) async def read_users_me(current_user: models.User Depends(auth.get_current_user)): return current_user # 受保护的、异步的、带后台任务的路由示例 app.post(/items-with-notify/) async def create_item_notify( item: schemas.ItemCreate, background_tasks: BackgroundTasks, db: AsyncSession Depends(get_async_db), current_user: models.User Depends(auth.get_current_user) ): db_item await crud.create_user_item(dbdb, itemitem, user_idcurrent_user.id) # 假设有个发送通知的函数 background_tasks.add_task(send_notification, f新项目 {item.name} 已创建) return db_item必须牢记的避坑点1️⃣SQLite适用于开发与轻量生产。用户量巨大或写入频繁时考虑PostgreSQL或MySQL。2️⃣SECRET_KEY是生命线。必须通过环境变量传入且定期更换。3️⃣连接池参数需按压调整。pool_size和max_overflow不是越大越好根据实际负载测试找到甜蜜点。4️⃣善用异步async/await。在I/O等待时释放CPU但确保整个调用链数据库驱动、HTTP客户端都支持异步。5️⃣验证所有输入。始终使用Pydantic模型验证请求数据这是防止注入和错误数据的第一道防线。
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2473295.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!