diff --git a/events/attributevalue.go b/events/attributevalue.go index 530f54db..0e645753 100644 --- a/events/attributevalue.go +++ b/events/attributevalue.go @@ -167,6 +167,14 @@ func NewNumberAttribute(value string) DynamoDBAttributeValue { return av } +// NewNumberSetAttribute creates an DynamoDBAttributeValue containing a NumberSet +func NewNumberSetAttribute(value []string) DynamoDBAttributeValue { + var av DynamoDBAttributeValue + av.value = value + av.dataType = DataTypeNumberSet + return av +} + // NewNullAttribute creates an DynamoDBAttributeValue containing a Null func NewNullAttribute() DynamoDBAttributeValue { var av DynamoDBAttributeValue diff --git a/events/attributevalue_test.go b/events/attributevalue_test.go index 7b298c25..5bb63f32 100644 --- a/events/attributevalue_test.go +++ b/events/attributevalue_test.go @@ -296,6 +296,12 @@ func Test_DynamoDBAttributeValue_NewAttribute(t *testing.T) { assert.Nil(t, err) assert.Equal(t, float64(1.1), v) } + { + value := []string{"2", "2.2"} + av := NewNumberSetAttribute(value) + assert.Equal(t, DataTypeNumberSet, av.DataType()) + assert.Equal(t, value, av.NumberSet()) + } { av := NewNullAttribute() assert.Equal(t, DataTypeNull, av.DataType())