add NewNumberSetAttribute (#167)

Author: daviskoh

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

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())