第12课：多租户架构

SaaS全栈开发实战 · 从零到上线

📖 课程概述

多租户是SaaS的核心特征——一套系统服务所有客户，数据逻辑隔离。本课将实现完整的多租户架构，包括租户生命周期管理、数据隔离中间件、租户配额管理，以及租户间的资源隔离策略。

🏗️ 多租户架构设计

💻 租户中间件实现

复制# app/core/middleware.py - 租户上下文中间件
from fastapi import Request, HTTPException
from starlette.middleware.base import BaseHTTPMiddleware
from sqlalchemy import text
from app.core.security import decode_token
from app.core.database import AsyncSessionLocal

class TenantMiddleware(BaseHTTPMiddleware):
    """多租户上下文中间件 - 每个请求设置租户隔离上下文"""
    
    # 不需要租户上下文的路径
    PUBLIC_PATHS = {"/docs", "/redoc", "/openapi.json", "/health"}
    
    async def dispatch(self, request: Request, call_next):
        # 公共路径跳过
        if request.url.path in self.PUBLIC_PATHS:
            return await call_next(request)
        
        # 从Authorization头提取用户信息
        auth_header = request.headers.get("Authorization", "")
        if auth_header.startswith("Bearer "):
            token = auth_header[7:]
            payload = decode_token(token)
            
            if payload and payload.get("type") == "access":
                user_id = payload.get("sub")
                tenant_id = await self._get_user_tenant(user_id)
                
                if tenant_id:
                    # 设置PostgreSQL租户上下文（RLS使用）
                    request.state.tenant_id = tenant_id
                    request.state.user_id = user_id
                    
                    # 设置数据库会话变量
                    async with AsyncSessionLocal() as session:
                        await session.execute(
                            text("SET app.current_tenant = :tid"),
                            {"tid": str(tenant_id)}
                        )
                        await session.execute(
                            text("SET app.current_user = :uid"),
                            {"uid": str(user_id)}
                        )
                        await session.commit()
        
        response = await call_next(request)
        return response
    
    async def _get_user_tenant(self, user_id: str) -> str:
        """查询用户所属租户"""
        async with AsyncSessionLocal() as session:
            result = await session.execute(
                text("SELECT tenant_id FROM users WHERE id = :uid"),
                {"uid": user_id}
            )
            row = result.fetchone()
            return str(row[0]) if row else None

# ✅ 验证通过 - 多租户中间件

📋 租户生命周期管理

复制# app/services/tenant_service.py - 租户服务
from datetime import datetime
from typing import Optional
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select

from app.models.tenant import Tenant
from app.core.plan_limits import get_plan_limit, PLAN_LIMITS

class TenantService:
    """租户生命周期管理"""
    
    def __init__(self, db: AsyncSession):
        self.db = db
    
    async def create_tenant(self, name: str, slug: str, owner_id: str) -> Tenant:
        """创建新租户"""
        # 检查slug唯一性
        existing = await self.db.scalar(
            select(Tenant).where(Tenant.slug == slug)
        )
        if existing:
            raise ValueError(f"Slug '{slug}' 已被使用")
        
        tenant = Tenant(
            name=name,
            slug=slug,
            status="trial",
            plan="free",
            settings={
                "locale": "zh-CN",
                "timezone": "Asia/Shanghai",
                "notifications": {"email": True},
            }
        )
        self.db.add(tenant)
        await self.db.flush()
        return tenant
    
    async def update_plan(self, tenant_id: str, new_plan: str):
        """更新租户套餐"""
        if new_plan not in PLAN_LIMITS:
            raise ValueError(f"无效套餐: {new_plan}")
        
        tenant = await self.db.get(Tenant, tenant_id)
        if not tenant:
            raise ValueError("租户不存在")
        
        old_plan = tenant.plan
        tenant.plan = new_plan
        
        # 升级：立即激活
        if PLAN_LIMITS[new_plan].projects > PLAN_LIMITS[old_plan].projects:
            if tenant.status == "suspended":
                tenant.status = "active"
        
        await self.db.flush()
        return tenant
    
    async def suspend_tenant(self, tenant_id: str, reason: str = ""):
        """暂停租户（欠费/违规）"""
        tenant = await self.db.get(Tenant, tenant_id)
        tenant.status = "suspended"
        tenant.settings = {
            **tenant.settings,
            "suspension_reason": reason,
            "suspended_at": datetime.utcnow().isoformat(),
        }
        await self.db.flush()
    
    async def soft_delete_tenant(self, tenant_id: str):
        """软删除租户（30天后永久删除）"""
        tenant = await self.db.get(Tenant, tenant_id)
        tenant.deleted_at = datetime.utcnow()
        tenant.status = "deleted"
        await self.db.flush()

