如何基于DynamoDB GSI按指定名称及SSN列表查询数据？

针对你的需求，有两种高效的实现方式，避免低效的全量过滤：

1. 使用BatchGetItem批量获取 #

因为你的GSI中name（分区键）+ssn（排序键）可以唯一定位一条记录，所以直接用BatchGetItem API一次性批量查询多个ssn对应的条目是最推荐的方式，能有效节省读取容量（RCU）。

示例代码：

import boto3

dynamodb = boto3.resource('dynamodb')
table = dynamodb.Table('你的表名')

name = 'Alex'
ssns = ['ssn1', 'ss2', 'ss3', 'ssn4']

# 构造批量查询的键列表
keys = [{'name': name, 'ssn': ssn} for ssn in ssns]

response = table.batch_get_item(
    RequestItems={
        '你的表名': {
            'Keys': keys,
            'IndexName': 'lookup-by-name'  # 指定要使用的GSI
        }
    }
)

# 获取结果
items = response['Responses']['你的表名']
print(items)

注意：BatchGetItem单次最多支持100个键，如果你的ssn列表超过100个，需要分批次处理。

2. 并行执行多个Query请求 #

如果不想用批量接口，也可以针对每个ssn单独发起Query请求，再合并结果。为了提升效率，用多线程或异步方式并行执行这些请求，避免串行等待的耗时。

示例代码（用concurrent.futures实现并行）：

import boto3
from concurrent.futures import ThreadPoolExecutor
from boto3.dynamodb.conditions import Key

dynamodb = boto3.resource('dynamodb')
table = dynamodb.Table('你的表名')

name = 'Alex'
ssns = ['ssn1', 'ss2', 'ss3', 'ssn4']

def query_by_ssn(ssn):
    response = table.query(
        IndexName='lookup-by-name',
        KeyConditionExpression=Key('name').eq(name) & Key('ssn').eq(ssn)
    )
    return response['Items']

# 并行执行所有查询
with ThreadPoolExecutor(max_workers=len(ssns)) as executor:
    results = executor.map(query_by_ssn, ssns)

# 合并所有结果
all_items = []
for res in results:
    all_items.extend(res)

print(all_items)

不推荐的方式：使用FilterExpression #

不要直接在Query里加FilterExpression=Attr('ssn').is_in(ssns)，这种方式会先把name='Alex'的所有记录都查出来，再在客户端过滤出ssn在列表中的条目，会浪费大量RCU，而且如果该name下的记录很多，性能会非常差。

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