高并发场景，关系型数据库不该是唯一选择

"""DynamoDB 单表设计示例：用户 - 订单 - 商品

访问模式（先定义清楚，再设计键）：
    AP1  按 userId 查用户资料
    AP2  按 userId 查该用户的订单（按时间倒序，可分页）
    AP3  按 orderId 查订单详情与明细
    AP4  按订单状态查所有订单（跨用户）—— 用 GSI1
    AP5  按 skuId 查该商品的销售记录 —— 用 GSI2
"""

import boto3
from datetime import datetime, timezone
from typing import Any

table = boto3.resource("dynamodb").Table("mushan-commerce")


def put_user(user_id: str, profile: dict[str, Any]) -> None:
    """AP1：用户主记录。"""
    table.put_item(Item={
        "PK": f"USER#{user_id}",
        "SK": "PROFILE",
        "entityType": "User",
        **profile,
    })


def put_order(user_id: str, order_id: str, status: str, amount: str) -> None:
    """AP2 / AP3 / AP4：订单头记录，同时写 GSI1 键。"""
    now = datetime.now(timezone.utc).isoformat()
    table.put_item(Item={
        "PK": f"USER#{user_id}",
        "SK": f"ORDER#{now}#{order_id}",
        "entityType": "Order",
        "orderId": order_id,
        "status": status,
        "amount": amount,
        "createdAt": now,
        # GSI1：按状态查订单
        "GSI1PK": f"STATUS#{status}",
        "GSI1SK": now,
    })


def query_user_orders(user_id: str, limit: int = 20, cursor: dict | None = None):
    """AP2：查某用户订单，时间倒序分页。一次 Query 搞定，不用 Scan。"""
    kwargs = {
        "KeyConditionExpression": (
            boto3.dynamodb.conditions.Key("PK").eq(f"USER#{user_id}")
            & boto3.dynamodb.conditions.Key("SK").begins_with("ORDER#")
        ),
        "ScanIndexForward": False,   # 倒序，最新的在前
        "Limit": limit,
    }
    if cursor:
        kwargs["ExclusiveStartKey"] = cursor
    resp = table.query(**kwargs)
    return resp.get("Items", []), resp.get("LastEvaluatedKey")


def query_orders_by_status(status: str, limit: int = 50):
    """AP4：跨用户按状态查订单，走 GSI1。"""
    resp = table.query(
        IndexName="GSI1",
        KeyConditionExpression=boto3.dynamodb.conditions.Key("GSI1PK").eq(f"STATUS#{status}"),
        ScanIndexForward=False,
        Limit=limit,
    )
    return resp.get("Items", [])