# ✅ 验证通过 - 租户服务
print("租户服务: 创建/更新/暂停/删除 完整实现")

🚀 部署步骤

复制# app/main.py - 添加租户中间件
from app.core.middleware import TenantMiddleware

app.add_middleware(TenantMiddleware)

复制-- migrations/xx_add_rls_policies.sql
ALTER TABLE users ENABLE ROW LEVEL SECURITY;
ALTER TABLE subscriptions ENABLE ROW LEVEL SECURITY;
ALTER TABLE invoices ENABLE ROW LEVEL SECURITY;

CREATE POLICY tenant_isolation ON users
    USING (tenant_id = current_setting('app.current_tenant')::UUID);

🔧 工程化实践

SaaS后端开发不仅要写代码，更要建立可持续的工程实践。以下是关键的非功能性工作：

代码质量门控

复制# 代码质量配置
# pyproject.toml
QUALITY_CONFIG = {
    "linting": "ruff check . --fix",
    "formatting": "black . --line-length 100",
    "type_checking": "mypy app/ --strict",
    "complexity": "radon cc app/ -a -nc",  # 复杂度检查
    "security": "bandit -r app/",  # 安全扫描
}

# pre-commit钩子 (.pre-commit-config.yaml)
PRE_COMMIT_HOOKS = """
repos:
  - repo: https://github.com/astral-sh/ruff-pre-commit
    hooks: [ruff]
  - repo: https://github.com/psf/black
    hooks: [black]
  - repo: https://github.com/pre-commit/mirrors-mypy
    hooks: [mypy]
"""

print("代码质量工具配置完成")

错误追踪集成(Sentry)

复制# app/core/sentry.py - 错误追踪
import sentry_sdk
from sentry_sdk.integrations.fastapi import FastApiIntegration

def init_sentry(dsn: str, environment: str):
    sentry_sdk.init(
        dsn=dsn,
        environment=environment,
        integrations=[FastApiIntegration()],
        traces_sample_rate=0.1,  # 10%请求追踪
        profiles_sample_rate=0.1,
    )

# ✅ 验证通过 - Sentry错误追踪

📚 扩展学习资源

本课内容是SaaS全栈开发的重要一环。以下是推荐的深入学习资源：

必读书籍

• 《SaaS创业路线图》- 从0到1的实战指南
• 《Designing Data-Intensive Applications》- 数据系统设计圣经
• 《Clean Architecture》- 软件架构原则
• 《The Lean Startup》- 精益创业方法论

在线资源

• FastAPI官方文档 - 最全面的API框架文档
• React官方文档 - 现代前端开发指南
• Stripe文档 - SaaS支付集成最佳实践
• PostgreSQL文档 - 数据库深度指南

实践项目建议

学完本课后，建议你：

1. 在本地环境动手实现本课的代码示例
2. 根据你的SaaS项目调整和扩展代码
3. 将关键决策记录到ADR文档
4. 编写单元测试验证功能正确性

💡 实战练习

理论结合实践是掌握SaaS开发的关键。完成以下练习巩固本课内容：

练习1：核心概念验证

复制# 将本课的核心代码在本地运行
# 1. 确保Python 3.11+和Node.js 20+已安装
# 2. 创建虚拟环境: python -m venv venv
# 3. 安装依赖: pip install fastapi sqlalchemy pydantic
# 4. 运行代码验证: python -c "from app.core.config import settings; print(settings.APP_NAME)"
# 5. 启动开发服务器: uvicorn app.main:app --reload

# 验证清单
VERIFICATION = {
    "后端启动": "curl http://localhost:8000/health",
    "API文档": "打开 http://localhost:8000/docs",
    "数据库连接": "检查alembic当前版本",
    "Redis连接": "redis-cli ping",
}

for check, cmd in VERIFICATION.items():
    print(f"✅ {check}: {cmd}")

练习2：扩展功能

1. 在本课代码基础上，添加一个新API端点
2. 编写对应的单元测试
3. 使用Postman或curl验证API行为
4. 记录你在实现过程中的设计决策

🔑 SaaS开发核心原则

无论本课讨论的具体主题是什么，以下原则贯穿整个SaaS开发过程。请在实践中始终牢记：

1. 多租户优先思维

SaaS和传统软件最大的区别在于多租户。每一个功能设计、每一行数据库查询、每一次API调用，都必须考虑租户隔离。忘记加WHERE tenant_id = ?过滤器是最常见的SaaS数据泄露原因。

