AWS无服务器实时聊天应用：DynamoDB单表下connectionId关联查询问题

针对你遇到的核心问题——仅通过connectionId快速找到对应chatId的连接记录，避免全表扫描，以下是两种贴合DynamoDB单表设计的实用方案：

方案1：添加反向查找条目（推荐） #

在connect函数写入连接记录的同时，额外写入一条以connectionId为分区键的反向关联记录，实现双向映射，保证断开连接时的快速查询：

写入逻辑（Connect函数） #

利用DynamoDB事务操作，确保两条记录原子性写入，避免数据不一致：

await dynamoDb.transactWrite({
  TransactItems: [
    // 原连接记录：用于会话内消息广播
    {
      Put: {
        TableName: 'ChatTable',
        Item: {
          PK: `chat#${chatId}`,
          SK: `connection#${connectionId}`,
          // 可选：添加用户ID、连接时间等业务属性
        }
      }
    },
    // 反向关联记录：用于断开连接时快速定位chatId
    {
      Put: {
        TableName: 'ChatTable',
        Item: {
          PK: `connection#${connectionId}`,
          SK: `chat#${chatId}`,
          chatId: chatId // 冗余存储chatId，简化后续读取逻辑
        }
      }
    }
  ]
}).promise();

删除逻辑（Disconnect函数） #

通过connectionId查询反向记录拿到chatId，再事务删除两条关联记录：

// 第一步：获取反向记录，拿到对应的chatId
const reverseRecord = await dynamoDb.get({
  TableName: 'ChatTable',
  Key: {
    PK: `connection#${connectionId}`,
    SK: `chat#${chatId}`
  }
}).promise();

if (reverseRecord.Item) {
  const chatId = reverseRecord.Item.chatId;
  // 第二步：原子删除原连接记录和反向记录
  await dynamoDb.transactWrite({
    TransactItems: [
      {
        Delete: {
          TableName: 'ChatTable',
          Key: {
            PK: `chat#${chatId}`,
            SK: `connection#${connectionId}`
          }
        }
      },
      {
        Delete: {
          TableName: 'ChatTable',
          Key: {
            PK: `connection#${connectionId}`,
            SK: `chat#${chatId}`
          }
        }
      }
    ]
  }).promise();
}

方案2：使用全局二级索引（GSI） #

若无法修改写入逻辑，可通过GSI实现反向查询：

1. 在DynamoDB表中创建GSI，将SK设为GSI的分区键，投影类型选择ALL或仅包含PK字段
2. 在disconnect函数中，通过GSI查询SK = connection#${connectionId}的条目，直接获取对应的PK（即chat#xxx）
3. 拿到PK后即可删除原连接记录

注意：GSI会产生额外的存储和读写开销，查询延迟略高于主键查询，适合无法调整写入逻辑的场景。

内容的提问来源于stack exchange，提问作者Ilya Shcherbakov