如何用Go的DynamoDB Expression包实现多条件Query查询

你需要用Go语言实现对应以下AWS CLI的DynamoDB Query操作：

aws dynamodb query \
  --table-name myTable \
  --index-name name-index \
  --key-condition-expression "_pk = :dpid and begins_with(st, :st)" \
  --expression-attribute-values  '{":dpid":{"S":"dp#1"}, ":st":{"S":"jam"}}'

使用AWS SDK v2的Expression包时遇到困惑，当前代码如下：

func (p *PatientStore) SearchPatients(ctx context.Context, searchTerm string) ([]PatientSearchResponseItem, error) {
    dentalPracticeId := "dp#1"
    lowerCaseSearchTerm := strings.ToLower(searchTerm)
    var err error
    var response *dynamodb.QueryOutput
    var patients []PatientSearchResponseItem
    keyEx := expression.Key("_pk").Equal(expression.Value(dentalPracticeId))
    gsiCondition := expression.BeginsWith(expression.Name("st"), lowerCaseSearchTerm) // gsiCondition declared but not used error
    expr, err := expression.NewBuilder().WithKeyCondition(keyEx).Build()
    if err != nil {
        log.Printf("Couldn't build epxression for query. Here's why: %v\n", err)
    } else {
        response, err = p.client.Query(context.TODO(), &dynamodb.QueryInput{
            TableName:                 aws.String(p.tableName),
            IndexName:                 jsii.String("name-index"),
            ExpressionAttributeNames:  expr.Names(),
            ExpressionAttributeValues: expr.Values(),
            KeyConditionExpression:    expr.KeyCondition(),
        })
        if err != nil {
            log.Printf("Couldn't query for patients matching %v. Here's why: %v\n", lowerCaseSearchTerm, err)
        } else {
            err = attributevalue.UnmarshalListOfMaps(response.Items, &patients)
            if err != nil {
                log.Printf("Couldn't unmarshal query response. Here's why: %v\n", err)
            }
        }
    }

    return patients, err
}

当前遇到gsiCondition is declared but not used错误，不知道如何组合多个Key条件。

解决方案 #

要组合多个Key条件，需用expression.And()把两个条件合并后传入WithKeyCondition()，同时修正代码里的几个细节问题：

• 替换jsii.String为aws.String（jsii属于CDK包，此处应使用AWS SDK v2的类型）
• 用函数传入的ctx替代context.TODO()，保证上下文传递正确

修改后的代码如下：

func (p *PatientStore) SearchPatients(ctx context.Context, searchTerm string) ([]PatientSearchResponseItem, error) {
    dentalPracticeId := "dp#1"
    lowerCaseSearchTerm := strings.ToLower(searchTerm)
    var err error
    var response *dynamodb.QueryOutput
    var patients []PatientSearchResponseItem

    // 组合两个Key条件：_pk等于指定值 且 st以搜索词开头
    pkCondition := expression.Key("_pk").Equal(expression.Value(dentalPracticeId))
    stCondition := expression.Key("st").BeginsWith(lowerCaseSearchTerm) // st是GSI排序键，需用expression.Key引用
    keyCondition := expression.And(pkCondition, stCondition)

    expr, err := expression.NewBuilder().WithKeyCondition(keyCondition).Build()
    if err != nil {
        log.Printf("Couldn't build expression for query. Here's why: %v\n", err)
        return nil, err
    }

    response, err = p.client.Query(ctx, &dynamodb.QueryInput{
        TableName:                 aws.String(p.tableName),
        IndexName:                 aws.String("name-index"),
        ExpressionAttributeNames:  expr.Names(),
        ExpressionAttributeValues: expr.Values(),
        KeyConditionExpression:    expr.KeyCondition(),
    })
    if err != nil {
        log.Printf("Couldn't query for patients matching %v. Here's why: %v\n", lowerCaseSearchTerm, err)
        return nil, err
    }

    err = attributevalue.UnmarshalListOfMaps(response.Items, &patients)
    if err != nil {
        log.Printf("Couldn't unmarshal query response. Here's why: %v\n", err)
        return nil, err
    }

    return patients, nil
}

关键说明 #

1. 组合Key条件：DynamoDB Query的KeyConditionExpression支持用AND组合分区键与排序键条件，通过expression.And()合并两个条件后，传给WithKeyCondition()即可生成正确的表达式。
2. 用expression.Key而非Name：st是GSI的排序键，属于Key属性范畴，必须用expression.Key()引用；expression.Name()仅用于非Key属性（如FilterExpression中的属性）。
3. 修正类型与上下文：替换错误的字符串类型转换函数，同时使用传入的上下文对象，确保资源能正确释放、请求上下文一致。

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