DynamoDB Marshal 和 unmarshal golang 时间。自纪元以来的时间(以毫秒为单位)

Jas*_*son 3 go amazon-web-services amazon-dynamodb aws-sdk-go

默认情况下,sdk 将time.Time值编组为 RFC3339 字符串。您如何选择以其他方式编组和解组,例如自纪元以来的毫秒数?

SDK 提到了 Marshaler 和 Unmarshaler 接口,但没有解释如何使用它们。

Jas*_*son 8

(当我正要发布我的问题时,我通过研究UnixTime工作原理找到了答案)。

要使用自定义编组器和解组器,您可以创建自定义类型。

type MillisTime time.Time

func (e MillisTime) MarshalDynamoDBAttributeValue(av *dynamodb.AttributeValue) error {
    millis := timeAsMillis(time.Time(e))
    millisStr := fmt.Sprintf("%d", millis)
    av.N = &millisStr
    return nil
}

func (e *MillisTime) UnmarshalDynamoDBAttributeValue(av *dynamodb.AttributeValue) error {
    millis, err := strconv.ParseInt(*av.N, 10, 0)
    if err != nil {
        return err
    }
    *e = MillisTime(millisAsTime(millis))
    return nil
}

func timeAsMillis(t time.Time) int64 {
    nanosSinceEpoch := t.UnixNano()
    return (nanosSinceEpoch / 1_000_000_000) + (nanosSinceEpoch % 1_000_000_000)
}

func millisAsTime(millis int64) time.Time {
    seconds := millis / 1_000
    nanos := (millis % 1_000) * 1_000_000
    return time.Unix(seconds, nanos)
}
Run Code Online (Sandbox Code Playgroud)

注意:上面的示例使用了 go 1.13 中引入的新数字文字语法。

MarshalMap您可以使用and轻松编组和解组结构,UnmarshalMap但缺点是结构类型中的字段必须使用 MillisTime 而不是time.Time。转换并不容易,但却是可能的。

SDK 定义了一种UnixTime类型,该类型将在自纪元以来的 <=> 秒之间处理编组和解组time.Time