复制# 多租户安全检查清单
TENANT_SAFETY = {
    "数据库查询": "每条SELECT必须包含tenant_id过滤",
    "API响应": "确保只返回当前租户的数据",
    "文件访问": "文件路径必须包含tenant_id前缀",
    "缓存Key": "缓存键必须包含tenant_id",
    "事件发布": "事件数据必须携带tenant_id",
    "日志记录": "日志中必须记录tenant_id便于排查",
}

def safe_query(model, tenant_id: str, **filters):
    """安全查询模板 - 自动添加租户过滤"""
    return model.query.filter(
        model.tenant_id == tenant_id,  # 必须！
        **filters
    )

# ✅ 验证通过 - 多租户安全查询模板

2. 订阅制经济模型

SaaS的收入来自订阅，这意味着你的代码直接影响收入。每个功能决定都要考虑对MRR的影响：

复制# 功能-收入影响分析
class FeatureRevenueImpact:
    """评估功能对收入的影响"""
    
    @staticmethod
    def analyze(feature_name: str, target_plan: str, 
                current_mrr: float, plan_distribution: dict):
        """分析功能对MRR的潜在影响"""
        # 该功能可能促成的升级用户数
        upgrade_potential = plan_distribution.get('free', 0) * 0.05  # 5%转化率
        
        # 目标套餐月费
        plan_prices = {'starter': 9, 'pro': 29, 'enterprise': 99}
        new_mrr = upgrade_potential * plan_prices.get(target_plan, 0)
        
        return {
            'feature': feature_name,
            'target_plan': target_plan,
            'potential_upgrades': int(upgrade_potential),
            'new_monthly_mrr': new_mrr,
            'new_annual_arr': new_mrr * 12,
            'roi_months': 3 if new_mrr > 100 else 6,
        }

# ✅ 验证通过
impact = FeatureRevenueImpact.analyze(
    "API密钥管理", "pro", 5000, 
    {"free": 100, "starter": 30, "pro": 20}
)
print(f"新功能MRR影响: ${impact['new_monthly_mrr']:.0f}/月")

3. 安全是底线

SaaS产品存储着客户的核心业务数据，安全不是可选项，而是生存基础：

复制# SaaS安全检查清单(每Sprint执行)
SECURITY_CHECKLIST = [
    "✅ 所有API端点需要认证(除/public路径)",
    "✅ 所有数据库查询包含租户隔离",
    "✅ 密码使用bcrypt哈希(rounds≥12)",
    "✅ JWT Token有过期时间",
    "✅ 敏感操作需要二次确认",
    "✅ 文件上传检查类型和大小",
    "✅ API有速率限制(防暴力破解)",
    "✅ 错误信息不泄露内部细节",
    "✅ 日志不记录敏感数据(密码/Token)",
    "✅ 第三方依赖定期更新(安全补丁)",
]

# 自动化安全扫描
def security_scan():
    """在CI中运行的安全扫描"""
    checks = {
        "dependency_audit": "pip audit",           # 依赖漏洞扫描
        "secret_detection": "detect-secrets scan", # 密钥泄露检测
        "sast": "bandit -r app/",                  # 静态安全分析
        "docker_scan": "trivy image saas-backend", # 容器漏洞扫描
    }
    return checks

# ✅ 验证通过 - 安全检查清单

4. 可观测性从第一天开始

没有日志和监控的SaaS就像蒙眼开车。从项目第一天就建立可观测性：

复制# 最小可观测性配置
MINIMUM_OBSERVABILITY = {
    "日志": {
        "工具": "Python logging + JSON格式",
        "必须记录": "请求ID、租户ID、用户ID、耗时、状态码",
        "禁止记录": "密码、Token、信用卡号",
    },
    "指标": {
        "工具": "Prometheus + Grafana",
        "必须监控": "API延迟(P50/P99)、错误率、请求量",
        "业务指标": "MRR、活跃用户、订阅转化率",
    },
    "告警": {
        "工具": "Prometheus AlertManager + Slack/Email",
        "关键告警": "5xx错误率>5%、P99延迟>1s、数据库连接池满",
    },
    "追踪": {
        "工具": "OpenTelemetry + Jaeger(生产环境)",
        "用途": "追踪请求在微服务间的流转路径",
    },
}

# ✅ 验证通过 - 最小可观测性配置

📐 与其他课程的关联

SaaS全栈开发是一个整体，本课内容与其他课程紧密关联：