如何在AppSync DynamoDB中高效批量删除指定字段匹配的条目？

针对你需要删除单个ChatRoom下数千条Messages的场景，仅用AppSync原生BatchWrite解析器无法满足需求，这里提供两种可行方案：

方案一：AppSync自定义解析器+分页循环删除 #

这种方案无需额外服务，但受限于AppSync解析器的30秒执行超时，适合数据量在几百条以内的场景：

1. 定义自定义GraphQL突变
   在你的schema中添加：

type Mutation {
  deleteMessagesByChatRoomId(chatRoomId: ID!, batchSize: Int = 25): Boolean
}

2. 编写VTL解析器
   创建对应的请求模板（Request Mapping Template）：

#set($limit = $util.defaultIfNull($ctx.args.batchSize, 25))
#set($query = {
  "version": "2018-05-29",
  "operation": "Query",
  "query": {
    "expression": "chatRoomId = :chatRoomId",
    "expressionValues": {
      ":chatRoomId": $util.dynamodb.toDynamoDB($ctx.args.chatRoomId)
    }
  },
  "limit": $limit
})

#set($deleteRequests = [])
#foreach($item in $ctx.result.items)
  #set($deleteRequest = {
    "DeleteRequest": {
      "Key": {
        "id": $util.dynamodb.toDynamoDB($item.id)
      }
    }
  })
  $util.qr($deleteRequests.add($deleteRequest))
#end

#set($batchWrite = {
  "version": "2018-05-29",
  "operation": "BatchWriteItem",
  "tables": {
    "Messages": $deleteRequests
  }
})

$util.toJson($batchWrite)

响应模板直接返回true即可。

注意：如果单次查询返回的条目数不足batchSize，说明已删除完毕；如果数据量过大，需要前端多次调用该突变，直到返回的删除条目数为0。

方案二：Lambda中间层处理大规模批量删除 #

这种方案适合数千条甚至更多数据的场景，利用Lambda的15分钟超时时间完成全量删除：

1. 创建Lambda函数
   用Node.js编写示例代码：

const AWS = require('aws-sdk');
const dynamodb = new AWS.DynamoDB.DocumentClient();

const BATCH_SIZE = 25;

exports.handler = async (event) => {
  const { chatRoomId } = event.arguments;
  let lastEvaluatedKey = null;

  do {
    // 分页查询当前批次的Message主键
    const queryParams = {
      TableName: 'Messages',
      KeyConditionExpression: 'chatRoomId = :chatRoomId',
      ExpressionAttributeValues: { ':chatRoomId': chatRoomId },
      ProjectionExpression: 'id',
      Limit: BATCH_SIZE,
      ExclusiveStartKey: lastEvaluatedKey
    };

    const queryResult = await dynamodb.query(queryParams).promise();
    const items = queryResult.Items;

    if (items.length === 0) break;

    // 构造批量删除请求
    const deleteRequests = items.map(item => ({
      DeleteRequest: {
        Key: { id: item.id }
      }
    }));

    const batchParams = {
      RequestItems: {
        'Messages': deleteRequests
      }
    };

    await dynamodb.batchWrite(batchParams).promise();
    lastEvaluatedKey = queryResult.LastEvaluatedKey;

  } while (lastEvaluatedKey);

  return true;
};

2. 在AppSync中配置Lambda解析器

• 在Amplify控制台中，将上述Lambda函数与AppSync关联
• 在schema中添加相同的deleteMessagesByChatRoomId突变，将其解析器设置为该Lambda函数

额外注意事项 #

• 确保Messages表已创建chatRoomId的GSI（全局二级索引）或主键包含chatRoomId，否则查询效率极低
• 批量删除操作会消耗DynamoDB的读写容量，大数量删除时注意监控容量使用情况，避免影响其他业务

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