如何用aws-sdk-go-v2将DynamoDB LastEvaluatedKey序列化为字符串

需要将DynamoDB查询返回的LastEvaluatedKey作为分页API的响应返回，让用户携带该参数请求下一页，询问能否通过aws-sdk-go-v2实现这一转换。尝试直接使用JSON序列化和反序列化但无法正常工作，代码示例如下：

lek := map[string]types.AttributeValue{
    "num":  &types.AttributeValueMemberN{Value: "1"},
    "text": &types.AttributeValueMemberS{Value: "text"},
}
barray, err := json.Marshal(lek)
if err != nil {
  fmt.Println(err)
}

lekDecoded := map[string]types.AttributeValue{}
err = json.Unmarshal(barray, &lekDecoded)
if err != nil {
 fmt.Println(err)
}

这段代码始终无法将数据反序列化为map[string]types.AttributeValue类型。

直接用标准库encoding/json序列化/反序列化map[string]types.AttributeValue会失败，核心原因是types.AttributeValue是接口类型，JSON序列化时只能记录其底层实现的类型标识和值，但反序列化时Go无法自动推断要实例化哪个具体的AttributeValue实现（比如AttributeValueMemberN、AttributeValueMemberS）。

aws-sdk-go-v2专门提供了github.com/aws/aws-sdk-go-v2/service/dynamodb/types/dynamodbattribute包来处理这类转换，下面是正确的实现方式：

1. 序列化LastEvaluatedKey为可返回给前端的JSON #

将LastEvaluatedKey（map[string]types.AttributeValue）转换为通用JSON结构，再序列化为字符串返回：

import (
    "encoding/json"
    "fmt"

    "github.com/aws/aws-sdk-go-v2/service/dynamodb/types"
    "github.com/aws/aws-sdk-go-v2/service/dynamodb/types/dynamodbattribute"
)

// 序列化LastEvaluatedKey到JSON字符串
func serializeLEK(lek map[string]types.AttributeValue) (string, error) {
    // 先转成通用map结构
    lekMap, err := dynamodbattribute.MarshalMap(lek)
    if err != nil {
        return "", fmt.Errorf("marshal lek to map failed: %w", err)
    }
    // 再序列化为JSON字符串
    jsonBytes, err := json.Marshal(lekMap)
    if err != nil {
        return "", fmt.Errorf("marshal map to json failed: %w", err)
    }
    return string(jsonBytes), nil
}

2. 反序列化前端传入的参数为ExclusiveStartKey #

将前端携带的JSON字符串转换回map[string]types.AttributeValue，作为查询的ExclusiveStartKey：

// 反序列化JSON字符串到map[string]types.AttributeValue
func deserializeLEK(jsonStr string) (map[string]types.AttributeValue, error) {
    var lekMap map[string]interface{}
    // 先把JSON字符串转成通用map
    err := json.Unmarshal([]byte(jsonStr), &lekMap)
    if err != nil {
        return nil, fmt.Errorf("unmarshal json to map failed: %w", err)
    }
    // 再转成DynamoDB的AttributeValue map
    lek := make(map[string]types.AttributeValue)
    err = dynamodbattribute.UnmarshalMap(lekMap, &lek)
    if err != nil {
        return nil, fmt.Errorf("unmarshal map to attribute value failed: %w", err)
    }
    return lek, nil
}

完整使用示例 #

func main() {
    // 模拟DynamoDB返回的LastEvaluatedKey
    originalLEK := map[string]types.AttributeValue{
        "num":  &types.AttributeValueMemberN{Value: "1"},
        "text": &types.AttributeValueMemberS{Value: "text"},
    }

    // 序列化并返回给前端
    jsonStr, err := serializeLEK(originalLEK)
    if err != nil {
        fmt.Println("序列化失败:", err)
        return
    }
    fmt.Println("返回给前端的JSON:", jsonStr)

    // 模拟前端传入的参数，反序列化为ExclusiveStartKey
    decodedLEK, err := deserializeLEK(jsonStr)
    if err != nil {
        fmt.Println("反序列化失败:", err)
        return
    }
    fmt.Println("反序列化后的ExclusiveStartKey:", decodedLEK)
}

这段代码会正确完成序列化和反序列化，你可以将序列化后的JSON字符串作为分页API的last_evaluated_key字段返回，用户下次请求时携带该字段，你再反序列化为ExclusiveStartKey传入DynamoDB的查询请求即可。

内容的提问来源于stack exchange，提问作者Sreejith A G