[DynamoDB] Add ScanIndexForward option for Query operation (#26)

evalphobia · Feb 7, 2018 · 18ebc43 · 18ebc43
1 parent fefad5c
commit 18ebc43
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/dynamodb/condition.go b/dynamodb/condition.go
@@ -31,6 +31,7 @@ type ConditionList struct {
 	limit        int64
 	startKey     map[string]*SDK.AttributeValue
 	isConsistent bool
+	isDesc       bool // descending order
 }
 
 // NewConditionList returns initialized *ConditionList.
@@ -77,6 +78,11 @@ func (c *ConditionList) SetConsistent(b bool) {
 	c.isConsistent = b
 }
 
+// SetDesc sets descending order flag.
+func (c *ConditionList) SetDesc(b bool) {
+	c.isDesc = b
+}
+
 // SetStartKey sets ExclusiveStartKey.
 func (c *ConditionList) SetStartKey(startKey map[string]*SDK.AttributeValue) {
 	c.startKey = startKey

diff --git a/dynamodb/table.go b/dynamodb/table.go
@@ -247,7 +247,11 @@ func (t *Table) query(cond *ConditionList, in *SDK.QueryInput) (*QueryResult, er
 	if cond.isConsistent {
 		in.ConsistentRead = pointers.Bool(cond.isConsistent)
 	}
+	if cond.isDesc {
+		in.ScanIndexForward = pointers.Bool(false)
+	}
 
+	in.ExclusiveStartKey = cond.startKey
 	in.TableName = pointers.String(t.nameWithPrefix)
 	req, err := t.service.client.Query(in)
 	if err != nil {