在 dynamoDB 中更新或创建嵌套元素

Question

我想更新或创建 DynamoDB 项目以获取下一个元素：

{
    "id": 156, 
    "date": 12323233.000,
    "countries": {
        "ua": 1,
        "ru": 2}
}

我使用 python 和 boto3。所以我可以检查字段是否countries存在，如果不存在则添加它。但这将意味着 2 个 DB 请求。

table.update_item(
    Key={
        'id': 156,
        'date': date,
    },
    UpdateExpression='SET countries = if_not_exists(countries, :countries)',
    ExpressionAttributeValues={
        ':countries': {},
    },
)

table.update_item(
    Key={
        'id': 156,
        'date': date,
    },
    UpdateExpression='ADD countries.#country :inc',
    ExpressionAttributeNames={"#country": country},  
    ExpressionAttributeValues={
        ':inc': 1
    },
)

有没有办法将这 2 个请求合二为一？

Answer 1

我最近不得不做这样的事情，花了一段时间才弄清楚。如果我的“完成”页面集中不存在页面，我想增加一个计数。如果没有，它会增加页码并将其添加到集合中。花了一段时间才意识到您可以“附加”到列表，但必须“添加”到集合。

try:
    res = dbt.update_item(
        Key={'pk': 'doc3', 'sk': 'na'},
        ReturnConsumedCapacity='INDEXES', ReturnValues='ALL_NEW',
        ExpressionAttributeNames={
            '#count': 'count',
            '#done': 'done',
        },
        ExpressionAttributeValues={
            ':1': 1,
            ':page': page,
            ':pagelist': set([page]),
        },
        ConditionExpression="(NOT contains(done, :page))",
        UpdateExpression="ADD #done :pagelist, #count :1",
    )
    print(f'rand int page={page} count={res["Attributes"]["count"]}'
          f' CU={res["ConsumedCapacity"]["Table"]}')
except ClientError as err:
    if err.response['Error']['Code'] == 'ConditionalCheckFailedException':
        print('Already got page=%s (%s)' % (page, err))
    else:
        raise