[DynamoDB] feat: Add 'ConditionList.AddKeyAttribute' method (#50)

evalphobia · Feb 17, 2020 · 841859e · 841859e
1 parent b3e16ab
commit 841859e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/dynamodb/condition.go b/dynamodb/condition.go
@@ -241,6 +241,11 @@ func (c *ConditionList) getMergedConditions() map[string]*Condition {
 	return list
 }
 
+// AddKeyAttribute adds to attribute to keyAttributes.
+func (c *ConditionList) AddKeyAttribute(attr AttributeDefinition) {
+	c.keyAttributes[attr.Name] = attr.Type
+}
+
 // Condition contains condition.
 type Condition struct {
 	Condition string