使用dataset_id与begins_with查询DynamoDB，代码存问题求排查

问题描述 #

需要同时通过dataset_id和以'a'开头的image_name查询DynamoDB，使用的代码如下：

dynamodb = boto3.resource('dynamodb')
table = dynamodb.Table('table_name') 
response = table.query(
           IndexName='dataset_id',
           KeyConditionExpression='dataset_id = :value AND begins_with (image_name, :name)',
           ExpressionAttributeValues={
              ':value': str(dataset_id),
              ':name': {'S', 'a'}
           },
           Limit=int(results_per_page)

对应的全局二级索引（GSI）结构：分区键为dataset_id，需确认排序键是否为image_name。

代码中的错误点 #

• ExpressionAttributeValues格式错误：使用boto3的resource模式时，不需要用DynamoDB的低级数据类型格式（如{'S': 'a'}），直接传入Python原生类型即可，低级格式仅适用于client模式。
• KeyConditionExpression语法不规范：begins_with函数的参数和括号之间不应有空格，正确写法是begins_with(image_name, :name)。
• 查询条件依赖GSI结构：如果你的GSI仅将dataset_id作为分区键，没有把image_name设为排序键，那么无法在KeyConditionExpression中对image_name使用begins_with——因为该表达式只能针对索引的分区键和排序键进行条件组合。
• 代码语法不完整：table.query()方法调用最后缺少闭合的)，会导致语法报错。

修正后的代码 #

假设你的GSI确实是分区键dataset_id+排序键image_name，修正后的代码如下：

dynamodb = boto3.resource('dynamodb')
table = dynamodb.Table('table_name') 
response = table.query(
           IndexName='dataset_id',
           KeyConditionExpression='dataset_id = :value AND begins_with(image_name, :name)',
           ExpressionAttributeValues={
              ':value': str(dataset_id),
              ':name': 'a'
           },
           Limit=int(results_per_page)
)

额外说明 #

如果你的GSI没有将image_name设为排序键，那么需要重新创建符合要求的GSI（分区键dataset_id，排序键image_name），否则无法通过query实现该查询逻辑——此时只能改用scan，但scan性能较差，不建议用于大规模数据查询。

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